Update app.py

app.py

@@ -184,18 +184,11 @@ def generate_graph(investors, filtered_data):
         )
     )
 
-    legend_items = [
-        go.Scatter(
-            x=[None], y=[None], mode='markers',
-            marker=dict(size=10, color=investor_color_map[investor]),
-            legendgroup=investor, showlegend=True, name=investor
-        ) for investor in unique_investors
-    ]
-
     # Compute total market cap
     total_market_cap = filtered_data["Valuation_Billions"].sum()
 
-    fig = go.Figure(data=legend_items + [edge_trace, node_trace])
+    fig = go.Figure(data=[edge_trace, node_trace])
+
     fig.update_layout(
         title="",
         titlefont_size=24,
@@ -206,6 +199,7 @@ def generate_graph(investors, filtered_data):
         autosize=True,
         xaxis={'showgrid': False, 'zeroline': False, 'visible': False},
         yaxis={'showgrid': False, 'zeroline': False, 'visible': False},
+        showlegend=False,  # Hide the legend to maximize space
         annotations=[
             dict(
                 x=0.5, y=1.1, xref='paper', yref='paper',
@@ -223,9 +217,9 @@ def app(selected_country, selected_industry, selected_company, selected_investor
     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()
+        return go.Figure()
     graph = generate_graph(investors, filtered_data)
-    return ', '.join(investors), graph
+    return graph
 
 def main():
     country_list = ["All"] + sorted(data["Country"].dropna().unique())
@@ -245,13 +239,11 @@ def main():
             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,
                   exclude_country_filter, exclude_industry_filter, exclude_company_filter, exclude_investor_filter]
-        outputs = [investor_output, graph_output]
+        outputs = [graph_output]
 
         # Set up event triggers for all inputs
         for input_control in inputs: