Update app.py

app.py

@@ -205,7 +205,6 @@ def app(selected_country, selected_industry):
     investor_list, filtered_data = filter_investors_by_country_and_industry(selected_country, selected_industry)
     logger.info("Updating CheckboxGroup and filtered data holder.")
     
-    # Use gr.update() to create an update dictionary for the CheckboxGroup
     return gr.update(
         choices=investor_list,
         value=investor_list,
@@ -217,7 +216,6 @@ def main():
     country_list = ["All"] + sorted(data["Country"].dropna().unique())
     industry_list = ["All"] + sorted(data["Industry"].dropna().unique())
     
-    # Ensure the default values for dropdowns exist
     default_country = "United States" if "United States" in country_list else "All"
     default_industry = "Enterprise Tech" if "Enterprise Tech" in industry_list else "All"
     
@@ -226,7 +224,6 @@ def main():
     
     with gr.Blocks() as demo:
         with gr.Row():
-            # Set default value for country and industry dropdowns
             country_filter = gr.Dropdown(choices=country_list, label="Filter by Country", value=default_country)
             industry_filter = gr.Dropdown(choices=industry_list, label="Filter by Industry", value=default_industry)
         
@@ -236,7 +233,6 @@ def main():
         
         filtered_data_holder = gr.State()
         
-        # Event handlers for filters
         country_filter.change(
             app,
             inputs=[country_filter, industry_filter],
@@ -248,25 +244,24 @@ def main():
             outputs=[filtered_investor_list, filtered_data_holder]
         )
         
-        # Generate graph when investors are selected
         filtered_investor_list.change(
             generate_graph,
             inputs=[filtered_investor_list, filtered_data_holder],
             outputs=graph_output
         )
         
-        # Handle plot click to display valuation
-        def display_valuation(plotly_event):
-            if not plotly_event or "points" not in plotly_event or not plotly_event["points"]:
+        def display_valuation(click_data):
+            if not click_data or not click_data.get("points"):
                 return "Click on a company node to see its valuation."
-            point_data = plotly_event["points"][0]
-            if "customdata" in point_data and point_data["customdata"]:
-                return f"**Valuation:** {point_data['customdata']}"
+            
+            point = click_data["points"][0]
+            if "customdata" in point and point["customdata"]:
+                return f"**Valuation:** {point['customdata']}"
             return "Click on a company node to see its valuation."
         
-        graph_output.events().on_click(
+        graph_output.select(
             fn=display_valuation,
-            inputs=[graph_output],
+            inputs=graph_output,
             outputs=valuation_display
         )