Spaces:
Build error
Build error
add healthcheck to server
Browse files- main.py +6 -1
- src/server.py +6 -1
main.py
CHANGED
@@ -14,6 +14,7 @@ import logging
|
|
14 |
from dotenv import load_dotenv
|
15 |
import uvicorn
|
16 |
import sys
|
|
|
17 |
|
18 |
# Import from src modules
|
19 |
from src.processor import process_leaderboards
|
@@ -79,8 +80,12 @@ def run_server_mode(args):
|
|
79 |
# Log startup information
|
80 |
logger.info("Running in server mode with periodic processing")
|
81 |
|
|
|
|
|
|
|
|
|
82 |
# Run the FastAPI server
|
83 |
-
uvicorn.run(app, host="0.0.0.0", port=
|
84 |
except KeyboardInterrupt:
|
85 |
logger.info("Server stopped by user")
|
86 |
except Exception as e:
|
|
|
14 |
from dotenv import load_dotenv
|
15 |
import uvicorn
|
16 |
import sys
|
17 |
+
import os
|
18 |
|
19 |
# Import from src modules
|
20 |
from src.processor import process_leaderboards
|
|
|
80 |
# Log startup information
|
81 |
logger.info("Running in server mode with periodic processing")
|
82 |
|
83 |
+
# Get port from environment variable or use default HF Spaces port
|
84 |
+
port = int(os.environ.get("PORT", 7860))
|
85 |
+
logger.info(f"Starting server on port {port}")
|
86 |
+
|
87 |
# Run the FastAPI server
|
88 |
+
uvicorn.run(app, host="0.0.0.0", port=port)
|
89 |
except KeyboardInterrupt:
|
90 |
logger.info("Server stopped by user")
|
91 |
except Exception as e:
|
src/server.py
CHANGED
@@ -190,4 +190,9 @@ async def trigger_run():
|
|
190 |
# Start processing in a separate thread
|
191 |
threading.Thread(target=lambda: process_leaderboards()).start()
|
192 |
|
193 |
-
return {"status": "started", "message": "Processing started"}
|
|
|
|
|
|
|
|
|
|
|
|
190 |
# Start processing in a separate thread
|
191 |
threading.Thread(target=lambda: process_leaderboards()).start()
|
192 |
|
193 |
+
return {"status": "started", "message": "Processing started"}
|
194 |
+
|
195 |
+
@app.get("/health")
|
196 |
+
async def health_check():
|
197 |
+
"""Health check endpoint for HF Spaces to verify the app is running"""
|
198 |
+
return {"status": "ok"}
|