Update app.py

app.py

@@ -8,10 +8,10 @@ from gradio.routes import mount_gradio_app
 from retriever import get_relevant_passages
 from reranker import rerank
 
-# FastAPI app instance
+# --- Initialize FastAPI ---
 app = FastAPI()
 
-# --- CSV Loading and Cleaning ---
+# --- Load and clean CSV ---
 def clean_df(df):
     df = df.copy()
     second_col = df.iloc[:, 2].astype(str)
@@ -25,11 +25,7 @@ def clean_df(df):
     df["adaptive_support"] = df.iloc[:, 4].map(lambda x: "Yes" if x == "T" else "No")
     df["test_type"] = df.iloc[:, 5].apply(lambda x: eval(x) if isinstance(x, str) else x)
     df["description"] = df.iloc[:, 6]
-
-    df["duration"] = pd.to_numeric(
-        df.iloc[:, 9].astype(str).str.extract(r'(\d+)')[0],
-        errors='coerce'
-    )
+    df["duration"] = pd.to_numeric(df.iloc[:, 9].astype(str).str.extract(r'(\d+)')[0], errors='coerce')
 
     return df[["url", "adaptive_support", "remote_support", "description", "duration", "test_type"]]
 
@@ -40,7 +36,7 @@ except Exception as e:
     print(f"Error loading data: {e}")
     df_clean = pd.DataFrame(columns=["url", "adaptive_support", "remote_support", "description", "duration", "test_type"])
 
-# --- Utility ---
+# --- Fix URLs ---
 def validate_and_fix_urls(candidates):
     for candidate in candidates:
         if not isinstance(candidate, dict):
@@ -56,7 +52,7 @@ def validate_and_fix_urls(candidates):
             candidate['url'] = f"https://www.shl.com{url}" if url.startswith('/') else f"https://www.shl.com/{url}"
     return candidates
 
-# --- Core Recommend Logic ---
+# --- Recommendation Logic ---
 def recommend(query):
     if not query.strip():
         return {"error": "Please enter a job description"}
@@ -90,33 +86,39 @@ iface = gr.Interface(
     description="Paste a job description to get the most relevant SHL assessments."
 )
 
-mount_gradio_app(app, iface, path="/") 
-# --- Mount Gradio at "/" ---
-#app = mount_gradio_app(app, iface, path="/")
+# ✅ Mount Gradio UI without overwriting `app`
+mount_gradio_app(app, iface, path="/")
 
-# --- /health Endpoint ---
-@app.get("/health", response_class=JSONResponse)
+# --- Health Endpoint ---
+@app.get("/health")
 async def health_check():
     return JSONResponse(
-        status_code=200,
         content={"status": "healthy"},
-        media_type="application/json"
+        media_type="application/json",
+        status_code=200
     )
 
-# --- /recommend Endpoint ---
-@app.post("/recommend", response_class=JSONResponse)
+# --- API Recommendation Endpoint ---
+@app.post("/recommend")
 async def recommend_api(request: Request):
-    body = await request.json()
-    query = body.get("query", "").strip()
-    if not query:
+    try:
+        body = await request.json()
+        query = body.get("query", "").strip()
+        if not query:
+            return JSONResponse(
+                content={"error": "Missing 'query' in request body"},
+                media_type="application/json",
+                status_code=400
+            )
+        result = recommend(query)
         return JSONResponse(
-            status_code=400,
-            content={"error": "Missing 'query' in request body"},
-            media_type="application/json"
+            content=result,
+            media_type="application/json",
+            status_code=200
+        )
+    except Exception as e:
+        return JSONResponse(
+            content={"error": str(e)},
+            media_type="application/json",
+            status_code=500
         )
-    result = recommend(query)
-    return JSONResponse(
-        status_code=200,
-        content=result,
-        media_type="application/json"
-    )