Update blaxel_main.py

blaxel_main.py

@@ -1,8 +1,10 @@
-# blaxel_main.py (Smoke Test)
+# blaxel_main.py (Smoke Test - Now with Blaxel-specific ports)
 from fastapi import FastAPI
 from pydantic import BaseModel
 from sse_starlette.sse import EventSourceResponse
 import asyncio
+import os  # <-- 1. Import os
+import uvicorn # <-- 2. Import uvicorn
 
 app = FastAPI()
 
@@ -23,4 +25,18 @@ async def solve_problem(request: ProblemRequest):
         await asyncio.sleep(0.5)
         yield "Test complete. The connection is good!"
 
-    return EventSourceResponse(stream_test())
+    return EventSourceResponse(stream_test())
+
+# --- 3. This is the new, critical part ---
+if __name__ == "__main__":
+    # Blaxel provides the host and port through environment variables
+    # We must use these.
+    HOST = os.getenv("BL_SERVER_HOST", "0.0.0.0")
+    
+    # We use 'int()' to convert the port from a string to a number
+    # Use 8080 as a fallback for local testing
+    PORT = int(os.getenv("BL_SERVER_PORT", "8080"))
+
+    # This line replaces the "Start Command" you were looking for.
+    # It tells our app how to run.
+    uvicorn.run(app, host=HOST, port=PORT)