Update app.py

app.py

@@ -236,7 +236,7 @@ def continue_chat(chat_history, user_message, analysis_text):
 def build_demo():
     with gr.Blocks(title="AURA Chat — Hedge Fund Picks") as demo:
 
-        # Inject custom CSS in a totally safe, backward-compatible way
+        # Inject custom CSS safely
         gr.HTML("""
         <style>
             .gradio-container { max-width: 1100px; margin: 18px auto; }
@@ -280,29 +280,57 @@ def build_demo():
                 send_btn = gr.Button("Send")
 
         # States
-        analysis_state = gr.State("")  
-        chat_state = gr.State([])      
-
-        # Analyze click
+        analysis_state = gr.State("")   # holds the analysis text
+        chat_state = gr.State([])       # holds list of (user, assistant) tuples
+
+        # ---- Handler functions (must be defined before wiring) ----
+        def on_analyze(prompts_text):
+            """
+            Called when user clicks Analyze.
+            Returns: analysis_out (string), error_box (gr.Markdown value), analysis_state (string), chat_state (list)
+            """
+            analysis_text, initial_chat = analyze_and_seed_chat(prompts_text)
+            if analysis_text.startswith("ERROR"):
+                # Show error in the error_box; leave other outputs unchanged
+                return "", f"**Error:** {analysis_text}", "", []
+            # Success: fill analysis_out, hide error box (empty string), set analysis_state and seed chat_state
+            return analysis_text, "", analysis_text, initial_chat
+
+        def on_send(chat_state_list, user_msg, analysis_text):
+            """
+            Called when user sends a chat message.
+            Returns: updated chat_state, cleared user_input
+            """
+            if not user_msg or not user_msg.strip():
+                return chat_state_list or [], ""
+            updated_history = continue_chat(chat_state_list or [], user_msg, analysis_text)
+            return updated_history, ""  # second value clears the user_input box
+
+        def render_chat(chat_state_list):
+            """Render the chat_state into the Chatbot component."""
+            return chat_state_list or []
+
+        # ---- Wire handlers to UI components ----
         analyze_btn.click(
             fn=on_analyze,
             inputs=[prompts],
             outputs=[analysis_out, error_box, analysis_state, chat_state]
         )
 
-        # Chat send
         send_btn.click(
             fn=on_send,
             inputs=[chat_state, user_input, analysis_state],
             outputs=[chat_state, user_input]
         )
+
+        # Allow Enter-key submission from the textbox
         user_input.submit(
             fn=on_send,
             inputs=[chat_state, user_input, analysis_state],
             outputs=[chat_state, user_input]
         )
 
-        # Render chat
+        # Keep Chatbot display updated whenever chat_state changes
         chat_state.change(
             fn=render_chat,
             inputs=[chat_state],
@@ -312,6 +340,7 @@ def build_demo():
     return demo
 
 
+
 # -----------------------
 # Clean shutdown helper
 # -----------------------