Update app.py

app.py

@@ -3,9 +3,7 @@ import uuid
 from typing import Any, Dict, List, Optional
 
 import gradio as gr
-from fastapi import FastAPI
-from fastapi.middleware.cors import CORSMiddleware
-from pydantic import BaseModel
+from pydantic import BaseModel  # ok to keep even if unused
 
 from backend_app.flows import FlowManager
 from backend_app.ingest import run_ingestion
@@ -23,49 +21,7 @@ if not (os.path.exists(FAISS_INDEX_PATH) and os.path.exists(DOCSTORE_PATH)):
 rag = RAGEngineHF()
 flows = FlowManager()
 
-# --------- FastAPI (for your website widget) ----------
-api = FastAPI(title="SysLink Chat API")
-
-api.add_middleware(
-    CORSMiddleware,
-    allow_origins=["*"],  # keep open for testing; lock later
-    allow_methods=["*"],
-    allow_headers=["*"],
-)
-
-class ChatIn(BaseModel):
-    session_id: str
-    message: str
-
-@api.get("/init")
-def init() -> Dict[str, Any]:
-    return {
-        "bot_name": BOT_NAME,
-        "logo_url": BOT_LOGO_URL,
-        "welcome": BOT_WELCOME,
-        "suggestions": flows.default_suggestions(),
-    }
-
-@api.post("/chat")
-def chat(payload: ChatIn) -> Dict[str, Any]:
-    session_id = payload.session_id
-    user_text = (payload.message or "").strip()
-
-    flow_result = flows.handle_message(session_id=session_id, user_message=user_text)
-
-    if flow_result["action"] == "rag":
-        rag_result = rag.answer(user_text, preferred_lang=flow_result.get("lang"))
-        return {
-            "answer": rag_result.get("answer", ""),
-            "suggestions": flow_result.get("suggestions", []),
-        }
-
-    return {
-        "answer": flow_result.get("answer", ""),
-        "suggestions": flow_result.get("suggestions", []),
-    }
-
-# --------- Gradio UI (optional, also hosted at /) ----------
+# --------- Gradio UI ----------
 def init_state():
     return {"session_id": "sess-" + uuid.uuid4().hex[:12], "initialized": False}
 
@@ -81,7 +37,7 @@ def render_suggestions(btns: Optional[List[str]]):
 
 def on_open(chat_history, state):
     chat_history = chat_history or []
-    if not state["initialized"]:
+    if not state.get("initialized", False):
         chat_history.append({"role": "assistant", "content": BOT_WELCOME})
         state["initialized"] = True
     return chat_history, state, *render_suggestions(flows.default_suggestions())
@@ -90,18 +46,30 @@ def handle_user_message(user_text, chat_history, state):
     chat_history = chat_history or []
     session_id = state["session_id"]
 
+    user_text = (user_text or "").strip()
+    if not user_text:
+        return "", chat_history, state, *render_suggestions([])
+
+    # add user message
     chat_history.append({"role": "user", "content": user_text})
 
-    flow_result = flows.handle_message(session_id=session_id, user_message=user_text)
+    try:
+        flow_result = flows.handle_message(session_id=session_id, user_message=user_text)
+
+        # RAG path
+        if flow_result["action"] == "rag":
+            rag_result = rag.answer(user_text, preferred_lang=flow_result.get("lang"))
+            bot_reply = rag_result.get("answer", "")
+            chat_history.append({"role": "assistant", "content": bot_reply})
+            return "", chat_history, state, *render_suggestions(flow_result.get("suggestions", []))
 
-    if flow_result["action"] == "rag":
-        rag_result = rag.answer(user_text, preferred_lang=flow_result.get("lang"))
-        bot_reply = rag_result.get("answer", "")
-        chat_history.append({"role": "assistant", "content": bot_reply})
+        # Flow-only reply
+        chat_history.append({"role": "assistant", "content": flow_result.get("answer", "")})
         return "", chat_history, state, *render_suggestions(flow_result.get("suggestions", []))
 
-    chat_history.append({"role": "assistant", "content": flow_result["answer"]})
-    return "", chat_history, state, *render_suggestions(flow_result.get("suggestions", []))
+    except Exception:
+        chat_history.append({"role": "assistant", "content": "Something went wrong temporarily. Please try again."})
+        return "", chat_history, state, *render_suggestions(flows.default_suggestions())
 
 def suggestion_click(text, chat_history, state):
     return handle_user_message(text, chat_history, state)
@@ -132,4 +100,5 @@ with gr.Blocks(title="SysLink Food System Chatbot") as demo:
     for b in [s1, s2, s3, s4, s5, s6]:
         b.click(suggestion_click, inputs=[b, chat, state], outputs=[user_in, chat, state, s1, s2, s3, s4, s5, s6])
 
-app = demo
+# ✅ Required for Hugging Face Gradio runtime stability
+demo.queue()