Update app.py

app.py

@@ -7,6 +7,7 @@ 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
 
+# Ensure data directory exists
 os.makedirs(DATA_DIR, exist_ok=True)
 
 # Build index if missing
@@ -16,8 +17,13 @@ if not (os.path.exists(FAISS_INDEX_PATH) and os.path.exists(DOCSTORE_PATH)):
 rag = RAGEngineHF()
 flows = FlowManager()
 
+
 def init_state():
-    return {"session_id": "sess-" + uuid.uuid4().hex[:12], "initialized": False}
+    return {
+        "session_id": "sess-" + uuid.uuid4().hex[:12],
+        "initialized": False
+    }
+
 
 def render_suggestions(btns):
     btns = (btns or [])[:6]
@@ -29,65 +35,98 @@ def render_suggestions(btns):
             updates.append(gr.update(value="", visible=False))
     return updates
 
+
 def on_open(chat_history, state):
     chat_history = chat_history or []
+
     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.append({"role": "assistant", "content": welcome})
     state["initialized"] = True
+
     return chat_history, state, *render_suggestions(flows.default_suggestions())
 
+
 def handle_user_message(user_text, chat_history, state):
     chat_history = chat_history or []
     session_id = state["session_id"]
 
-    # add user message
+    # Add user message
     chat_history.append({"role": "user", "content": user_text})
 
     try:
-        flow_result = flows.handle_message(session_id=session_id, user_message=user_text)
+        flow_result = flows.handle_message(
+            session_id=session_id,
+            user_message=user_text
+        )
 
+        # RAG path
         if flow_result["action"] == "rag":
             try:
-                rag_result = rag.answer(user_text, preferred_lang=flow_result.get("lang"))
+                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]])
+                    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."
+                bot_reply = (
+                    "I ran into a temporary error while generating the answer. "
+                    "Please try again."
+                )
 
             chat_history.append({"role": "assistant", "content": bot_reply})
-            return "", chat_history, state, *render_suggestions(flow_result.get("suggestions", []))
 
-        
-        # flow reply
-        chat_history.append({"role": "assistant", "content": flow_result["answer"]})
-        return "", chat_history, state, *render_suggestions(flow_result.get("suggestions", []))
+            return (
+                "",
+                chat_history,
+                state,
+                *render_suggestions(flow_result.get("suggestions", []))
+            )
+
+        # Flow-only response
+        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())
+        chat_history.append({
+            "role": "assistant",
+            "content": "Something went wrong temporarily. Please try again."
+        })
 
-            chat_history = chat_history + [(user_text, bot_reply)]
-            return "", chat_history, state, *render_suggestions(flow_result.get("suggestions", []))
+        return (
+            "",
+            chat_history,
+            state,
+            *render_suggestions(flows.default_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, type="messages")
@@ -100,14 +139,37 @@ with gr.Blocks(title="SysLink Food System Chatbot") as demo:
         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")
+    user_in = gr.Textbox(
+        placeholder="Type your message...",
+        label="Message"
+    )
 
-    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])
+    send_btn = gr.Button("Send")
+    open_btn = gr.Button("Open Chat")
+
+    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])
+        b.click(
+            suggestion_click,
+            inputs=[b, chat, state],
+            outputs=[user_in, chat, state, s1, s2, s3, s4, s5, s6]
+        )
 
 demo.launch()