Update app.py

app.py

@@ -64,7 +64,7 @@ def create_video_from_frames(frame_files, output_path, duration_seconds, width,
 
     video.release()
 
-def process_svg_to_video(input_svg_path, original_width, original_height, video_duration_seconds=10, chunk_size=None):
+def process_svg_to_video(input_svg_path, original_width, original_height, video_duration_seconds=10, chunk_size=30):
     """Process SVG file and create a video with specified duration using exact row slicing"""
     # Read SVG file as a table to maintain exact row slicing logic
     df = pd.read_table(input_svg_path, header=None)
@@ -75,10 +75,12 @@ def process_svg_to_video(input_svg_path, original_width, original_height, video_
     # Use the original image dimensions
     width, height = original_width, original_height
     
-    # Calculate chunk size if not specified
+    # If chunk_size is 0, use automatic calculation (total_rows // 30)
     total_rows = len(df_middle)
-    if chunk_size is None:
-        chunk_size = max(1, total_rows // 30)  # Default to ~30 steps
+    if chunk_size == 0:
+        chunk_size = max(1, total_rows // 30)
+    else:
+        chunk_size = max(1, min(chunk_size, total_rows))  # Ensure it's within valid range
     
     # Create a temporary directory for images
     temp_dir = tempfile.mkdtemp()
@@ -113,6 +115,7 @@ def process_svg_to_video(input_svg_path, original_width, original_height, video_
 def convert_to_vector_and_video(
     image, 
     video_duration=10,
+    chunk_size=30,
     colormode="color", 
     hierarchical="stacked", 
     mode="spline", 
@@ -158,7 +161,8 @@ def convert_to_vector_and_video(
         output_svg_path,
         original_width,
         original_height,
-        video_duration_seconds=video_duration
+        video_duration_seconds=video_duration,
+        chunk_size=chunk_size
     )
     
     # Read the SVG output
@@ -179,6 +183,7 @@ def clear_inputs():
     return (
         gr.Image(value=None), 
         gr.Slider(value=10),
+        gr.Slider(value=30),
         gr.Radio(value="color"), 
         gr.Radio(value="stacked"), 
         gr.Radio(value="spline"), 
@@ -243,6 +248,8 @@ with gr.Blocks(css=css) as app:
             with gr.Column():
                 image_input = gr.Image(type="pil", label="Upload Image")
                 video_duration = gr.Slider(1, 60, value=10, step=1, label="Video Duration (seconds)")
+                chunk_size = gr.Slider(0, 1000, value=30, step=1, label="Chunk Size (0=auto)", 
+                                     info="Number of SVG path elements to add per frame (0 for automatic calculation)")
                 
                 with gr.Accordion("Advanced Settings", open=False):
                     with gr.Accordion("Clustering", open=False):
@@ -300,6 +307,7 @@ with gr.Blocks(css=css) as app:
         outputs=[
             image_input,
             video_duration,
+            chunk_size,
             colormode,
             hierarchical,
             mode,
@@ -319,6 +327,7 @@ with gr.Blocks(css=css) as app:
         inputs=[
             image_input,
             video_duration,
+            chunk_size,
             colormode,
             hierarchical,
             mode,