Update app.py

But the Gradio interface doesn't expose a /search_medqa endpoint. We need to add FastAPI back for the API.

✅ Web UI at: https://imurra-medqa-api-online.hf.space/
✅ API endpoint at: https://imurra-medqa-api-online.hf.space/search_medqa

app.py

@@ -3,6 +3,8 @@ import zipfile
 import chromadb
 from sentence_transformers import SentenceTransformer
 import gradio as gr
+from fastapi import FastAPI
+from pydantic import BaseModel
 
 # Extract database
 DB_PATH = "./medqa_db"
@@ -21,14 +23,19 @@ print("Loading model...")
 model = SentenceTransformer('ncbi/MedCPT-Query-Encoder')
 print("Ready!")
 
-def search_medqa(query, num_results=3):
+# Search function
+def search_function(query, num_results=3):
+    embedding = model.encode(query).tolist()
+    results = collection.query(query_embeddings=[embedding], n_results=int(num_results))
+    return results
+
+# Gradio interface
+def search_gradio(query, num_results=3):
     if not query.strip():
         return "Please enter a search query."
     
     try:
-        embedding = model.encode(query).tolist()
-        results = collection.query(query_embeddings=[embedding], n_results=int(num_results))
-        
+        results = search_function(query, num_results)
         output = ""
         for i in range(len(results['documents'][0])):
             output += f"\n{'='*60}\nExample {i+1}\n{'='*60}\n"
@@ -39,9 +46,8 @@ def search_medqa(query, num_results=3):
     except Exception as e:
         return f"Error: {str(e)}"
 
-# Create Gradio interface
 demo = gr.Interface(
-    fn=search_medqa,
+    fn=search_gradio,
     inputs=[
         gr.Textbox(label="Medical Query", placeholder="e.g., hyponatremia", lines=2),
         gr.Slider(1, 5, value=3, step=1, label="Number of Results")
@@ -52,5 +58,28 @@ demo = gr.Interface(
     examples=[["hyponatremia", 3], ["myocardial infarction", 2]]
 )
 
-# Launch - this is the key line!
-demo.launch(server_name="0.0.0.0", server_port=7860)
+# FastAPI for ChatGPT
+app = FastAPI()
+
+class SearchRequest(BaseModel):
+    query: str
+    num_results: int = 3
+
+class SearchResponse(BaseModel):
+    results: list[dict]
+
+@app.post("/search_medqa")
+async def api_search(request: SearchRequest):
+    results = search_function(request.query, request.num_results)
+    formatted = []
+    for i in range(len(results['documents'][0])):
+        formatted.append({
+            "example_number": i + 1,
+            "question": results['documents'][0][i],
+            "answer": results['metadatas'][0][i].get('answer', 'N/A'),
+            "distance": results['distances'][0][i]
+        })
+    return SearchResponse(results=formatted)
+
+# Mount Gradio on FastAPI
+app = gr.mount_gradio_app(app, demo, path="/")