Update app.py

app.py

@@ -17,7 +17,7 @@ def categorize_rank(r):
 
 def process_data(file):
     if file is None:
-        return None, None, None
+        return None, None, None, gr.update(choices=[]), gr.update(choices=[])
     
     excel = pd.ExcelFile(file.name)
     first_sheet = excel.sheet_names[0]
@@ -72,36 +72,35 @@ def process_data(file):
             x='Range', 
             y='Keyword Count', 
             color='Month',
-            barmode='group', # <--- This ensures side-by-side bars
+            barmode='group', # Side-by-side bars
             text_auto=True,
-            title="Keyword Performance Distribution",
-            category_orders={"Range": correct_order} # <--- This enforces your custom sort order
+            title="Keyword Performance Distribution (Grouped)",
+            category_orders={"Range": correct_order}
         )
-        # Enable clicking
-        fig.update_layout(clickmode='event+select', dragmode=False)
     else:
         fig = None
     
-    return final_df, fig, final_df
+    # Return data and update dropdown choices
+    unique_months = list(full_viz_df['Month'].unique()) if not full_viz_df.empty else []
+    return (
+        final_df, 
+        fig, 
+        final_df, 
+        gr.Dropdown.update(choices=unique_months, value=unique_months[0] if unique_months else None),
+        gr.Dropdown.update(choices=correct_order)
+    )
 
-def filter_table(full_data, selection_data: gr.SelectData):
-    if not selection_data: return full_data
+def filter_table(full_data, selected_month, selected_range):
+    if full_data is None or full_data.empty: return None
     
-    # Extract what was clicked
-    clicked_range = selection_data.value['x'] # e.g. "01-03"
+    # Filter by specific month AND range
+    target_col = f"{selected_month}_Rank Range"
     
-    # Try to find which specific month (color) was clicked
-    clicked_month = None
-    # Plotly passes legend info differently depending on interaction
-    # We check the raw selection data
-    if selection_data.value.get('curveNumber') is not None:
-        # We can try to infer, but filtering by Range alone is safer and more robust
-        pass
-
-    # Filter Logic: Return keywords that match this range in ANY of the months
-    range_cols = [c for c in full_data.columns if "Rank Range" in c]
-    filtered_df = full_data[full_data[range_cols].apply(lambda row: clicked_range in row.values, axis=1)]
-    return filtered_df
+    if target_col in full_data.columns:
+        filtered_df = full_data[full_data[target_col] == selected_range]
+        return filtered_df
+    
+    return full_data
 
 def reset_table(full_data):
     return full_data
@@ -114,28 +113,35 @@ with gr.Blocks() as demo:
     
     with gr.Row():
         file_input = gr.File(label="Upload SEO Excel", file_types=[".xlsx"])
-        with gr.Column():
-            process_btn = gr.Button("📊 Build Dashboard", variant="primary")
-            reset_btn = gr.Button("🔄 Reset Filters")
+        process_btn = gr.Button("📊 Build Dashboard", variant="primary")
 
-    plot_output = gr.Plot(label="Interactive Chart")
+    # The Chart
+    plot_output = gr.Plot(label="Keyword Distribution")
     
-    gr.Markdown("### 🔍 Keywords List")
+    gr.Markdown("### 🔍 Filter Keywords")
+    with gr.Row():
+        month_dropdown = gr.Dropdown(label="Select Month", choices=[])
+        range_dropdown = gr.Dropdown(label="Select Rank Range", choices=[])
+        filter_btn = gr.Button("Apply Filter")
+        reset_btn = gr.Button("Show All")
+
     table_output = gr.DataFrame(interactive=False)
 
+    # 1. Process File
     process_btn.click(
         fn=process_data,
         inputs=file_input,
-        outputs=[table_output, plot_output, full_data_state]
+        outputs=[table_output, plot_output, full_data_state, month_dropdown, range_dropdown]
     )
 
-    # This .select method requires Gradio >= 4.0
-    plot_output.select(
+    # 2. Filter Table (Button Click)
+    filter_btn.click(
         fn=filter_table,
-        inputs=[full_data_state],
+        inputs=[full_data_state, month_dropdown, range_dropdown],
         outputs=table_output
     )
 
+    # 3. Reset Table
     reset_btn.click(
         fn=reset_table,
         inputs=full_data_state,
@@ -143,4 +149,4 @@ with gr.Blocks() as demo:
     )
 
 if __name__ == "__main__":
-    demo.launch(theme=gr.themes.Soft())
+    demo.launch()