Update app.py

app.py

@@ -50,11 +50,12 @@ def process_frame(frame, bg_type, bg, fast_mode, bg_frame_index, background_fram
         return frame, bg_frame_index
 
 @spaces.GPU
-def fn(vid, bg_type="Color", bg_image=None, bg_video=None, color="#00FF00", video_handling="slow_down"):
+def fn(vid, bg_type="Color", bg_image=None, bg_video=None, color="#00FF00", fps=0, video_handling="slow_down", fast_mode=True, max_workers=10):
     try:
         start_time = time.time()  # Start the timer
         video = VideoFileClip(vid)
-        fps = video.fps  # Use original fps
+        if fps == 0:
+            fps = video.fps
         
         audio = video.audio
         frames = list(video.iter_frames(fps=fps))
@@ -75,9 +76,9 @@ def fn(vid, bg_type="Color", bg_image=None, bg_video=None, color="#00FF00", vide
         
         bg_frame_index = 0  # Initialize background frame index
 
-        with ThreadPoolExecutor(max_workers=10) as executor:  # Fixed max_workers
+        with ThreadPoolExecutor(max_workers=max_workers) as executor:
             # Pass bg_frame_index as part of the function arguments
-            futures = [executor.submit(process_frame, frames[i], bg_type, bg_image, True, bg_frame_index + i, background_frames, color) for i in range(len(frames))]  # Fixed fast_mode to True
+            futures = [executor.submit(process_frame, frames[i], bg_type, bg_image, fast_mode, bg_frame_index + i, background_frames, color) for i in range(len(frames))] 
             for i, future in enumerate(futures):
                 result, _ = future.result() #  No need to update bg_frame_index here
                 processed_frames.append(result)
@@ -101,7 +102,7 @@ def fn(vid, bg_type="Color", bg_image=None, bg_video=None, color="#00FF00", vide
         yield gr.update(visible=False), gr.update(visible=True)
         yield None, f"Error processing video: {e}"
 
-def process(image, bg, fast_mode=True):  # Default to fast_mode=True
+def process(image, bg, fast_mode=False):
     image_size = image.size
     input_images = transform_image(image).unsqueeze(0).to(device)
     model = birefnet_lite if fast_mode else birefnet
@@ -123,37 +124,72 @@ def process(image, bg, fast_mode=True):  # Default to fast_mode=True
     image = Image.composite(image, background, mask)
     return image
 
-with gr.Blocks(theme=gr.themes.Ocean()) as demo:
+# Create custom dark theme
+dark_theme = gr.themes.Soft(
+    primary_hue="purple",
+    secondary_hue="blue",
+    neutral_hue="slate",
+).set(
+    body_background_fill_dark="#0f0f23",
+    block_background_fill_dark="#1a1b2e",
+    block_border_color_dark="#16213e",
+    input_background_fill_dark="#16213e",
+    button_primary_background_fill_dark="#6366f1",
+    button_primary_background_fill_hover_dark="#4f46e5",
+    button_secondary_background_fill_dark="#374151",
+    button_secondary_background_fill_hover_dark="#4b5563"
+)
+
+with gr.Blocks(theme=dark_theme) as demo:
     
     with gr.Row():
         in_video = gr.Video(label="Input Video", interactive=True)
         stream_image = gr.Image(label="Streaming Output", visible=False)
         out_video = gr.Video(label="Final Output Video")
     
-    # Settings panels aligned below input video with limited width
-    with gr.Row():
-        with gr.Column(scale=1):
-            # Background Type selection
+    # Settings panels aligned below input video
+    with gr.Column():
+        # Row 1: Background Type and FPS (two smaller panels)
+        with gr.Row():
             bg_type = gr.Radio(["Color", "Image", "Video"], label="Background Type", value="Color", interactive=True)
-            
-            # Dynamic background options
-            color_picker = gr.ColorPicker(label="Background Color", value="#00FF00", visible=True, interactive=True)
-            bg_image = gr.Image(label="Background Image", type="filepath", visible=False, interactive=True)
-            bg_video = gr.Video(label="Background Video", visible=False, interactive=True)
-            
-            # Video handling options (only visible when Video is selected)
-            with gr.Column(visible=False) as video_handling_options:
-                video_handling_radio = gr.Radio(["slow_down", "loop"], label="Video Handling", value="slow_down", interactive=True)
-            
-            # Styled button with less rounded corners
-            submit_button = gr.Button(
-                "🚀 Change Background", 
-                interactive=True, 
-                variant="primary",
-                size="lg"
+            fps_slider = gr.Slider(
+                minimum=0,
+                maximum=60,
+                step=1,
+                value=0,
+                label="Output FPS (0 = original fps)",
+                interactive=True
+            )
+        
+        # Row 2: Dynamic background options (full width)
+        color_picker = gr.ColorPicker(label="Background Color", value="#00FF00", visible=True, interactive=True)
+        bg_image = gr.Image(label="Background Image", type="filepath", visible=False, interactive=True)
+        bg_video = gr.Video(label="Background Video", visible=False, interactive=True)
+        
+        # Row 3: Video handling options (only visible when Video is selected)
+        with gr.Row(visible=False) as video_handling_options:
+            video_handling_radio = gr.Radio(["slow_down", "loop"], label="Video Handling", value="slow_down", interactive=True)
+        
+        # Row 4: Processing options (two smaller panels)
+        with gr.Row():
+            fast_mode_checkbox = gr.Checkbox(label="Fast Mode (Use BiRefNet_lite)", value=True, interactive=True)
+            max_workers_slider = gr.Slider(
+                minimum=1, 
+                maximum=32, 
+                step=1, 
+                value=10, 
+                label="Max Workers", 
+                info="Parallel processing threads", 
+                interactive=True
             )
-        with gr.Column(scale=1):
-            pass  # Empty column to balance layout
+    
+    # Styled button with rocket emoji and rounded corners
+    submit_button = gr.Button(
+        "🚀 Change Background", 
+        interactive=True, 
+        variant="primary",
+        size="lg"
+    )
 
     def update_visibility(bg_type):
         if bg_type == "Color":