Update app.py

app.py

@@ -63,7 +63,7 @@ def setup_environment():
 
 setup_environment()
 
-@spaces.GPU(requires=True)
+@spaces.GPU(required=True)
 def interpolate_video(input_video_path, multi_factor):
     if input_video_path is None: return None
     factor = str(multi_factor).replace("x", "").strip()
@@ -89,10 +89,20 @@ def interpolate_video(input_video_path, multi_factor):
         os.chdir(BASE_DIR)
 
 # --- Stitching Logic ---
-def stitch_videos(video_files):
+def stitch_videos(video_files, resolution_choice):
     if not video_files: return None
     
-    print(f"Stitching {len(video_files)} videos...")
+    # Parse resolution (e.g., "1024x1024" -> w=1024, h=1024)
+    try:
+        target_w, target_h = resolution_choice.split("x")
+        target_w = target_w.strip()
+        target_h = target_h.strip()
+    except:
+        # Fallback default
+        target_w, target_h = "1920", "1080"
+
+    print(f"Stitching {len(video_files)} videos into {target_w}x{target_h}...")
+    
     stitch_list_path = os.path.join(BASE_DIR, "stitch_list.txt")
     output_stitched = os.path.join(BASE_DIR, "final_stitched.mp4")
     temp_dir = os.path.join(BASE_DIR, "temp_stitch")
@@ -100,18 +110,21 @@ def stitch_videos(video_files):
     if os.path.exists(temp_dir): shutil.rmtree(temp_dir)
     os.makedirs(temp_dir)
 
-    # 1. Normalize Step: Convert all inputs to same Codec, FPS (60), and Scale
-    # This prevents the "glitch" when combining different framerates
+    # 1. Normalize Step: Convert all inputs to same Codec, FPS (60), and User-Selected Scale
     normalized_files = []
     for i, vid_path in enumerate(video_files):
         temp_filepath = os.path.join(temp_dir, f"norm_{i}.mp4")
         print(f"Normalizing clip {i+1}...")
         
-        # We force 60fps (-r 60) and 1080p scale to ensure they match
+        # FFmpeg Filter Breakdown:
+        # scale=W:H:force_original_aspect_ratio=decrease -> Resize video to FIT inside the box
+        # pad=W:H:(ow-iw)/2:(oh-ih)/2 -> Create the box and center the video (adding black bars)
+        scale_filter = f"scale={target_w}:{target_h}:force_original_aspect_ratio=decrease,pad={target_w}:{target_h}:(ow-iw)/2:(oh-ih)/2"
+        
         cmd = [
             'ffmpeg', '-i', vid_path,
-            '-r', '60',                 # Force Framerate
-            '-vf', 'scale=1920:1080:force_original_aspect_ratio=decrease,pad=1920:1080:(ow-iw)/2:(oh-ih)/2', # Fit to 1080p box
+            '-r', '60',                 # Force Framerate to 60
+            '-vf', scale_filter,        # Apply Dynamic Scaling
             '-c:v', 'libx264', '-crf', '23',
             '-c:a', 'aac',              # Force Audio Codec
             '-ar', '44100',             # Force Audio Rate
@@ -152,14 +165,23 @@ with gr.Blocks() as demo:
 
         # TAB 2: Stitch
         with gr.TabItem("2. Stitch Videos"):
-            gr.Markdown("Upload multiple videos. They will be **normalized to 60fps 1080p** and joined together.")
+            gr.Markdown("Upload multiple videos. They will be normalized to **60fps** and your selected resolution.")
             with gr.Row():
                 with gr.Column():
-                    # file_count="multiple" allows selecting multiple files
                     stitch_inputs = gr.File(label="Upload Clips", file_count="multiple")
+                    
+                    # New Resolution Dropdown
+                    res_select = gr.Dropdown(
+                        choices=["1920x1080", "1280x1280", "1024x1024"], 
+                        value="1920x1080", 
+                        label="Output Resolution"
+                    )
+                    
                     stitch_btn = gr.Button("Stitch Videos", variant="primary")
                 with gr.Column():
                     stitch_output = gr.Video(label="Stitched Result")
-            stitch_btn.click(stitch_videos, inputs=stitch_inputs, outputs=stitch_output)
+            
+            # Updated click event passing the resolution choice
+            stitch_btn.click(stitch_videos, inputs=[stitch_inputs, res_select], outputs=stitch_output)
 
 demo.launch()