Update app.py

app.py

@@ -14,8 +14,7 @@ from contextlib import redirect_stdout
 # --- Configuration ---
 warnings.filterwarnings('ignore')
 
-# --- NEW: Expert-Crafted Dark Theme CSS ---
-# This CSS is designed to work with a dark theme, ensuring all elements are visible and stylish.
+# --- Expert-Crafted Dark Theme CSS ---
 CSS = """
 /* --- Phoenix UI Custom Dark CSS --- */
 /* Stat Card Styling */
@@ -104,13 +103,11 @@ def load_and_process_file(file_obj, state_dict):
         
         status_msg = f"✅ **{state_dict['filename']}** loaded successfully."
         
-        # Update UI elements with new data context
         cockpit_update = gr.update(visible=True)
         deep_dive_update = gr.update(visible=False)
         copilot_update = gr.update(visible=False)
         welcome_update = gr.update(visible=False)
         
-        # Stat cards
         rows, cols = metadata['shape']
         quality = metadata['data_quality']
         
@@ -136,7 +133,6 @@ def extract_dataset_metadata(df: pd.DataFrame):
 
 # --- Page Navigation ---
 def switch_page(page_name):
-    """Controls visibility of main content pages."""
     if page_name == "cockpit":
         return gr.update(visible=True), gr.update(visible=False), gr.update(visible=False)
     elif page_name == "deep_dive":
@@ -147,7 +143,6 @@ def switch_page(page_name):
 
 # --- Page 1: Data Cockpit ---
 def get_ai_suggestions(state_dict, api_key):
-    """Generates proactive analytical suggestions from the AI."""
     if not api_key: return "Enter your Gemini API key to get suggestions.", *[gr.update(visible=False)]*5
     if not state_dict: return "Upload data first.", *[gr.update(visible=False)]*5
     
@@ -176,17 +171,13 @@ def get_ai_suggestions(state_dict, api_key):
         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
+        gr.update(visible=False), gr.update(visible=False),
+        gr.update(visible=True),  question_text
     )
 
 # --- Page 2: Deep Dive Dashboard ---
 def add_plot_to_dashboard(state_dict, x_col, y_col, plot_type):
-    """Generates a plot and adds it to the state-managed dashboard."""
     if not x_col: return state_dict, gr.update()
     
     df = state_dict['df']
@@ -213,13 +204,11 @@ def add_plot_to_dashboard(state_dict, x_col, y_col, plot_type):
         return state_dict, gr.update()
 
 def clear_dashboard(state_dict):
-    """Clears all plots from the dashboard."""
     state_dict['dashboard_plots'] = []
     return state_dict, gr.Accordion(label="Your Dashboard Plots", children=[])
 
 # --- Page 3: AI Co-pilot ---
 def respond_to_chat(user_message, history, state_dict, api_key):
-    """Handles the advanced chat interaction with the AI Co-pilot."""
     if not api_key:
         history.append((user_message, "I need a Gemini API key to function. Please provide it in the sidebar."))
         return history, *[gr.update(visible=False)] * 4
@@ -292,20 +281,19 @@ def respond_to_chat(user_message, history, state_dict, api_key):
 
 # --- Gradio UI Definition ---
 def create_gradio_interface():
-    # --- CORRECTED: Using a standard dark theme + custom dark CSS ---
-    with gr.Blocks(theme=gr.themes.Default(primary_hue="indigo", secondary_hue="blue").dark(), css=CSS, title="Phoenix AI Data Explorer") as demo:
+    # --- CORRECTED THEME DEFINITION ---
+    # We use gr.themes.Glass which is dark by default and allows hue customization.
+    # This is the syntactically correct and visually appropriate way to create the theme.
+    with gr.Blocks(theme=gr.themes.Glass(primary_hue="indigo", secondary_hue="blue"), css=CSS, title="Phoenix AI Data Explorer") as demo:
         global_state = gr.State({})
 
         with gr.Row():
-            # --- Sidebar ---
+            # Sidebar
             with gr.Column(scale=1, elem_classes="sidebar"):
                 gr.Markdown("## 🚀 Phoenix UI")
-                
-                # Navigation buttons
                 cockpit_btn = gr.Button("📊 Data Cockpit", elem_classes="selected")
                 deep_dive_btn = gr.Button("🔍 Deep Dive Builder")
                 copilot_btn = gr.Button("🤖 AI Co-pilot")
-                
                 gr.Markdown("---")
                 file_input = gr.File(label="📁 Upload New CSV", file_types=[".csv"])
                 status_output = gr.Markdown("Status: Awaiting data...")
@@ -313,16 +301,13 @@ def create_gradio_interface():
                 api_key_input = gr.Textbox(label="🔑 Gemini API Key", type="password", placeholder="Enter key here...")
                 suggestion_btn = gr.Button("Get Smart Suggestions", variant="secondary")
 
-            # --- Main Content Area ---
+            # Main Content Area
             with gr.Column(scale=4):
-                
-                # Welcome Page (Visible initially)
                 with gr.Column(visible=True) as welcome_page:
-                    gr.Markdown("# Welcome to the AI Data Explorer (Phoenix UI)", elem_id="welcome-header")
+                    gr.Markdown("# Welcome to the AI Data Explorer (Phoenix UI)")
                     gr.Markdown("Please **upload a CSV file** and **enter your Gemini API key** in the sidebar to begin.")
                     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:
                     gr.Markdown("## 📊 Data Cockpit")
                     with gr.Row():
@@ -338,15 +323,12 @@ def create_gradio_interface():
                         with gr.Column(elem_classes="stat-card"):
                             gr.Markdown("<div class='stat-card-title'>Date/Time Cols</div>")
                             time_cols_stat = gr.Textbox("0", show_label=False, elem_classes="stat-card-value")
-
                     suggestion_status = gr.Markdown(visible=True)
                     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:
                     gr.Markdown("## 🔍 Deep Dive Dashboard Builder")
-                    gr.Markdown("Create a custom dashboard by adding multiple plots to investigate your data.")
                     with gr.Row():
                         plot_type_dd = gr.Dropdown(['histogram', 'bar', 'scatter', 'box'], label="Plot Type", value='histogram')
                         x_col_dd = gr.Dropdown([], label="X-Axis / Column")
@@ -356,23 +338,19 @@ def create_gradio_interface():
                         clear_plots_btn = gr.Button("Clear Dashboard")
                     dashboard_accordion = gr.Accordion(label="Your Dashboard Plots", open=True)
 
-                # Page 3: AI Co-pilot (Hidden initially)
                 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", show_copy_button=True)
-                    
                     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")
+                        copilot_code = gr.Code(language="python", visible=False, label="Executed Python Code")
                         copilot_plot = gr.Plot(visible=False, label="Generated Visualization")
                         copilot_table = gr.Dataframe(visible=False, label="Generated Table", wrap=True)
-
                     with gr.Row():
                         chat_input = gr.Textbox(label="Your Question", placeholder="e.g., 'What is the correlation between age and salary?'", scale=4)
                         chat_submit_btn = gr.Button("Submit", variant="primary")
 
-        # --- Event Handlers ---
+        # Event Handlers
         pages = [cockpit_page, deep_dive_page, copilot_page]
         nav_buttons = [cockpit_btn, deep_dive_btn, copilot_btn]