Update app.py

app.py

@@ -1,29 +1,34 @@
+from fastapi import FastAPI, Request
 import gradio as gr
 from chatbot_functions import DataProcessor, Chatbot
 
-# Initialize data and chatbot
-data_processor = DataProcessor()  # loads cleaned_medquad.csv
+# Initialize
+data_processor = DataProcessor()
 chatbot = Chatbot(data_processor)
 
-def chat_with_backend(message, history=None):
-    if not message.strip():
-        return "Please enter a question."
-    
-    try:
-        best_question, response, source = chatbot.get_response(message)
-        if best_question == "UNKNOWN":
-            return response
-        else:
-            return f"Matched question: {best_question}\nAnswer: {response}\nSource: {source}"
-    except Exception as e:
-        return f"Error: {str(e)}"
+app = FastAPI()
 
-# Gradio interface
+@app.post("/api/chat")
+async def chat(request: Request):
+    data = await request.json()
+    message = data.get("message", "").strip()
+
+    if not message:
+        return {"reply": "Please enter a valid question."}
+
+    best_question, response, source = chatbot.get_response(message)
+    if best_question == "UNKNOWN":
+        return {"reply": response}
+    else:
+        return {
+            "reply": f"Matched question: {best_question}\nAnswer: {response}\nSource: {source}"
+        }
+
+# Gradio UI (optional for Hugging Face Space)
 iface = gr.ChatInterface(
-    fn=chat_with_backend,
-    title="Care Companion Chatbot",
-    description="Ask health-related questions and get reliable answers from Care Companion."
+    fn=lambda msg, history=None: chatbot.get_response(msg)[1],
+    title="Care Companion Chatbot"
 )
 
-if __name__ == "__main__":
-    iface.launch(server_name="0.0.0.0", server_port=7860, share=True, debug=True)
+import threading
+threading.Thread(target=lambda: iface.launch(server_name="0.0.0.0", server_port=7860, share=False)).start()