Update app.py

app.py

@@ -77,7 +77,7 @@ def safe_exec(code_string: str, local_vars: dict):
 def load_and_process_file(file_obj, state_dict):
     """Loads a CSV, processes it, and updates the entire UI state."""
     if file_obj is None:
-        return state_dict, "Please upload a file.", *[gr.update(visible=False)] * 3
+        return state_dict, "Please upload a file.", *[gr.update(visible=False)] * 4
     try:
         df = pd.read_csv(file_obj.name, low_memory=False)
         for col in df.select_dtypes(include=['object']).columns:
@@ -111,7 +111,7 @@ def load_and_process_file(file_obj, state_dict):
                 gr.update(value=f"{len(metadata['datetime_cols'])}"),
                 gr.update(choices=metadata['columns']), gr.update(choices=metadata['columns']), gr.update(choices=metadata['columns']))
     except Exception as e:
-        return state_dict, f"❌ **Error:** {e}", *[gr.update()] * 10
+        return state_dict, f"❌ **Error:** {e}", *[gr.update()] * 11
 
 def extract_dataset_metadata(df: pd.DataFrame):
     rows, cols = df.shape
@@ -160,11 +160,23 @@ def get_ai_suggestions(state_dict, api_key):
         suggestions = json.loads(response.text)
         
         # Create a button for each suggestion
-        buttons = [gr.Button(s, variant="secondary") for s in suggestions]
-        return gr.update(visible=False), gr.Accordion(label="✨ AI Smart Suggestions", open=True, children=buttons)
+        buttons = [gr.Button(s, variant="secondary", visible=True) for s in suggestions]
+        # Pad with hidden buttons to always have 5 outputs
+        buttons += [gr.Button(visible=False)] * (5 - len(buttons))
+
+        return gr.update(visible=False), *buttons
     
     except Exception as e:
-        return f"Could not generate suggestions: {e}", gr.update(visible=False)
+        return f"Could not generate suggestions: {e}", *[gr.update(visible=False)]*5
+
+def handle_suggestion_click(question_text):
+    """When a suggestion button is clicked, switch to the co-pilot page and populate the input."""
+    return (
+        gr.update(visible=False), # Hide cockpit
+        gr.update(visible=False), # Hide deep dive
+        gr.update(visible=True),  # Show co-pilot
+        question_text # Populate the chat input
+    )
 
 # --- Page 2: Deep Dive Dashboard ---
 def add_plot_to_dashboard(state_dict, x_col, y_col, plot_type):
@@ -261,7 +273,7 @@ def respond_to_chat(user_message, history, state_dict, api_key):
         if df_result is not None: output_updates[3] = gr.update(visible=True, value=df_result)
         if stdout:
             # Append stdout to explanation if it exists
-            new_explanation = (output_updates[0]['value'] if output_updates[0]['value'] else "") + f"\n\n**Console Output:**\n```\n{stdout}\n```"
+            new_explanation = (output_updates[0]['value'] if output_updates[0]['visible'] else "") + f"\n\n**Console Output:**\n```\n{stdout}\n```"
             output_updates[0] = gr.update(visible=True, value=new_explanation)
         if error:
             error_explanation = f"**Phoenix Co-pilot:** I encountered an error. Here's the details:\n\n`{error}`"
@@ -304,7 +316,8 @@ def create_gradio_interface():
                 with gr.Column(visible=True) as welcome_page:
                     gr.Markdown("# Welcome to the AI Data Explorer (Phoenix UI)", elem_id="welcome-header")
                     gr.Markdown("Please **upload a CSV file** and **enter your Gemini API key** in the sidebar to begin.")
-                    gr.Image(value="https://i.imgur.com/gY5wSjV.png", label="Workflow", show_label=False, show_download_button=False, container=False) # Placeholder image
+                    # CORRECTED: Uses a local file, 'workflow.png', which must be in the same directory.
+                    gr.Image(value="workflow.png", label="Workflow", show_label=False, show_download_button=False, container=False)
 
                 # Page 1: Data Cockpit (Hidden initially)
                 with gr.Column(visible=False) as cockpit_page:
@@ -324,7 +337,8 @@ def create_gradio_interface():
                             time_cols_stat = gr.Textbox("0", show_label=False, elem_classes="stat-card-value")
 
                     suggestion_status = gr.Markdown(visible=True)
-                    suggestion_accordion = gr.Accordion(label="✨ AI Smart Suggestions", open=False, visible=False)
+                    with gr.Accordion(label="✨ AI Smart Suggestions", open=True):
+                         suggestion_buttons = [gr.Button(visible=False) for _ in range(5)]
 
                 # Page 2: Deep Dive Dashboard Builder (Hidden initially)
                 with gr.Column(visible=False) as deep_dive_page:
