Update app.py

app.py

@@ -4,6 +4,8 @@ from skimage.color import rgb2gray
 from skimage.transform import resize
 from skimage.draw import line
 import gradio as gr
+import tempfile
+import os
 
 def preprocess_image(image, size=(200, 200)):
     if image.ndim == 3:
@@ -28,7 +30,7 @@ def line_darkness_score(img, pin1, pin2):
     rr = np.clip(rr, 0, h-1)
     cc = np.clip(cc, 0, w-1)
     line_pixels = img[rr, cc]
-    score = np.sum(1 - line_pixels)  # invert because black is 0
+    score = np.sum(1 - line_pixels)  # black = 0, so 1 - pixel for darkness
     return score
 
 def generate_connections(img, pins, num_connections):
@@ -61,22 +63,23 @@ def draw_string_art(image, num_pins, num_connections):
     connections = generate_connections(img, pins, num_connections)
     # Draw the string art
     fig, ax = plt.subplots(figsize=(8, 8))
-    ax.set_facecolor("black")
+    ax.set_facecolor("white")  # White background for visibility
     ax.axis("off")
     for c in connections:
         p1, p2 = pins[c[0]], pins[c[1]]
-        ax.plot([p1[0], p2[0]], [p1[1], p2[1]], 'w-', linewidth=0.3)
-    ax.scatter(pins[:,0], pins[:,1], c='red', s=20)
+        ax.plot([p1[0], p2[0]], [p1[1], p2[1]], 'k-', linewidth=0.7, alpha=0.9)  # Black, thicker lines
+    ax.scatter(pins[:,0], pins[:,1], c='red', s=16)  # Red pins, moderate size
+    plt.tight_layout(pad=0)
     plt.close(fig)
     # Prepare connections file (1-based indexing)
-    connections_1_based = [(c[0]+1, c[1]+1) for c in connections]
-    flat = [str(pin) for conn in connections_1_based for pin in conn]
-    connections_str = ', '.join(flat)
-    # Save to file
-    filename = "string_art_connections.txt"
-    with open(filename, "w") as f:
-        f.write(connections_str)
-    return fig, filename
+    connections_1_based = [str(c[0]+1) for c in connections]
+    # Make continuous: start at pin 1, then next, etc.
+    connections_str = ', '.join(connections_1_based)
+    # Save to temporary file
+    temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".txt", mode="w", encoding="utf-8")
+    temp_file.write(connections_str)
+    temp_file.close()
+    return fig, temp_file.name
 
 def interface(image, num_pins, num_connections):
     fig, filename = draw_string_art(image, num_pins, num_connections)
@@ -94,7 +97,7 @@ demo = gr.Interface(
         gr.File(label="Connections File")
     ],
     title="String Art Generator",
-    description="Upload an image, set number of pins and connections, and generate circular string art with downloadable connection file."
+    description="Upload an image, set the number of pins and connections, and generate circular string art with a downloadable connection file."
 )
 
 if __name__ == "__main__":