fix: unicorn app (#244)

petercat-ai · Aug 21, 2024 · f4a89b0 · f4a89b0
2 parents 8dc07c0 + 8176b14
commit f4a89b0
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/server/main.py b/server/main.py
@@ -1,3 +1,6 @@
+import os
+
+import uvicorn
 
 from fastapi import FastAPI
 from starlette.middleware.sessions import SessionMiddleware
@@ -49,8 +52,8 @@
 app.include_router(task.router)
 
 
-# if __name__ == "__main__":
-#     if is_dev:
-#         uvicorn.run("main:app", host="0.0.0.0", port=int(os.environ.get("PORT", "8080")), reload=True)
-#     else:
-#         uvicorn.run(app, host="0.0.0.0", port=int(os.environ.get("PORT", "8080")))
+if __name__ == "__main__":
+    if is_dev:
+        uvicorn.run("main:app", host="0.0.0.0", port=int(os.environ.get("PORT", "8080")), reload=True)
+    else:
+        uvicorn.run(app, host="0.0.0.0", port=int(os.environ.get("PORT", "8080")))