@@ -343,9 +357,8 @@ def create_gradio_interface():
                 with gr.Column(visible=False) as copilot_page:
                     gr.Markdown("## 🤖 AI Co-pilot")
                     gr.Markdown("Ask complex questions in natural language. The Co-pilot will write and execute code to find the answer.")
-                    chatbot = gr.Chatbot(height=400, label="Conversation with Co-pilot")
+                    chatbot = gr.Chatbot(height=400, label="Conversation with Co-pilot", show_copy_button=True)
                     
-                    # AI's multi-modal response area
                     with gr.Accordion("Co-pilot's Response Details", open=True):
                         copilot_explanation = gr.Markdown(visible=False, elem_classes="explanation-block")
                         copilot_code = gr.Code(language="python", visible=False, label="Executed Python Code", elem_classes="code-block")
@@ -357,55 +370,49 @@ def create_gradio_interface():
                         chat_submit_btn = gr.Button("Submit", variant="primary")
 
         # --- Event Handlers ---
-        # Page Navigation
         pages = [cockpit_page, deep_dive_page, copilot_page]
         nav_buttons = [cockpit_btn, deep_dive_btn, copilot_btn]
         
         for i, btn in enumerate(nav_buttons):
-            btn.click(lambda i=i: (gr.update(visible=i==0), gr.update(visible=i==1), gr.update(visible=i==2)),
-                      outputs=pages).then(
-                      lambda: [gr.update(elem_classes="selected" if j==i else "") for j in range(len(nav_buttons))],
-                      outputs=nav_buttons)
+            btn.click(
+                lambda i=i: (gr.update(visible=i==0), gr.update(visible=i==1), gr.update(visible=i==2)),
+                outputs=pages
+            ).then(
+                lambda i=i: [gr.update(elem_classes="selected" if j==i else "") for j in range(len(nav_buttons))],
+                outputs=nav_buttons
+            )
 
-        # File Upload
         file_input.upload(
             fn=load_and_process_file,
             inputs=[file_input, global_state],
             outputs=[global_state, status_output, welcome_page, cockpit_page, deep_dive_page, copilot_page,
                      rows_stat, cols_stat, quality_stat, time_cols_stat,
-                     x_col_dd, y_col_dd, copilot_input_col_ref_for_dynamic_update] # This last one is a dummy to trigger updates
+                     x_col_dd, y_col_dd, plot_type_dd]
+        ).then(
+            lambda: (gr.update(visible=True), gr.update(visible=False), gr.update(visible=False)), outputs=pages
         ).then(
-            fn=lambda: (gr.update(elem_classes="selected"), gr.update(elem_classes=""), gr.update(elem_classes="")),
-            outputs=nav_buttons
+            lambda: (gr.update(elem_classes="selected"), gr.update(elem_classes=""), gr.update(elem_classes="")), outputs=nav_buttons
         )
         
-        # Suggestions Button
         suggestion_btn.click(
             get_ai_suggestions,
             [global_state, api_key_input],
-            [suggestion_status, suggestion_accordion]
-        ).then(
-            fn=lambda: [gr.Button.update(visible=True) for _ in range(5)], # Assumes max 5 suggestions for demo
-            outputs=[b for b in suggestion_accordion.children] if isinstance(suggestion_accordion, gr.Accordion) and suggestion_accordion.children else []
+            [suggestion_status, *suggestion_buttons]
         )
-        # Handle suggestion button clicks to populate chat
-        if isinstance(suggestion_accordion, gr.Accordion):
-            for button in suggestion_accordion.children:
-                button.click(
-                    fn=lambda q=button.value: (gr.update(visible=False), gr.update(visible=False), gr.update(visible=True), q),
-                    outputs=[cockpit_page, deep_dive_page, copilot_page, chat_input]
-                ).then(
-                    fn=lambda: (gr.update(elem_classes=""), gr.update(elem_classes=""), gr.update(elem_classes="selected")),
-                    outputs=nav_buttons
-                )
-
-
-        # Dashboard Builder
+        
+        for btn in suggestion_buttons:
+            btn.click(
+                handle_suggestion_click,
+                inputs=[btn],
+                outputs=[cockpit_page, deep_dive_page, copilot_page, chat_input]
+            ).then(
+                lambda: (gr.update(elem_classes=""), gr.update(elem_classes=""), gr.update(elem_classes="selected")),
+                outputs=nav_buttons
+            )
+
         add_plot_btn.click(add_plot_to_dashboard, [global_state, x_col_dd, y_col_dd, plot_type_dd], [global_state, dashboard_accordion])
         clear_plots_btn.click(clear_dashboard, [global_state], [global_state, dashboard_accordion])
 
-        # AI Co-pilot Chat
-        copilot_input_col_ref_for_dynamic_update = x_col_dd # Dummy placeholder for dynamic updates
         chat_submit_btn.click(
             respond_to_chat, 
             [chat_input, chatbot, global_state, api_key_input],