Update app.py

app.py

@@ -144,23 +144,24 @@ def generate_graph(investors, filtered_data):
     )
 
     node_x, node_y, node_text = [], [], []
-    node_color, node_size, node_hovertext = [], [], []
+    node_color, node_size, node_hovertext, node_textposition = [], [], [], []
 
     for node in G.nodes():
         x, y = pos[node]
         node_x.append(x)
         node_y.append(y)
         if node in investors:
-            node_text.append("")  # Remove investor labels
+            node_text.append(node)  # Add investor labels
             node_color.append(investor_color_map[node])
             node_size.append(30)
             node_hovertext.append(f"Investor: {node}")
+            node_textposition.append('top center')
         else:
             valuation = filtered_data.loc[filtered_data["Company"] == node, "Valuation_Billions"].values
             industry = filtered_data.loc[filtered_data["Company"] == node, "Industry"].values
             size = valuation[0] * 5 if len(valuation) > 0 and not pd.isnull(valuation[0]) else 15
             node_size.append(max(size, 10))
-            node_text.append("")  # No text label for companies
+            node_text.append(node)  # Add company labels
             node_color.append("#a6d854")
             hovertext = f"Company: {node}"
             if len(industry) > 0 and not pd.isnull(industry[0]):
@@ -168,14 +169,17 @@ def generate_graph(investors, filtered_data):
             if len(valuation) > 0 and not pd.isnull(valuation[0]):
                 hovertext += f"<br>Valuation: ${valuation[0]:.2f}B"
             node_hovertext.append(hovertext)
+            node_textposition.append('bottom center')
 
     node_trace = go.Scatter(
         x=node_x,
         y=node_y,
         text=node_text,
-        mode='markers',
+        textposition=node_textposition,
+        mode='markers+text',
         hoverinfo='text',
         hovertext=node_hovertext,
+        textfont=dict(size=10),  # Adjust label font size
         marker=dict(
             showscale=False,
             size=node_size,
@@ -197,8 +201,8 @@ def generate_graph(investors, filtered_data):
         width=1200,
         height=800,
         autosize=True,
-        xaxis={'showgrid': False, 'zeroline': False, 'visible': False},
-        yaxis={'showgrid': False, 'zeroline': False, 'visible': False},
+        xaxis=dict(showgrid=False, zeroline=False, visible=False),
+        yaxis=dict(showgrid=False, zeroline=False, visible=False),
         showlegend=False,  # Hide the legend to maximize space
         annotations=[
             dict(
@@ -214,8 +218,10 @@ def generate_graph(investors, filtered_data):
 # Gradio app
 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)
+    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 go.Figure()
     graph = generate_graph(investors, filtered_data)