Update app.py

app.py

@@ -2,7 +2,6 @@ import streamlit as st
 import matplotlib.pyplot as plt
 import networkx as nx
 import bz2
-import numpy as np
 
 # Sidebar for selecting an option
 sidebar_option = st.sidebar.radio("Select an option", 
@@ -12,13 +11,19 @@ sidebar_option = st.sidebar.radio("Select an option",
                                   "Drawing: Chess Masters"])
 
 # Helper function to draw and display graph
-def draw_graph(G, pos=None, title="Graph Visualization"):
+def draw_graph(G, pos=None, title="Graph Visualization", edgewidth=None, nodesize=None):
+    if edgewidth is None:
+        edgewidth = [1] * len(G.edges())  # Default edge width if not provided
+
+    if nodesize is None:
+        nodesize = [300] * len(G.nodes())  # Default node size if not provided
+
     plt.figure(figsize=(12, 12))
     nx.draw_networkx_edges(G, pos, alpha=0.3, width=edgewidth, edge_color="m")
     nx.draw_networkx_nodes(G, pos, node_size=nodesize, node_color="#210070", alpha=0.9)
     label_options = {"ec": "k", "fc": "white", "alpha": 0.7}
     nx.draw_networkx_labels(G, pos, font_size=14, bbox=label_options)
-    
+
     # Title/legend
     font = {"fontname": "Helvetica", "color": "k", "fontweight": "bold", "fontsize": 14}
     ax = plt.gca()
@@ -128,7 +133,7 @@ def display_chess_masters_graph():
         pos["Smyslov, Vassily V"] += (0.05, -0.03)
 
         # Draw the graph
-        draw_graph(H, pos, title="World Chess Championship Games: 1886 - 1985")
+        draw_graph(H, pos, title="World Chess Championship Games: 1886 - 1985", edgewidth=edgewidth, nodesize=nodesize)
 
     elif option == "Create your own":
         uploaded_file = st.file_uploader("Upload your own PGN file", type="pgn")
@@ -149,7 +154,7 @@ def display_chess_masters_graph():
                     wins[v] += 1.0
             nodesize = [wins[v] * 50 for v in H_custom]
             pos_custom = nx.kamada_kawai_layout(H_custom)
-            draw_graph(H_custom, pos_custom, title="Custom Chess Game Graph")
+            draw_graph(H_custom, pos_custom, title="Custom Chess Game Graph", edgewidth=edgewidth, nodesize=nodesize)
 
 # Display other sections
 def display_basic_properties():