Update app.py

app.py

@@ -13,6 +13,7 @@ from fastapi.middleware.cors import CORSMiddleware
 from pydantic import BaseModel
 import uvicorn
 import json
+import gradio as gr
 
 # Disable SSL warnings
 urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)
@@ -200,8 +201,30 @@ async def search_endpoint(request: SearchRequest):
         
         raise HTTPException(status_code=500, detail=error_response)
 
+def gradio_search(question, mode):
+    import requests
+    try:
+        resp = requests.post(
+            "/search",  # Internal call on Spaces
+            json={"question": question, "mode": mode},
+            timeout=60
+        )
+        data = resp.json()
+        answer = data.get("answer", "")
+        sources = data.get("sources", [])
+        sources_md = "\n".join([f"- [{src['title']}]({src['url']})" for src in sources])
+        return answer, sources_md
+    except Exception as e:
+        return f"Error: {e}", ""
+
+with gr.Blocks() as demo:
+    gr.Markdown("# Search Assistant")
+    question = gr.Textbox(label="Question", placeholder="Ask anything...")
+    mode = gr.Radio(choices=["quick", "deep"], value="quick", label="Mode")
+    answer = gr.Markdown(label="Answer")
+    sources = gr.Markdown(label="Sources")
+    btn = gr.Button("Search")
+    btn.click(gradio_search, inputs=[question, mode], outputs=[answer, sources])
+
 if __name__ == "__main__":
-    port = int(os.getenv("PORT", 5000))
-    print("🚀 Starting Search Assistant Server...")
-    print(f"📱 Port: {port}")
-    uvicorn.run(app, host="0.0.0.0", port=port)
+    demo.launch(server_name="0.0.0.0", server_port=int(os.getenv("PORT", 7860)))