Update app.py

app.py

@@ -532,14 +532,20 @@ if uploaded_file is not None:
     
     fig4 = plt.figure(figsize=(10, 10))
     pos = nx.spring_layout(G)
+    
+    # Calculate edge widths properly
+    edge_weights = [G[u][v]['weight'] for u,v in G.edges()]
+    max_weight = max(edge_weights) if edge_weights else 1
+    
     nx.draw(G, pos, with_labels=True, 
             node_color='lightblue',
             node_size=1000,
             font_size=12,
-            width=[G[u][v]['weight']/max(dict(G.edges).values())*5 for u,v in G.edges()])
+            width=[G[u][v]['weight']/max_weight * 5 for u,v in G.edges()])
     plt.title("Top Character Connections")
     st.pyplot(fig4)
 
+
     # 5. Line Length Distribution
     fig5 = plt.figure(figsize=(10, 6))
     sns.histplot(line_lengths)