Update app.py

app.py

@@ -1,6 +1,7 @@
 # app.py
-# Fixed Mutual Fund Churn Explorer - Gradio app
-# Fixes ValueError: numpy.float64 passed to layout.width by coercing to int and enforcing minimums.
+# Mutual Fund Churn Explorer - final version with collapsible sidebar + deep green theme
+# Save as app.py and run with: python app.py
+# requirements.txt should include: gradio, networkx, plotly, pandas, numpy
 
 import gradio as gr
 import pandas as pd
@@ -85,6 +86,7 @@ def maps_from_dataframe(df, amc_col="AMC", company_col="Company", action_col="Ac
         elif act in ("fresh_buy", "fresh", "new"):
             fresh_buy[a].append(c)
         else:
+            # fallback heuristics
             if "sell" in act:
                 sell_map[a].append(c)
             elif "exit" in act:
@@ -180,13 +182,13 @@ def build_graph(AMCS, COMPANIES, BUY_MAP, SELL_MAP, COMPLETE_EXIT, FRESH_BUY, in
 # Plotly visualizer (coerce width/height -> int with minimums)
 # ---------------------------
 def graph_to_plotly(G,
-                    node_color_amc="#9EC5FF",
-                    node_color_company="#FFCF9E",
+                    node_color_amc="#0f5132",       # deep green theme: use darker green for AMCs by default
+                    node_color_company="#ffc107",    # amber for companies — user can change
                     node_shape_amc="circle",
                     node_shape_company="circle",
-                    edge_color_buy="#2ca02c",
-                    edge_color_sell="#d62728",
-                    edge_color_transfer="#888888",
+                    edge_color_buy="#28a745",
+                    edge_color_sell="#dc3545",
+                    edge_color_transfer="#6c757d",
                     edge_thickness_base=1.2,
                     show_labels=True,
                     width=1400,
@@ -205,6 +207,7 @@ def graph_to_plotly(G,
     if height < 360:
         height = 360
 
+    # layout: spring layout for clarity
     pos = nx.spring_layout(G, seed=42, k=1.4)
 
     node_x = []
@@ -233,12 +236,14 @@ def graph_to_plotly(G,
         hoverinfo='text'
     )
 
+    # edges rendered individually so they can be styled
     edge_traces = []
     for u, v, attrs in G.edges(data=True):
         x0, y0 = pos[u]
         x1, y1 = pos[v]
         actions = attrs.get("actions", [])
         weight = float(attrs.get("weight", 1.0))
+        # priority styling
         if "complete_exit" in actions:
             color = edge_color_sell
             dash = "solid"
@@ -273,7 +278,7 @@ def graph_to_plotly(G,
 
     fig = go.Figure(data=edge_traces + [node_trace],
                     layout=go.Layout(
-                        title_text="Mutual Fund Churn Network (AMCs: blue, Companies: orange)",
+                        title_text="Mutual Fund Churn Network (AMCs: green, Companies: amber)",
                         title_x=0.5,
                         showlegend=False,
                         margin=dict(b=20,l=5,r=5,t=40),
@@ -354,35 +359,47 @@ def build_initial_graph_and_data():
 (AMCS, COMPANIES, BUY_MAP, SELL_MAP, COMPLETE_EXIT, FRESH_BUY, G_initial, initial_fig) = build_initial_graph_and_data()
 
 # ---------------------------
-# Gradio UI
+# GRADIO UI: deep-green theme + collapsible sidebar
 # ---------------------------
-with gr.Blocks() as demo:
-    gr.Markdown("# Mutual Fund Churn Explorer (fixed layout issue)")
+# Create a deep green theme using gradio theme API (primary hue green, darker accents)
+deep_green_theme = gr.themes.Soft(primary_hue="green", secondary_hue="teal", spacing_size="md")
+
+with gr.Blocks(theme=deep_green_theme) as demo:
+    gr.Markdown("# Mutual Fund Churn Explorer — Deep Green Theme")
     with gr.Row():
-        with gr.Column(scale=3):
-            csv_uploader = gr.File(label="Upload CSV (optional). Columns: AMC,Company,Action", file_types=['.csv'])
-            node_color_company = gr.ColorPicker(value="#FFCF9E", label="Company node color")
-            node_color_amc = gr.ColorPicker(value="#9EC5FF", label="AMC node color")
-            node_shape_company = gr.Dropdown(choices=["circle","square","diamond"], value="circle", label="Company node shape")
-            node_shape_amc = gr.Dropdown(choices=["circle","square","diamond"], value="circle", label="AMC node shape")
-            edge_color_buy = gr.ColorPicker(value="#2ca02c", label="BUY edge color")
-            edge_color_sell = gr.ColorPicker(value="#d62728", label="SELL edge color")
-            edge_color_transfer = gr.ColorPicker(value="#888888", label="Transfer edge color")
-            edge_thickness = gr.Slider(minimum=0.5, maximum=8.0, value=1.4, step=0.1, label="Edge thickness base")
-            include_transfers_chk = gr.Checkbox(value=True, label="Infer AMC→AMC transfers (show loops)")
-            update_btn = gr.Button("Update network")
-            gr.Markdown("## Inspect")
-            company_selector = gr.Dropdown(choices=COMPANIES, label="Select Company (show buyers/sellers)")
-            amc_selector = gr.Dropdown(choices=AMCS, label="Select AMC (inferred transfers)")
-        with gr.Column(scale=7):
+        # Collapsible sidebar using Accordion
+        with gr.Accordion("⚙️ Settings (click to expand / collapse)", open=False):
+            with gr.Column():
+                gr.Markdown("### Data Input")
+                csv_uploader = gr.File(label="Upload CSV (optional). Columns: AMC,Company,Action", file_types=['.csv'])
+                gr.Markdown("### Node Appearance")
+                node_color_company = gr.ColorPicker(value="#ffc107", label="Company node color")
+                node_color_amc = gr.ColorPicker(value="#0f5132", label="AMC node color")
+                node_shape_company = gr.Dropdown(choices=["circle","square","diamond"], value="circle", label="Company node shape")
+                node_shape_amc = gr.Dropdown(choices=["circle","square","diamond"], value="circle", label="AMC node shape")
+                gr.Markdown("### Edge Appearance")
+                edge_color_buy = gr.ColorPicker(value="#28a745", label="BUY edge color")
+                edge_color_sell = gr.ColorPicker(value="#dc3545", label="SELL edge color")
+                edge_color_transfer = gr.ColorPicker(value="#6c757d", label="Transfer edge color")
+                edge_thickness = gr.Slider(minimum=0.5, maximum=8.0, value=1.4, step=0.1, label="Edge thickness base")
+                include_transfers_chk = gr.Checkbox(value=True, label="Infer AMC→AMC transfers (show loops)")
+                update_btn = gr.Button("Update network")
+                gr.Markdown("### Inspect")
+                # do not set default value to avoid mismatch warnings after dropdown choices update
+                company_selector = gr.Dropdown(choices=COMPANIES, label="Select Company (show buyers/sellers)")
+                amc_selector = gr.Dropdown(choices=AMCS, label="Select AMC (inferred transfers)")
+        # Graph column
+        with gr.Column():
             network_plot = gr.Plot(value=initial_fig, label="Network graph (drag to zoom)")
 
+    # outputs for inspection
     company_plot = gr.Plot(label="Company trade summary")
     company_table = gr.Dataframe(headers=["Role","AMC"], interactive=False, label="Trades (company)")
     amc_plot = gr.Plot(label="AMC inferred transfers")
     amc_table = gr.Dataframe(headers=["security","buyer_amc"], interactive=False, label="Inferred transfers (AMC)")
     loops_text = gr.Markdown()
 
+    # CSV loader helper
     def load_dataset_from_csv(file_obj):
         if file_obj is None:
             return AMCS, COMPANIES, BUY_MAP, SELL_MAP, COMPLETE_EXIT, FRESH_BUY
@@ -410,6 +427,7 @@ with gr.Blocks() as demo:
             print("CSV load error:", e)
             return AMCS, COMPANIES, BUY_MAP, SELL_MAP, COMPLETE_EXIT, FRESH_BUY
 
+    # Update callback builds new graph, detects loops and refreshes dropdown choices
     def on_update(csv_file, node_color_company_val, node_color_amc_val, node_shape_company_val, node_shape_amc_val,
                   edge_color_buy_val, edge_color_sell_val, edge_color_transfer_val, edge_thickness_val, include_transfers_val):
         amcs, companies, buy_map, sell_map, complete_exit, fresh_buy = load_dataset_from_csv(csv_file)
@@ -438,9 +456,23 @@ with gr.Blocks() as demo:
                              edge_color_buy, edge_color_sell, edge_color_transfer, edge_thickness, include_transfers_chk],
                      outputs=[network_plot, loops_text, company_selector, amc_selector])
 
+    # Defensive normalizer for dropdown values (sometimes Gradio returns list)
+    def normalize_dropdown_value(val):
+        if val is None:
+            return None
+        if isinstance(val, list):
+            return val[0] if len(val) > 0 else None
+        try:
+            return str(val)
+        except Exception:
+            return None
+
     def on_company_sel(company_name, csv_file):
+        cname = normalize_dropdown_value(company_name)
+        if cname is None:
+            return None, pd.DataFrame([], columns=["Role","AMC"])
         amcs, companies, buy_map, sell_map, complete_exit, fresh_buy = load_dataset_from_csv(csv_file)
-        fig, df = company_trade_summary(company_name, buy_map, sell_map, fresh_buy, complete_exit)
+        fig, df = company_trade_summary(cname, buy_map, sell_map, fresh_buy, complete_exit)
         if fig is None:
             return None, pd.DataFrame([], columns=["Role","AMC"])
         return fig, df
@@ -448,8 +480,11 @@ with gr.Blocks() as demo:
     company_selector.change(on_company_sel, inputs=[company_selector, csv_uploader], outputs=[company_plot, company_table])
 
     def on_amc_sel(amc_name, csv_file):
+        aname = normalize_dropdown_value(amc_name)
+        if aname is None:
+            return None, pd.DataFrame([], columns=["security","buyer_amc"])
         amcs, companies, buy_map, sell_map, complete_exit, fresh_buy = load_dataset_from_csv(csv_file)
-        fig, df = amc_transfer_summary(amc_name, buy_map, sell_map)
+        fig, df = amc_transfer_summary(aname, buy_map, sell_map)
         if fig is None:
             return None, pd.DataFrame([], columns=["security","buyer_amc"])
         return fig, df