1232

app.py

@@ -254,32 +254,36 @@ def chat_with_session(message, history, session_id=None):
     if session_id is None:
         session_id = str(uuid.uuid4())
     answer = get_context_and_answer(message, history, session_id)
-    return answer, session_id  # Return answer and session_id for state tracking
 
-if __name__ == "__main__":
-    # Gradio Blocks layout for combining ChatInterface + Clear Button
-    with gr.Blocks() as demo:
-        session_id_state = gr.State(str(uuid.uuid4()))
-
-        chat_interface = gr.ChatInterface(
-            fn=chat_with_session,
-            additional_inputs=[session_id_state],
-            title="ASKXENO",
-            description="Ask anything about XENO's financial services.",
-            theme="soft",
-            examples=[
-                ["How do I open a XENO account?", None],
-                ["What are the fees?", None],
-                ["Tell me about investment options", None]
-            ]
-        )
+    if history is None:
+        history = []
+    # Append user message and assistant answer to chat history as dicts:
+    history.append({"role": "user", "content": message})
+    history.append({"role": "assistant", "content": answer})
 
-        clear_button = gr.Button("Clear Session")
+    return history, session_id
 
-        # When clear button is clicked, reset session_id state (clears conversation context)
-        clear_button.click(
-            fn=lambda: str(uuid.uuid4()),
-            outputs=[session_id_state]
-        )
-
-    demo.launch(share=False, server_name="0.0.0.0", server_port=7860)
+if __name__ == "__main__":
+    session_id_state = gr.State(value=str(uuid.uuid4()))
+    
+    iface = gr.ChatInterface(
+        fn=chat_with_session,
+        additional_inputs=[session_id_state],
+        title="ASKXENO",
+        description="Ask anything about XENO's financial services.",
+        theme="soft",
+        type="messages",  # Use 'messages' format (dict with role/content)
+        examples=[
+            ["How do I open a XENO account?", None],
+            ["What are the fees?", None],
+            ["Tell me about investment options", None]
+        ]
+    )
+    
+    # Reset session ID on clear button click:
+    iface.clear_btn.click(
+        fn=lambda: str(uuid.uuid4()),
+        outputs=[session_id_state]
+    )
+    
+    iface.launch(share=False, server_name="0.0.0.0", server_port=7860)