Update app.py

app.py

@@ -31,24 +31,40 @@ for _, row in data.iterrows():
                 investor_company_mapping[investor] = []
             investor_company_mapping[investor].append(company)
 
-# Gradio app function
-def generate_graph(selected_investors, selected_country, selected_industry):
+# Gradio app functions
+def filter_investors_by_country_and_industry(selected_country, selected_industry):
     filtered_data = data
 
-    # Apply country filter
+    # Apply filters
     if selected_country != "All":
         filtered_data = filtered_data[filtered_data["Country"] == selected_country]
-
-    # Apply industry filter
     if selected_industry != "All":
         filtered_data = filtered_data[filtered_data["Industry"] == selected_industry]
 
+    # Calculate total valuation per investor
+    investor_valuations = {}
+    for investor, companies in investor_company_mapping.items():
+        total_valuation = 0
+        for company in companies:
+            if company in filtered_data["Company"].values:
+                valuation = filtered_data.loc[filtered_data["Company"] == company, "Valuation_Billions"].values
+                total_valuation += valuation[0] if len(valuation) > 0 else 0
+        if total_valuation >= 20:  # Filter by total valuation
+            investor_valuations[investor] = total_valuation
+
+    return list(investor_valuations.keys()), filtered_data
+
+def generate_graph(selected_investors, filtered_data):
+    if not selected_investors:
+        return None
+
     # Filter the investor-to-company mapping
     filtered_mapping = {}
     for investor, companies in investor_company_mapping.items():
-        filtered_companies = [c for c in companies if c in filtered_data["Company"].values]
-        if filtered_companies:
-            filtered_mapping[investor] = filtered_companies
+        if investor in selected_investors:
+            filtered_companies = [c for c in companies if c in filtered_data["Company"].values]
+            if filtered_companies:
+                filtered_mapping[investor] = filtered_companies
 
     # Use the filtered mapping to build the graph
     G = nx.Graph()
@@ -99,42 +115,52 @@ def generate_graph(selected_investors, selected_country, selected_industry):
     image = Image.open(buf)
     return image
 
+def app(selected_country, selected_industry):
+    investor_list, filtered_data = filter_investors_by_country_and_industry(selected_country, selected_industry)
+
+    return gr.update(
+        choices=investor_list,
+        value=investor_list,
+        visible=True
+    ), filtered_data
+
 # Gradio Interface
 def main():
-    investor_list = sorted(investor_company_mapping.keys())
     country_list = ["All"] + sorted(data["Country"].dropna().unique())
     industry_list = ["All"] + sorted(data["Industry"].dropna().unique())
 
-    iface = gr.Interface(
-        fn=generate_graph,
-        inputs=[
-            gr.CheckboxGroup(
-                choices=investor_list,
-                label="Select Investors",
-                value=investor_list  # Default to all selected
-            ),
-            gr.Dropdown(
-                choices=country_list,
-                label="Filter by Country",
-                value="All"  # Default to no filter
-            ),
-            gr.Dropdown(
-                choices=industry_list,
-                label="Filter by Industry",
-                value="All"  # Default to no filter
-            )
-        ],
-        outputs=gr.Image(type="pil", label="Venture Network Graph"),
-        title="Venture Networks Visualization",
-        description=(
-            "Select investors and apply optional filters by country and industry "
-            "to visualize their investments. The graph shows connections between "
-            "investors and the companies they've invested in. Node sizes represent company valuations."
-        ),
-        flagging_mode="never"
-    )
-
-    iface.launch()
+    with gr.Blocks() as demo:
+        with gr.Row():
+            country_filter = gr.Dropdown(choices=country_list, label="Filter by Country", value="All")
+            industry_filter = gr.Dropdown(choices=industry_list, label="Filter by Industry", value="All")
+
+        filtered_investor_list = gr.CheckboxGroup(
+            choices=[],
+            label="Select Investors",
+            visible=False
+        )
+        graph_output = gr.Image(type="pil", label="Venture Network Graph")
+
+        filtered_data_holder = gr.State()
+
+        country_filter.change(
+            app,
+            inputs=[country_filter, industry_filter],
+            outputs=[filtered_investor_list, filtered_data_holder]
+        )
+        industry_filter.change(
+            app,
+            inputs=[country_filter, industry_filter],
+            outputs=[filtered_investor_list, filtered_data_holder]
+        )
+
+        filtered_investor_list.change(
+            generate_graph,
+            inputs=[filtered_investor_list, filtered_data_holder],
+            outputs=graph_output
+        )
+
+    demo.launch()
 
 if __name__ == "__main__":
     main()