Update app.py

app.py

@@ -1,5 +1,3 @@
-# venture_networks_gradio.py
-
 import networkx as nx
 import matplotlib.pyplot as plt
 import gradio as gr
@@ -10,27 +8,7 @@ from PIL import Image
 investors = {
     "Accel": ["Meta", "Dropbox", "Spotify", "Adroll", "PackLink", "Zoom", "Slack"],
     "Andreessen Horowitz": [
-        "Airbnb", "Lyft", "Pinterest", "Accolade", "Actifio", "Affirm", "Amify",
-        "Apptio", "Arimo", "Asana", "Asserts", "Authy", "Automatic", "Barefoot Networks",
-        "Bebop", "BeReal", "Blockspring", "Boku", "Box", "Bracket Computing", "Bump",
-        "BuzzFeed", "Capital", "Cardiogram", "Caviar", "Citiizen", "CipherCloud",
-        "ClearStory", "Coinbase", "Cumulus", "DeepMap", "Delphix", "DigitalOcean",
-        "Dollar Shave Club", "Earn", "EQRx", "Erasca", "Meta", "FiftyThree", "Flutter",
-        "Fusion-io", "Gainspeed", "Genius", "GitHub", "Gobble", "Golden", "Greenlight",
-        "Granular", "Greplify", "Groupon", "Hatch", "iCracked", "Imgur", "Instacart",
-        "Instagram", "Intrinsic", "Isovalent", "Jopwell", "Julep", "Jungla", "Keybase",
-        "Kno", "Knowable", "Knowmia", "Loom", "Lowkey", "Luma", "Lyrebird", "Lytro",
-        "Massive", "Morta Security", "Nautilus Biotechnology", "Nicira", "Oculus", "Okta",
-        "Onshape", "Opendoor", "OpenInvest", "Optimizely", "Optimyze", "PagerDuty",
-        "Parsec", "Patch Biosciences", "PatientPing", "PicCloud", "Pinterest", "Pixate",
-        "Platfora", "Product Hunt", "Quantopian", "RHL Ventures", "Reddit", "Reflektive",
-        "Rigetti", "Rival", "Robinhood", "Roblox", "Rockmelt", "RTFKT", "Samsara", "Seed",
-        "Shapeways", "ShoeDazzle", "SignalFx", "SigOpt", "Silver Tail Systems", "Skout",
-        "Skype", "Slack", "Socialcam", "Soylent", "Stack Overflow", "Standard Treasury",
-        "Swell", "Swiftype", "Tabular", "Talaria", "Tasty Labs", "Teleport", "Tenfold",
-        "Tidemark", "Tilt", "TinyCo", "Tmunity", "Tomfoolery", "Wise", "TXN", "Urban Engines",
-        "Usermind", "Viki", "Walker and Company", "Wit.ai", "Within", "Zynga", "Altitude Learning",
-        "Bromium", "Factual", "Instart", "Proven", "Rabbit", "UnifyID"
+        "Airbnb", "Lyft", "Pinterest", "Coinbase", "Robinhood", "Slack"
     ],
     "Google Ventures": ["Uber", "LendingClub"],
     "Greylock": ["Workday", "Palo Alto Networks"],
@@ -40,43 +18,63 @@ investors = {
     "Emergence Capital Partners": ["Zoom", "Box", "Salesforce"],
     "Trinity Ventures": ["New Relic", "Care.com", "TubeMogul"],
     "Citi Ventures": ["Square", "Nutanix"],
-    "Sequoia": [
-        "Alphabet (Google)", "NVIDIA", "Dropbox", "Airbnb", "Affirm",
-        "Airbnb", "Eventbrite", "Meta", "Lyft", "Opendoor", "Palantir", "Reddit", "Uber",
-        "23ANDME", "AMPLITUDE", "APPLE", "ATARI", "AURORA", "BARRACUDA", "BIRD", "BLOCK",
-        "BRIDGEBIO", "CISCO", "CONFLUENT", "DOORDASH", "ELECTRONIC ARTS", "FIREEYE", "GOOGLE",
-        "GREEN DOT", "GUARDANT HEALTH", "HUBSPOT", "INSTACART", "LINKEDIN", "MEDALLIA", "MONGODB",
-        "NATERA", "NETAPP", "NIMBLE STORAGE", "NUBANK", "OKTA", "ORACLE", "PALO ALTO NETWORKS",
-        "PAYPAL", "QUALTRICS", "QUANTENNA", "RINGCENTRAL", "ROBINHOOD", "RUCKUS NETWORKS",
-        "SERVICENOW", "SNOWFLAKE", "SUMO LOGIC", "SUNRUN", "TRULIA", "UIPATH", "UNITY", "YAHOO"
-    ],
-    "Y Combinator": ["Dropbox", "Airbnb", "Coinbase", "DoorDash", "Reddit", "Ginkgo Bioworks", "GitLab", "Instacart"]
+    "Sequoia": ["Alphabet (Google)", "NVIDIA", "Dropbox", "Airbnb", "Meta"],
+    "Y Combinator": ["Dropbox", "Airbnb", "Coinbase", "DoorDash", "Reddit"]
+}
+
+# Example market capitalization values (in billions USD)
+market_cap = {
+    "Meta": 900,
+    "Dropbox": 10,
+    "Spotify": 30,
+    "Zoom": 20,
+    "Slack": 27,
+    "Airbnb": 100,
+    "Lyft": 4,
+    "Pinterest": 14,
+    "Coinbase": 70,
+    "Robinhood": 10,
+    "Uber": 60,
+    "LendingClub": 1,
+    "Snap": 18,
+    "Grubhub": 6,
+    "AppDynamics": 1,
+    "WeWork": 0.9,
+    "Opendoor": 3,
+    "Box": 4,
+    "Salesforce": 200,
+    "Square": 90,
+    "Nutanix": 10,
+    "Alphabet (Google)": 1500,
+    "NVIDIA": 1200
 }
 
+# Assign default size for missing companies
+default_size = 5
+
 # Define a color map for the investors
 investor_colors = {
-    "Accel": "lightblue",
-    "Andreessen Horowitz": "lightgreen",
-    "Google Ventures": "lightred",
-    "Greylock": "purple",
-    "Lightspeed Venture Partners": "orange",
-    "Benchmark": "lightbrown",
-    "Norwest Venture Partners": "pink",
-    "Emergence Capital Partners": "yellow",
-    "Trinity Ventures": "cyan",
-    "Citi Ventures": "magenta",
-    "Sequoia": "lime",
-    "Y Combinator": "black"
+    "Accel": "#1f77b4",
+    "Andreessen Horowitz": "#ff7f0e",
+    "Google Ventures": "#2ca02c",
+    "Greylock": "#d62728",
+    "Lightspeed Venture Partners": "#9467bd",
+    "Benchmark": "#8c564b",
+    "Norwest Venture Partners": "#e377c2",
+    "Emergence Capital Partners": "#7f7f7f",
+    "Trinity Ventures": "#bcbd22",
+    "Citi Ventures": "#17becf",
+    "Sequoia": "#1b9e77",
+    "Y Combinator": "#d95f02"
 }
 
-
 def generate_graph(selected_investors):
     if not selected_investors:
         selected_investors = list(investors.keys())
 
     G = nx.Graph()
 
-    # Add edges based on selected investors
+    # Add edges and nodes based on selected investors
     for investor in selected_investors:
         companies = investors[investor]
         for company in companies:
@@ -85,13 +83,16 @@ def generate_graph(selected_investors):
     # Get edge colors
     edge_colors = [G[u][v]['color'] for u, v in G.edges]
 
-    # Set node colors: investors have their specific color, companies have a default color
+    # Set node colors and sizes
     node_colors = []
+    node_sizes = []
     for node in G.nodes:
         if node in investor_colors:
             node_colors.append(investor_colors[node])
+            node_sizes.append(2000)  # Fixed size for investors
         else:
             node_colors.append("#F0E68C")  # Khaki for companies
+            node_sizes.append(market_cap.get(node, default_size) * 100)  # Scale up sizes
 
     # Create plot
     plt.figure(figsize=(18, 18))
@@ -99,7 +100,7 @@ def generate_graph(selected_investors):
     nx.draw(
         G, pos,
         with_labels=True,
-        node_size=3000,
+        node_size=node_sizes,
         node_color=node_colors,
         font_size=10,
         font_weight="bold",
@@ -133,7 +134,7 @@ def main():
         ),
         outputs=gr.Image(type="pil", label="Venture Network Graph"),
         title="Venture Networks Visualization",
-        description="Select investors to visualize their investments in various companies. The graph shows connections between investors and the companies they've invested in.",
+        description="Select investors to visualize their investments in various companies. The graph shows connections between investors and the companies they've invested in. Node sizes represent market capitalization.",
         flagging_mode="never"
     )