Update app.py

app.py

@@ -66,8 +66,8 @@ def build_investor_company_mapping(df):
 investor_company_mapping = build_investor_company_mapping(data)
 logger.info("Investor to company mapping created.")
 
-# Filter investors by country, industry, and investor selection
-def filter_investors(selected_country, selected_industry, selected_investors):
+# Filter investors by country, industry, investor selection, and company selection
+def filter_investors(selected_country, selected_industry, selected_investors, selected_company):
     filtered_data = data.copy()
     if selected_country != "All":
         filtered_data = filtered_data[filtered_data["Country"] == selected_country]
@@ -76,7 +76,9 @@ def filter_investors(selected_country, selected_industry, selected_investors):
     if selected_investors:
         pattern = '|'.join([re.escape(inv) for inv in selected_investors])
         filtered_data = filtered_data[filtered_data["Select_Investors"].str.contains(pattern, na=False)]
-    
+    if selected_company != "All":
+        filtered_data = filtered_data[filtered_data["Company"] == selected_company]
+
     investor_company_mapping_filtered = build_investor_company_mapping(filtered_data)
     filtered_investors = list(investor_company_mapping_filtered.keys())
     return filtered_investors, filtered_data
@@ -101,7 +103,6 @@ def generate_graph(investors, filtered_data):
         "#f781bf",  # Pink
         "#999999",  # Grey
     ]
-    # Extend color_palette if necessary
     while num_colors > len(color_palette):
         color_palette.extend(color_palette)
     
@@ -145,24 +146,22 @@ def generate_graph(investors, filtered_data):
         node_x.append(x)
         node_y.append(y)
         if node in investors:
-            # Investor node
-            node_text.append(node)  # Label investors
-            node_color.append(investor_color_map[node])  # Color assigned to investor
-            node_size.append(30)  # Fixed size for investors
+            node_text.append(node)
+            node_color.append(investor_color_map[node])
+            node_size.append(30)
             node_hovertext.append(f"Investor: {node}")
         else:
-            # Company node
             valuation = filtered_data.loc[filtered_data["Company"] == node, "Valuation_Billions"].values
             industry = filtered_data.loc[filtered_data["Company"] == node, "Industry"].values
             if len(valuation) > 0 and not pd.isnull(valuation[0]):
-                size = valuation[0] * 5  # Scale size as needed
+                size = valuation[0] * 5
                 if size < 10:
-                    size = 10  # Minimum size
+                    size = 10
             else:
-                size = 15  # Default size
+                size = 15
             node_size.append(size)
-            node_text.append("")  # Hide company labels by default
-            node_color.append("#a6d854")  # Light green color for companies
+            node_text.append("")
+            node_color.append("#a6d854")
             hovertext = f"Company: {node}"
             if len(industry) > 0 and not pd.isnull(industry[0]):
                 hovertext += f"<br>Industry: {industry[0]}"
@@ -187,7 +186,6 @@ def generate_graph(investors, filtered_data):
         textfont=dict(size=12, color="#000000")
     )
     
-    # Add legend manually
     legend_items = []
     for investor in unique_investors:
         legend_items.append(
@@ -215,7 +213,6 @@ def generate_graph(investors, filtered_data):
         height=800
     )
     
-    # Improve layout responsiveness
     fig.update_layout(
         autosize=True,
         xaxis={'showgrid': False, 'zeroline': False, 'visible': False},
@@ -225,8 +222,8 @@ def generate_graph(investors, filtered_data):
     return fig
 
 # Gradio app
-def app(selected_country, selected_industry, selected_investors):
-    investors, filtered_data = filter_investors(selected_country, selected_industry, selected_investors)
+def app(selected_country, selected_industry, selected_company, selected_investors):
+    investors, filtered_data = filter_investors(selected_country, selected_industry, selected_investors, selected_company)
     if not investors:
         return "No investors found with the selected filters.", go.Figure()
     graph = generate_graph(investors, filtered_data)
@@ -234,9 +231,9 @@ def app(selected_country, selected_industry, selected_investors):
 
 # Main function
 def main():
-    import re  # Added import for regex
     country_list = ["All"] + sorted(data["Country"].dropna().unique())
     industry_list = ["All"] + sorted(data["Industry"].dropna().unique())
+    company_list = ["All"] + sorted(data["Company"].dropna().unique())
     investor_list = sorted(investor_company_mapping.keys())
     
     with gr.Blocks(title="Venture Networks Visualization") as demo:
@@ -245,43 +242,24 @@ def main():
         Explore the connections between investors and companies in the venture capital ecosystem. Use the filters below to customize the network graph.
         """)
         with gr.Row():
-            country_filter = gr.Dropdown(
-                choices=country_list,
-                label="Country",
-                value="All",
-                info="Filter companies by country."
-            )
-            industry_filter = gr.Dropdown(
-                choices=industry_list,
-                label="Industry",
-                value="All",
-                info="Filter companies by industry."
-            )
-            investor_filter = gr.Dropdown(
-                choices=investor_list,
-                label="Select Investors",
-                value=[],
-                multiselect=True,
-                info="Select one or more investors to visualize."
-            )
+            country_filter = gr.Dropdown(choices=country_list, label="Country", value="All")
+            industry_filter = gr.Dropdown(choices=industry_list, label="Industry", value="All")
+            company_filter = gr.Dropdown(choices=company_list, label="Company", value="All")
+            investor_filter = gr.Dropdown(choices=investor_list, label="Select Investors", value=[], multiselect=True)
         with gr.Row():
             investor_output = gr.Textbox(label="Filtered Investors", interactive=False)
         graph_output = gr.Plot(label="Network Graph")
         
-        inputs = [country_filter, industry_filter, investor_filter]
+        inputs = [country_filter, industry_filter, company_filter, investor_filter]
         outputs = [investor_output, graph_output]
         
-        # Update the graph when any filter changes
         country_filter.change(app, inputs, outputs)
         industry_filter.change(app, inputs, outputs)
+        company_filter.change(app, inputs, outputs)
         investor_filter.change(app, inputs, outputs)
         
         gr.Markdown("""
-        **Instructions:**
-        - **Country**: Select a country to filter companies based on their location.
-        - **Industry**: Choose an industry to focus on companies within that sector.
-        - **Select Investors**: Pick one or more investors to visualize their network.
-        
+        **Instructions:** Use the dropdowns to filter the network graph.
         """)
     
     demo.launch()