Update app.py

app.py

@@ -79,7 +79,7 @@ def generate_graph(selected_investors, filtered_data):
             node_sizes.append(2000)
         else:  # Company size based on valuation
             valuation = filtered_data.loc[filtered_data["Company"] == node, "Valuation_Billions"].values
-            node_sizes.append(valuation[0] * 100 if len(valuation) > 0 else 100)
+            node_sizes.append(valuation[0] * 50 if len(valuation) > 0 else 100)
 
     # Node colors
     node_colors = []
@@ -97,11 +97,19 @@ def generate_graph(selected_investors, filtered_data):
         with_labels=True,
         node_size=node_sizes,
         node_color=node_colors,
-        font_size=12,
+        alpha=0.8,  # Slight transparency for Tufte-inspired visuals
+        font_size=10,
         font_weight="bold",
-        edge_color="gray",
-        width=1.5
+        edge_color="#B0BEC5",  # Neutral, muted edge color
+        width=0.8  # Thin edges for minimal visual clutter
     )
+
+    # Add a legend for node size (valuation)
+    min_size, max_size = 50, 5000  # Example scale
+    for size, label in zip([min_size, max_size], ["$1B", "$100B"]):
+        plt.scatter([], [], s=size, color="#33FF57", label=f"{label} valuation")
+    plt.legend(scatterpoints=1, frameon=False, labelspacing=1.5, loc="lower left", fontsize=12)
+
     plt.title("Venture Funded Companies Visualization", fontsize=20)
     plt.axis('off')