PRANAV05092003 commited on
Commit
9c6cff5
·
1 Parent(s): e7eb0fa

Final multi-mode OpenEnv fix

Browse files
Files changed (2) hide show
  1. pyproject.toml +1 -1
  2. server/app.py +16 -0
pyproject.toml CHANGED
@@ -19,7 +19,7 @@ dependencies = [
19
  ]
20
 
21
  [project.scripts]
22
- server = "server.app:app"
23
 
24
  [tool.setuptools.packages.find]
25
  where = ["."]
 
19
  ]
20
 
21
  [project.scripts]
22
+ server = "server.app:main"
23
 
24
  [tool.setuptools.packages.find]
25
  where = ["."]
server/app.py CHANGED
@@ -19,6 +19,7 @@ from fastapi import FastAPI, HTTPException
19
  from fastapi.middleware.cors import CORSMiddleware
20
  from fastapi.responses import HTMLResponse, JSONResponse
21
  from openai import OpenAI
 
22
 
23
  # Ensure project root is importable when executed in Spaces/Docker.
24
  PROJECT_ROOT = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
@@ -394,3 +395,18 @@ def optimize(req: OptimizeRequest) -> OptimizeResponse:
394
  task_score=task_score,
395
  )
396
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
19
  from fastapi.middleware.cors import CORSMiddleware
20
  from fastapi.responses import HTMLResponse, JSONResponse
21
  from openai import OpenAI
22
+ import uvicorn
23
 
24
  # Ensure project root is importable when executed in Spaces/Docker.
25
  PROJECT_ROOT = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
 
395
  task_score=task_score,
396
  )
397
 
398
+
399
+ def main() -> None:
400
+ """
401
+ Entry point for OpenEnv multi-mode deployment.
402
+
403
+ - API mode: OpenEnv imports `server.app:app`
404
+ - CLI mode: OpenEnv / HF can run `server` script -> `server.app:main`
405
+ """
406
+ port = int(os.getenv("PORT", "7860"))
407
+ uvicorn.run("server.app:app", host="0.0.0.0", port=port)
408
+
409
+
410
+ if __name__ == "__main__":
411
+ main()
412
+