Update app.py

app.py

@@ -106,17 +106,63 @@ def filter_investors_by_country_and_industry(selected_country, selected_industry
     
     return list(investor_valuations.keys()), filtered_data
 
+# Function to generate the Plotly graph
+def generate_graph(investor_list, filtered_data):
+    if not investor_list:
+        logger.warning("No investors selected. Returning empty figure.")
+        return go.Figure()
+
+    G = nx.Graph()
+    for investor in investor_list:
+        companies = filtered_data[filtered_data["Select_Investors"].str.contains(investor, na=False)]["Company"].tolist()
+        for company in companies:
+            G.add_edge(investor, company)
+
+    pos = nx.spring_layout(G, k=0.2, seed=42)
+
+    # Create Plotly traces for edges and nodes
+    edge_trace = go.Scatter(
+        x=[],
+        y=[],
+        line=dict(width=0.5, color='#888'),
+        hoverinfo='none',
+        mode='lines'
+    )
+
+    for edge in G.edges():
+        x0, y0 = pos[edge[0]]
+        x1, y1 = pos[edge[1]]
+        edge_trace['x'] += [x0, x1, None]
+        edge_trace['y'] += [y0, y1, None]
+
+    node_trace = go.Scatter(
+        x=[],
+        y=[],
+        text=[],
+        mode='markers',
+        hoverinfo='text',
+        marker=dict(
+            showscale=True,
+            colorscale='YlGnBu',
+            size=10,
+            colorbar=dict(thickness=15, title='Node Valuation')
+        )
+    )
+
+    for node in G.nodes():
+        x, y = pos[node]
+        node_trace['x'] += [x]
+        node_trace['y'] += [y]
+        node_trace['text'] += [f"{node}"]
+
+    fig = go.Figure(data=[edge_trace, node_trace])
+    return fig
+
 # Gradio app function
 def app(selected_country, selected_industry, valuation_threshold):
     investor_list, filtered_data = filter_investors_by_country_and_industry(selected_country, selected_industry, valuation_threshold)
-    
-    if not investor_list:
-        return (
-            "No investors meet the selected criteria. Try reducing the valuation threshold or selecting different filters.",
-            None
-        )
-    
-    return investor_list, generate_graph(investor_list, filtered_data)
+    graph = generate_graph(investor_list, filtered_data)
+    return investor_list, graph
 
 # Gradio Interface
 def main():
@@ -132,8 +178,8 @@ def main():
             industry_filter = gr.Dropdown(choices=industry_list, label="Filter by Industry", value="All")
             valuation_threshold = gr.Slider(minimum=0, maximum=50, step=1, value=20, label="Valuation Threshold (in B)")
 
-        graph_output = gr.Plot(label="Venture Network Graph")
         investor_output = gr.Text(label="Investor Results")
+        graph_output = gr.Plot(label="Venture Network Graph")
         
         country_filter.change(
             app,