Update app.py

app.py

@@ -570,12 +570,13 @@ def create_avatar():
        draw.line([(225 + i*3, 790), (223 + i*3, 800)], fill=(0, 255, 255, 200), width=2)
        draw.line([(365 + i*3, 790), (363 + i*3, 800)], fill=(0, 255, 255, 200), width=2)
   
-# Neural network lines
-for _ in range(100):
-    start = (np.random.randint(0, AVATAR_WIDTH), np.random.randint(0, AVATAR_HEIGHT))
-    end = (np.random.randint(0, AVATAR_WIDTH), np.random.randint(0, AVATAR_HEIGHT))
-    draw.line([start, end], fill=(0, 255, 255, 50), width=1)
-return img
+def generate_neural_network_lines(img, draw):
+    # Neural network lines
+    for _ in range(100):
+        start = (np.random.randint(0, AVATAR_WIDTH), np.random.randint(0, AVATAR_HEIGHT))
+        end = (np.random.randint(0, AVATAR_WIDTH), np.random.randint(0, AVATAR_HEIGHT))
+        draw.line([start, end], fill=(0, 255, 255, 50), width=1)
+    return img
     
 
 # Create and display avatar with heatmap