Update app.py

app.py

@@ -70,17 +70,31 @@ investor_company_mapping = build_investor_company_mapping(data)
 logger.info("Investor to company mapping created.")
 
 # Filter investors by country, industry, investor selection, and company selection
-def filter_investors(selected_country, selected_industry, selected_investors, selected_company):
+def filter_investors(selected_country, selected_industry, selected_investors, selected_company,
+                     exclude_countries, exclude_industries, exclude_companies, exclude_investors):
     filtered_data = data.copy()
+
+    # Inclusion filters
     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_company != "All":
+        filtered_data = filtered_data[filtered_data["Company"] == selected_company]
     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]
+
+    # Exclusion filters
+    if exclude_countries:
+        filtered_data = filtered_data[~filtered_data["Country"].isin(exclude_countries)]
+    if exclude_industries:
+        filtered_data = filtered_data[~filtered_data["Industry"].isin(exclude_industries)]
+    if exclude_companies:
+        filtered_data = filtered_data[~filtered_data["Company"].isin(exclude_companies)]
+    if exclude_investors:
+        pattern = '|'.join([re.escape(inv) for inv in exclude_investors])
+        filtered_data = filtered_data[~filtered_data["Select_Investors"].str.contains(pattern, na=False)]
 
     investor_company_mapping_filtered = build_investor_company_mapping(filtered_data)
     filtered_investors = list(investor_company_mapping_filtered.keys())
@@ -204,8 +218,10 @@ def generate_graph(investors, filtered_data):
     return fig
 
 # Gradio app
-def app(selected_country, selected_industry, selected_company, selected_investors):
-    investors, filtered_data = filter_investors(selected_country, selected_industry, selected_investors, selected_company)
+def app(selected_country, selected_industry, selected_company, selected_investors,
+        exclude_countries, exclude_industries, exclude_companies, exclude_investors):
+    investors, filtered_data = filter_investors(selected_country, selected_industry, selected_investors, selected_company,
+                                                exclude_countries, exclude_industries, exclude_companies, exclude_investors)
     if not investors:
         return "No investors found with the selected filters.", go.Figure()
     graph = generate_graph(investors, filtered_data)
@@ -224,20 +240,25 @@ def main():
             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():
+            exclude_country_filter = gr.Dropdown(choices=country_list[1:], label="Exclude Country", value=[], multiselect=True)
+            exclude_industry_filter = gr.Dropdown(choices=industry_list[1:], label="Exclude Industry", value=[], multiselect=True)
+            exclude_company_filter = gr.Dropdown(choices=company_list[1:], label="Exclude Company", value=[], multiselect=True)
+            exclude_investor_filter = gr.Dropdown(choices=investor_list, label="Exclude 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, company_filter, investor_filter]
+
+        inputs = [country_filter, industry_filter, company_filter, investor_filter,
+                  exclude_country_filter, exclude_industry_filter, exclude_company_filter, exclude_investor_filter]
         outputs = [investor_output, graph_output]
-        
-        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:** Use the dropdowns to filter the network graph.")
-    
+
+        # Set up event triggers for all inputs
+        for input_control in inputs:
+            input_control.change(app, inputs, outputs)
+
+        gr.Markdown("**Instructions:** Use the dropdowns to filter the network graph. You can include or exclude specific countries, industries, companies, or investors.")
+
     demo.launch()
 
 if __name__ == "__main__":