Update app.py

app.py

@@ -234,66 +234,84 @@ def continue_chat(chat_history, user_message, analysis_text):
 # Gradio UI
 # -----------------------
 def build_demo():
-    with gr.Blocks(title="AURA Chat — Hedge Fund Picks", css="""
-        .gradio-container { max-width: 1100px; margin: 18px auto; }
-        .header {text-align: left; margin-bottom: 6px;}
-        .muted {color: #7d8590; font-size: 14px;}
-        .analysis-box { background: #ffffff; border-radius: 8px; padding: 12px; box-shadow: 0 4px 14px rgba(0,0,0,0.06);}
-    """) as demo:
+    with gr.Blocks(title="AURA Chat — Hedge Fund Picks") as demo:
+
+        # Inject custom CSS in a totally safe, backward-compatible way
+        gr.HTML("""
+        <style>
+            .gradio-container { max-width: 1100px; margin: 18px auto; }
+            .header { text-align: left; margin-bottom: 6px; }
+            .muted { color: #7d8590; font-size: 14px; }
+            .analysis-box {
+                background: #ffffff;
+                border-radius: 8px;
+                padding: 12px;
+                box-shadow: 0 4px 14px rgba(0,0,0,0.06);
+            }
+        </style>
+        """)
+
         gr.Markdown("# AURA Chat — Hedge Fund Picks")
         gr.Markdown("**Enter one or more data prompts (one per line)** — e.g. `SEC insider transactions october 2025 company XYZ`.\n\nOnly input prompts; model, tokens and timing are fixed. Press **Analyze** to fetch & generate the picks. After analysis you can chat with the assistant about the results.")
 
         with gr.Row():
             with gr.Column(scale=1):
-                prompts = gr.Textbox(lines=6, label="Data Prompts (one per line)", placeholder="SEC insider transactions october 2025\n13F filings Q3 2025\ncompany: ACME corp insider buys")
+                prompts = gr.Textbox(
+                    lines=6,
+                    label="Data Prompts (one per line)",
+                    placeholder="SEC insider transactions october 2025\n13F filings Q3 2025\ncompany: ACME corp insider buys"
+                )
                 analyze_btn = gr.Button("Analyze", variant="primary")
                 error_box = gr.Markdown("", visible=False)
                 gr.Markdown(f"**Fixed settings:** Model = `{LLM_MODEL}`  •  Max tokens = `{MAX_TOKENS}`  •  Scrape delay = `{SCRAPE_DELAY}s`")
                 gr.Markdown("**Important:** Add your `OPENAI_API_KEY` to Space Secrets before running.")
             with gr.Column(scale=1):
-                analysis_out = gr.Textbox(label="Generated Analysis (Top picks with Investment Duration)", lines=18, interactive=False)
+                analysis_out = gr.Textbox(
+                    label="Generated Analysis (Top picks with Investment Duration)",
+                    lines=18,
+                    interactive=False
+                )
                 gr.Markdown("**Chat with AURA about this analysis**")
                 chatbot = gr.Chatbot(label="AURA Chat", height=420)
-                user_input = gr.Textbox(placeholder="Ask a follow-up question about the analysis...", label="Your question")
+                user_input = gr.Textbox(
+                    placeholder="Ask a follow-up question about the analysis...",
+                    label="Your question"
+                )
                 send_btn = gr.Button("Send")
 
-        # State to hold the analysis text and chat history
-        analysis_state = gr.State("")       # holds the analysis text (string)
-        chat_state = gr.State([])           # holds list of (user, assistant) tuples
-
-        # Hook: Analyze button
-        def on_analyze(prompts_text):
-            analysis_text, initial_chat = analyze_and_seed_chat(prompts_text)
-            if analysis_text.startswith("ERROR"):
-                return gr.update(value=analysis_text), gr.update(visible=True, value=f"**Error:** {analysis_text}"), "", []
-            # Set outputs: analysis_out, clear chatbot and seed it
-            # analysis_text displayed in the big box; the chat seeded with the pair
-            return analysis_text, gr.update(visible=False, value=""), analysis_text, initial_chat
-
-        analyze_btn.click(fn=on_analyze, inputs=[prompts], outputs=[analysis_out, error_box, analysis_state, chat_state])
-
-        # Hook: Send follow-up chat message
-        def on_send(chat_state_list, user_msg, analysis_text):
-            if not user_msg.strip():
-                return chat_state_list, ""
-            # chat_state_list is list of tuples
-            updated_history = continue_chat(chat_state_list or [], user_msg, analysis_text)
-            # Clear user input after sending
-            return updated_history, ""
-
-        send_btn.click(fn=on_send, inputs=[chat_state, user_input, analysis_state], outputs=[chat_state, user_input])
-        # Also allow pressing Enter in the user_input box
-        user_input.submit(fn=on_send, inputs=[chat_state, user_input, analysis_state], outputs=[chat_state, user_input])
-
-        # Render chat_state into the Chatbot UI whenever it updates
-        def render_chat(chat_state_list):
-            # gr.Chatbot expects a list of (user, assistant) pairs
-            return chat_state_list or []
-
-        chat_state.change(fn=render_chat, inputs=[chat_state], outputs=[chatbot])
+        # States
+        analysis_state = gr.State("")  
+        chat_state = gr.State([])      
+
+        # Analyze click
+        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]
+        )
+        user_input.submit(
+            fn=on_send,
+            inputs=[chat_state, user_input, analysis_state],
+            outputs=[chat_state, user_input]
+        )
+
+        # Render chat
+        chat_state.change(
+            fn=render_chat,
+            inputs=[chat_state],
+            outputs=[chatbot]
+        )
 
     return demo
 
+
 # -----------------------
 # Clean shutdown helper
 # -----------------------