Update app.py

app.py

@@ -1,82 +1,93 @@
-import os
-import uuid
-import gradio as gr
-
-from backend_app.flows import FlowManager
-from backend_app.ingest import run_ingestion
-from backend_app.rag_hf import RAGEngineHF
-from backend_app.config import DATA_DIR, FAISS_INDEX_PATH, DOCSTORE_PATH
-
-os.makedirs(DATA_DIR, exist_ok=True)
-
-# Build index if missing
-if not (os.path.exists(FAISS_INDEX_PATH) and os.path.exists(DOCSTORE_PATH)):
-    run_ingestion()
-
-rag = RAGEngineHF()
-flows = FlowManager()
-
-def init_state():
-    return {"session_id": "sess-" + uuid.uuid4().hex[:12], "initialized": False}
-
-def render_suggestions(btns):
-    btns = btns[:6]
-    updates = [gr.Button.update(visible=True, value=b) for b in btns]
-    updates += [gr.Button.update(visible=False) for _ in range(6 - len(btns))]
-    return updates
-
-def on_open(chat_history, state):
-    if state["initialized"]:
-        return chat_history, state, *render_suggestions(flows.default_suggestions())
-    welcome = "Welcome to SysLink Food System 👋 How can I help you today?"
-    chat_history = chat_history + [("", welcome)]
-    state["initialized"] = True
-    return chat_history, state, *render_suggestions(flows.default_suggestions())
-
-def handle_user_message(user_text, chat_history, state):
-    session_id = state["session_id"]
-    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"))
-        bot_reply = rag_result["answer"]
-
-        if rag_result.get("sources"):
-            src_lines = "\n".join([f"- {s['title']}" for s in rag_result["sources"][:3]])
-            bot_reply += f"\n\nSources:\n{src_lines}"
-
-        chat_history = chat_history + [(user_text, bot_reply)]
-        return "", chat_history, state, *render_suggestions(flow_result.get("suggestions", []))
-
-    chat_history = chat_history + [(user_text, flow_result["answer"])]
-    return "", chat_history, state, *render_suggestions(flow_result.get("suggestions", []))
-
-def suggestion_click(suggestion_text, chat_history, state):
-    return handle_user_message(suggestion_text, chat_history, state)
-
-with gr.Blocks(title="SysLink Food System Chatbot") as demo:
-    state = gr.State(init_state())
-    gr.Markdown("## SysLink Food System Assistant")
-
-    chat = gr.Chatbot(height=420)
-    open_btn = gr.Button("Open Chat / Start")
-
-    with gr.Row():
-        s1 = gr.Button(visible=False)
-        s2 = gr.Button(visible=False)
-        s3 = gr.Button(visible=False)
-        s4 = gr.Button(visible=False)
-        s5 = gr.Button(visible=False)
-        s6 = gr.Button(visible=False)
-
-    user_in = gr.Textbox(placeholder="Type your message...", label="Message")
-    send_btn = gr.Button("Send")
-
-    open_btn.click(on_open, inputs=[chat, state], outputs=[chat, state, s1, s2, s3, s4, s5, s6])
-    send_btn.click(handle_user_message, inputs=[user_in, chat, state], outputs=[user_in, chat, state, s1, s2, s3, s4, s5, s6])
-    user_in.submit(handle_user_message, inputs=[user_in, chat, state], outputs=[user_in, chat, state, s1, s2, s3, s4, s5, s6])
-
-    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])
-
+import os
+import uuid
+import gradio as gr
+
+from backend_app.flows import FlowManager
+from backend_app.ingest import run_ingestion
+from backend_app.rag_hf import RAGEngineHF
+from backend_app.config import DATA_DIR, FAISS_INDEX_PATH, DOCSTORE_PATH
+
+os.makedirs(DATA_DIR, exist_ok=True)
+
+# Build index if missing
+if not (os.path.exists(FAISS_INDEX_PATH) and os.path.exists(DOCSTORE_PATH)):
+    run_ingestion()
+
+rag = RAGEngineHF()
+flows = FlowManager()
+
+def init_state():
+    return {"session_id": "sess-" + uuid.uuid4().hex[:12], "initialized": False}
+
+def render_suggestions(btns):
+    btns = btns[:6]
+    updates = [gr.Button.update(visible=True, value=b) for b in btns]
+    updates += [gr.Button.update(visible=False) for _ in range(6 - len(btns))]
+    return updates
+
+def on_open(chat_history, state):
+    if state["initialized"]:
+        return chat_history, state, *render_suggestions(flows.default_suggestions())
+    welcome = "Welcome to SysLink Food System 👋 How can I help you today?"
+    chat_history = chat_history + [("", welcome)]
+    state["initialized"] = True
+    return chat_history, state, *render_suggestions(flows.default_suggestions())
+
+def handle_user_message(user_text, chat_history, state):
+    session_id = state["session_id"]
+
+    try:
+        flow_result = flows.handle_message(session_id=session_id, user_message=user_text)
+
+        if flow_result["action"] == "rag":
+            try:
+                rag_result = rag.answer(user_text, preferred_lang=flow_result.get("lang"))
+                bot_reply = rag_result.get("answer", "")
+                if rag_result.get("sources"):
+                    src_lines = "\n".join([f"- {s['title']}" for s in rag_result["sources"][:3]])
+                    bot_reply += f"\n\nSources:\n{src_lines}"
+            except Exception:
+                bot_reply = (
+                    "I ran into a temporary error while generating the answer. "
+                    "Please try again in a moment."
+                )
+
+            chat_history = chat_history + [(user_text, bot_reply)]
+            return "", chat_history, state, *render_suggestions(flow_result.get("suggestions", []))
+
+        chat_history = chat_history + [(user_text, flow_result["answer"])]
+        return "", chat_history, state, *render_suggestions(flow_result.get("suggestions", []))
+
+    except Exception:
+        chat_history = chat_history + [(user_text, "Something went wrong temporarily. Please try again.")]
+        return "", chat_history, state, *render_suggestions(flows.default_suggestions())
+
+def suggestion_click(suggestion_text, chat_history, state):
+    return handle_user_message(suggestion_text, chat_history, state)
+
+with gr.Blocks(title="SysLink Food System Chatbot") as demo:
+    state = gr.State(init_state())
+    gr.Markdown("## SysLink Food System Assistant")
+
+    chat = gr.Chatbot(height=420)
+    open_btn = gr.Button("Open Chat / Start")
+
+    with gr.Row():
+        s1 = gr.Button(visible=False)
+        s2 = gr.Button(visible=False)
+        s3 = gr.Button(visible=False)
+        s4 = gr.Button(visible=False)
+        s5 = gr.Button(visible=False)
+        s6 = gr.Button(visible=False)
+
+    user_in = gr.Textbox(placeholder="Type your message...", label="Message")
+    send_btn = gr.Button("Send")
+
+    open_btn.click(on_open, inputs=[chat, state], outputs=[chat, state, s1, s2, s3, s4, s5, s6])
+    send_btn.click(handle_user_message, inputs=[user_in, chat, state], outputs=[user_in, chat, state, s1, s2, s3, s4, s5, s6])
+    user_in.submit(handle_user_message, inputs=[user_in, chat, state], outputs=[user_in, chat, state, s1, s2, s3, s4, s5, s6])
+
+    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])
+
 demo.launch()