Update app.py

app.py

@@ -1,16 +1,23 @@
 import pandas as pd
 import gradio as gr
+from fastapi import FastAPI, Request
+from fastapi.responses import JSONResponse
 from retriever import get_relevant_passages
 from reranker import rerank
 
+# === FastAPI App ===
+app = FastAPI()
+
 # === Load and Clean CSV ===
 def clean_df(df):
     df = df.copy()
     second_col = df.iloc[:, 2].astype(str)
+    
     if second_col.str.contains('http').any() or second_col.str.contains('www').any():
         df["url"] = second_col
     else:
         df["url"] = "https://www.shl.com" + second_col.str.replace(r'^(?!/)', '/', regex=True)
+    
     df["remote_support"] = df.iloc[:, 3].map(lambda x: "Yes" if x == "T" else "No")
     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)
@@ -25,6 +32,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 ===
 def validate_and_fix_urls(candidates):
     for candidate in candidates:
         if not isinstance(candidate, dict):
@@ -40,6 +48,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
 
+# === Recommendation Logic ===
 def recommend(query):
     if not query.strip():
         return {"error": "Please enter a job description"}
@@ -65,13 +74,31 @@ def recommend(query):
         return {"error": f"Error processing request: {str(e)}"}
 
 # === Gradio UI ===
-gr.Interface(
+gr_interface = gr.Interface(
     fn=recommend,
     inputs=gr.Textbox(label="Enter Job Description", lines=4),
     outputs="json",
     title="SHL Assessment Recommender",
     description="Paste a job description to get the most relevant SHL assessments."
-).launch()
+)
+
+# === FastAPI Endpoints ===
+@app.get("/health")
+async def health():
+    return JSONResponse(content={"status": "healthy"}, status_code=200)
+
+@app.post("/recommend")
+async def recommend_api(request: Request):
+    try:
+        data = await request.json()
+        query = data.get("query", "").strip()
+        if not query:
+            return JSONResponse(content={"error": "Missing query"}, status_code=400)
+        result = recommend(query)
+        return JSONResponse(content=result, status_code=200)
+    except Exception as e:
+        return JSONResponse(content={"error": str(e)}, status_code=500)
 
+# === Mount Gradio to FastAPI ===
 from gradio.routes import mount_gradio_app
 app = mount_gradio_app(app, gr_interface, path="/")