Update app.py

app.py

@@ -30,16 +30,22 @@ def render_suggestions(btns):
     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 = chat_history + [("", welcome)]
+    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
+    chat_history.append({"role": "user", "content": user_text})
+
     try:
         flow_result = flows.handle_message(session_id=session_id, user_message=user_text)
 
@@ -47,14 +53,25 @@ def handle_user_message(user_text, chat_history, state):
             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."
-                )
+                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", []))
+
+    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 = chat_history + [(user_text, bot_reply)]
             return "", chat_history, state, *render_suggestions(flow_result.get("suggestions", []))
@@ -73,8 +90,7 @@ 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")
+    chat = gr.Chatbot(height=420, type="messages")
 
     with gr.Row():
         s1 = gr.Button(visible=False)