Update app.py

app.py

@@ -1,23 +1,16 @@
 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)
@@ -32,7 +25,6 @@ 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):
@@ -48,7 +40,6 @@ 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"}
@@ -74,31 +65,10 @@ 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."
-)
-
-# === 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="/")
+).launch()