Update app.py

app.py

@@ -475,6 +475,9 @@ def make_arc_html(nodes, node_type, buys, sells, transfers, loops):
 
 initial_html = make_arc_html(NODES, NODE_TYPE, BUYS, SELLS, TRANSFERS, LOOPS)
 
+# ---------------------------
+# Gradio UI
+# ---------------------------
 # ---------------------------
 # Gradio UI
 # ---------------------------
@@ -488,16 +491,17 @@ with gr.Blocks(css=responsive_css, title="MF Churn — Semi-layer Arc Diagram (L
     gr.HTML(initial_html)
 
     gr.Markdown("### Inspect Company / AMC")
+    
     select_company = gr.Dropdown(choices=COMPANIES, label="Select company")
-    company_plot = gr.Plotly()
+    company_plot = gr.Plot()
     company_table = gr.DataFrame()
+    
     select_amc = gr.Dropdown(choices=AMCS, label="Select AMC")
-    amc_plot = gr.Plotly()
+    amc_plot = gr.Plot()
     amc_table = gr.DataFrame()
 
     def on_company(c):
         fig, df = company_trade_summary(c)
-        # Ensure DataFrame is returned as-is for gr.DataFrame
         return fig, df
 
     def on_amc(a):