Setup the Hugging face deployment

.env

@@ -0,0 +1 @@
+DATABASE_URL='postgresql://neondb_owner:npg_0ZThyfD5xJGw@ep-polished-base-ab35nufp.eu-west-2.aws.neon.tech/neondb?sslmode=require&channel_binding=require'

main.py

@@ -1,20 +1,23 @@
+import os
 from fastapi import FastAPI
-from fastapi.middleware.cors import CORSMiddleware
+from sqlalchemy import create_engine
+from sqlalchemy.orm import sessionmaker
 from dotenv import load_dotenv
-import os
 
-# Load variables from .env
+# This will pick up the secret you saved in Hugging Face
 load_dotenv()
 
-app = FastAPI()
+DATABASE_URL = os.getenv("DATABASE_URL")
 
-app.add_middleware(
-    CORSMiddleware,
-    allow_origins=["*"],
-    allow_methods=["*"],
-    allow_headers=["*"],
-)
+# Create the SQLAlchemy engine using the modern psycopg driver
+engine = create_engine(DATABASE_URL)
+SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
+
+app = FastAPI()
 
-@app.get("/")
-def read_root():
-    return {"status": "SmiloCAD API is Live", "db_connected": bool(os.getenv("DATABASE_URL"))}
+@app.get("/health")
+def health_check():
+    # This checks if the URL was loaded correctly from the secrets
+    if DATABASE_URL:
+        return {"status": "Database URL is loaded"}
+    return {"status": "Error: DATABASE_URL not found"}