edited

app.py

@@ -1,22 +1,21 @@
+
 import streamlit as st
 from PIL import Image
 import numpy as np
 import cv2
 import svgwrite
 import io
-import os
 
-st.set_page_config(page_title="Image to Embroidery", layout="centered")
-st.title("🧵 Image to Embroidery Converter")
-st.write("Upload an image and get both an SVG embroidery preview and a short video!")
+st.title("Image to Embroidery Converter")
+st.write("Upload an image and get a stitch-style SVG preview.")
 
-uploaded_file = st.file_uploader("📷 Upload an image", type=["jpg", "jpeg", "png"])
+uploaded_file = st.file_uploader("Choose an image", type=["jpg", "jpeg", "png"])
 
-# === Stitch Preview Function ===
 def process_image(uploaded_image):
     img = Image.open(uploaded_image).convert("L")
     img = img.resize((300, 300))
     img_np = np.array(img)
+
     edges = cv2.Canny(img_np, threshold1=100, threshold2=200)
 
     svg = svgwrite.Drawing(size=(img.width, img.height))
@@ -29,33 +28,10 @@ def process_image(uploaded_image):
     svg.write(svg_buffer)
     return svg_buffer.getvalue()
 
-# === Video Generator Function ===
-def create_video_from_image(image_file, duration=3, fps=10):
-    img = Image.open(image_file).convert("RGB")
-    img = img.resize((600, 600))
-    frame = np.array(img)
-    height, width, _ = frame.shape
-
-    output_path = "embroidery_preview.mp4"
-    total_frames = duration * fps
-    out = cv2.VideoWriter(output_path, cv2.VideoWriter_fourcc(*'mp4v'), fps, (width, height))
-
-    for _ in range(total_frames):
-        out.write(frame)
-
-    out.release()
-    return output_path
-
-# === App Logic ===
 if uploaded_file:
-    st.image(uploaded_file, caption="🖼️ Uploaded Image", use_column_width=True)
-
-    # SVG generation
+    st.image(uploaded_file, caption="Uploaded Image", use_column_width=True)
     svg_data = process_image(uploaded_file)
-    st.download_button("⬇️ Download SVG", data=svg_data, file_name="embroidery_preview.svg", mime="image/svg+xml")
-    st.code(svg_data, language="xml")
-
-    # Video generation
-    video_path = create_video_from_image(uploaded_file)
-    with open(video_path, "rb") as video_file:
-        st.download_button("🎥 Download MP4 Video", data=video_file, file_name="embroidery_preview.mp4", mime="video/mp4")
+    
+    st.download_button("Download SVG", data=svg_data, file_name="embroidery_preview.svg", mime="image/svg+xml")
+    st.markdown("### Embroidery Stitch Preview (SVG)")
+    st.code(svg_data, language="xml")