Merge branch 'main' of https://github.com/callaunchpad/aura-analyzer

main.py

@@ -9,6 +9,7 @@
 from sqlmodel import Field, Session, SQLModel, or_, create_engine, select
 from sqlalchemy import func
 import subprocess
+import uvicorn
 
 class AuraRequest(BaseModel):
     Department: Literal['menswear', 'womenswear']
@@ -56,6 +57,10 @@ def get_session():
 
 app = FastAPI()
 
+# Ensure the app listens on the correct port
+if __name__ == '__main__':
+    uvicorn.run('main:app', host='0.0.0.0', port=8000)
+
 app.add_middleware(
     CORSMiddleware,
     allow_origins=["*"],  # Allow all origins
@@ -68,6 +73,10 @@ def get_session():
 def read_root():
     return {"Hello": "World"}
 
+@app.get("/health")
+def health_check():
+    return {"status": "ok"}
+
 @app.on_event("startup")
 def on_startup():
     create_db_and_tables()

requirements.txt

@@ -12,7 +12,8 @@ opencv-python
 Pylette
 mediapipe
 imutils
-fastapi
+fastapi[all]
 uvicorn
 sqlmodel
 gunicorn
+SQLAlchemy

startup.sh

@@ -1 +1,3 @@
-gunicorn -w 4 -k uvicorn.workers.UvicornWorker main:app
+#!/bin/bash
+python3 -m pip install -r requirements.txt
+python -m uvicorn main:app --host 0.0.0.0