Update app.py

app.py

@@ -101,18 +101,18 @@ def filter_investors_by_country_and_industry(selected_country, selected_industry
     investor_valuations = {}
     for investor, companies in investor_company_mapping_filtered.items():
         total_valuation = filtered_data[filtered_data["Company"].isin(companies)]["Valuation_Billions"].sum()
-        if total_valuation >= 5:  # Investors with >= 20B total valuation
+        if total_valuation >= 20:  # Investors with >= 20B total valuation
             investor_valuations[investor] = total_valuation
     
     logger.info(f"Filtered investors with total valuation >= 20B: {len(investor_valuations)}")
     
     return list(investor_valuations.keys()), filtered_data
 
-# Function to generate the graph
-def generate_graph(selected_investors, filtered_data):
+# Function to generate the graph and return node information
+def generate_graph_and_get_node_info(selected_investors, filtered_data, clicked_node=None):
     if not selected_investors:
         logger.warning("No investors selected. Returning None for graph.")
-        return None
+        return None, "No investors selected."
 
     investor_company_mapping_filtered = build_investor_company_mapping(filtered_data)
     filtered_mapping = {inv: investor_company_mapping_filtered[inv] for inv in selected_investors if inv in investor_company_mapping_filtered}
@@ -169,21 +169,22 @@ def generate_graph(selected_investors, filtered_data):
     plt.close()
     buf.seek(0)
     
-    logger.info("Graph generated successfully.")
+    # Get node information if clicked
+    node_info = "No node clicked."
+    if clicked_node:
+        if clicked_node in filtered_data["Company"].values:
+            valuation = filtered_data.loc[filtered_data["Company"] == clicked_node, "Valuation_Billions"].iloc[0]
+            node_info = f"Company: {clicked_node}, Valuation: ${valuation}B"
+        elif clicked_node in filtered_mapping:
+            node_info = f"Investor: {clicked_node}"
     
-    return Image.open(buf)
+    return Image.open(buf), node_info
 
 # Gradio app function
-def app(selected_country, selected_industry):
+def app(selected_country, selected_industry, selected_investors, clicked_node=None):
     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,
-        visible=True
-    ), filtered_data
+    graph_image, node_info = generate_graph_and_get_node_info(selected_investors, filtered_data, clicked_node)
+    return graph_image, node_info, gr.update(choices=investor_list, value=selected_investors)
 
 # Gradio Interface
 def main():
@@ -195,30 +196,33 @@ def main():
     
     with gr.Blocks() as demo:
         with gr.Row():
-            # Set default value to "US" for country and "Enterprise Tech" for industry
-            country_filter = gr.Dropdown(choices=country_list, label="Filter by Country", value="United States")
+            country_filter = gr.Dropdown(choices=country_list, label="Filter by Country", value="US")
             industry_filter = gr.Dropdown(choices=industry_list, label="Filter by Industry", value="Enterprise Tech")
         
-        filtered_investor_list = gr.CheckboxGroup(choices=[], label="Select Investors", visible=False)
+        filtered_investor_list = gr.CheckboxGroup(choices=[], label="Select Investors")
+        clicked_node_input = gr.Textbox(label="Clicked Node (Enter name to simulate a click)", placeholder="Enter a node name")
         graph_output = gr.Image(type="pil", label="Venture Network Graph")
-        
-        filtered_data_holder = gr.State()
+        node_info_output = gr.Textbox(label="Node Information", interactive=False)
         
         country_filter.change(
             app,
-            inputs=[country_filter, industry_filter],
-            outputs=[filtered_investor_list, filtered_data_holder]
+            inputs=[country_filter, industry_filter, filtered_investor_list, clicked_node_input],
+            outputs=[graph_output, node_info_output, filtered_investor_list]
         )
         industry_filter.change(
             app,
-            inputs=[country_filter, industry_filter],
-            outputs=[filtered_investor_list, filtered_data_holder]
+            inputs=[country_filter, industry_filter, filtered_investor_list, clicked_node_input],
+            outputs=[graph_output, node_info_output, filtered_investor_list]
         )
-        
         filtered_investor_list.change(
-            generate_graph,
-            inputs=[filtered_investor_list, filtered_data_holder],
-            outputs=graph_output
+            app,
+            inputs=[country_filter, industry_filter, filtered_investor_list, clicked_node_input],
+            outputs=[graph_output, node_info_output, filtered_investor_list]
+        )
+        clicked_node_input.change(
+            app,
+            inputs=[country_filter, industry_filter, filtered_investor_list, clicked_node_input],
+            outputs=[graph_output, node_info_output, filtered_investor_list]
         )
     
     demo.launch()