Update app.py

app.py

@@ -54,19 +54,23 @@ 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 and industry (removed valuation threshold)
-def filter_investors(selected_country, selected_industry):
+# Filter investors by country, industry, and investor selection
+def filter_investors(selected_country, selected_industry, selected_investor):
     filtered_data = data.copy()
     if selected_country != "All":
         filtered_data = filtered_data[filtered_data["Country"] == selected_country]
     if selected_industry != "All":
         filtered_data = filtered_data[filtered_data["Industry"] == selected_industry]
+    if selected_investor != "All":
+        filtered_data = filtered_data[
+            filtered_data["Select_Investors"].str.contains(selected_investor, na=False)
+        ]
 
     investor_company_mapping_filtered = build_investor_company_mapping(filtered_data)
     filtered_investors = list(investor_company_mapping_filtered.keys())
     return filtered_investors, filtered_data
 
-# Generate Plotly graph with increased size and improved styling
+# Generate Plotly graph
 def generate_graph(investors, filtered_data):
     if not investors:
         logger.warning("No investors selected.")
@@ -105,21 +109,24 @@ def generate_graph(investors, filtered_data):
         x, y = pos[node]
         node_x.append(x)
         node_y.append(y)
-        node_text.append(node)
-        node_color.append(10)  # Use a fixed color or other logic
+        if node in investors:
+            node_text.append(node)  # Label investors
+            node_color.append(20)  # Fixed color for investors
+        else:
+            node_text.append("")  # Hide company labels by default
+            node_color.append(10)
 
     node_trace = go.Scatter(
         x=node_x,
         y=node_y,
         text=node_text,
-        mode='markers+text',
+        mode='markers',
         hoverinfo='text',
         marker=dict(
             showscale=False,
-            size=15,  # Increased size
+            size=15,
             color=node_color,
-        ),
-        textposition="top center"  # Improved label positioning
+        )
     )
 
     fig = go.Figure(data=[edge_trace, node_trace])
@@ -129,14 +136,14 @@ def generate_graph(investors, filtered_data):
         titlefont_size=20,
         margin=dict(l=20, r=20, t=50, b=20),
         hovermode='closest',
-        width=1200,  # Increased width
-        height=800   # Increased height
+        width=1200,
+        height=800
     )
     return fig
 
-# Update the Gradio app to remove valuation threshold
-def app(selected_country, selected_industry):
-    investors, filtered_data = filter_investors(selected_country, selected_industry)
+# Gradio app
+def app(selected_country, selected_industry, selected_investor):
+    investors, filtered_data = filter_investors(selected_country, selected_industry, selected_investor)
     graph = generate_graph(investors, filtered_data)
     return investors, graph
 
@@ -144,17 +151,26 @@ def app(selected_country, selected_industry):
 def main():
     country_list = ["All"] + sorted(data["Country"].dropna().unique())
     industry_list = ["All"] + sorted(data["Industry"].dropna().unique())
+    investor_list = ["All"] + sorted(investor_company_mapping.keys())
 
     with gr.Blocks() as demo:
         with gr.Row():
             country_filter = gr.Dropdown(choices=country_list, label="Country", value="All")
             industry_filter = gr.Dropdown(choices=industry_list, label="Industry", value="All")
+            investor_filter = gr.Dropdown(choices=investor_list, label="Investor", value="All")
 
         investor_output = gr.Textbox(label="Filtered Investors")
         graph_output = gr.Plot(label="Network Graph")
 
-        country_filter.change(app, [country_filter, industry_filter], [investor_output, graph_output])
-        industry_filter.change(app, [country_filter, industry_filter], [investor_output, graph_output])
+        country_filter.change(
+            app, [country_filter, industry_filter, investor_filter], [investor_output, graph_output]
+        )
+        industry_filter.change(
+            app, [country_filter, industry_filter, investor_filter], [investor_output, graph_output]
+        )
+        investor_filter.change(
+            app, [country_filter, industry_filter, investor_filter], [investor_output, graph_output]
+        )
 
     demo.launch()