Update app.py

app.py

@@ -6,6 +6,10 @@ import random
 
 # THIS IS THE FIX - PART 1
 os.environ['GRADIO_SUPPRESS_PROGRESS'] = 'true'
+# THIS IS THE FIX - PART 2: Clean up console logs from Gradio
+import logging
+logging.getLogger('gradio').setLevel(logging.ERROR)
+
 
 import cv2
 import numpy as np
@@ -54,6 +58,10 @@ os.makedirs(TEMP_DIR, exist_ok=True)
 model_load_lock = Lock()
 loaded_detectors = {}
 whisper_model = None
+# --- ACTION REQUIRED ---
+# Please download a sound file, name it 'finish_sound.mp3', 
+# and place it in the same directory as this app.py file.
+# A good one can be found here: https://cdn.pixabay.com/download/audio/2022/03/15/audio_2289a72212.mp3?filename=whistle-108713.mp3
 
 # --- Default Presets for Transfer Tab (Flat Dictionary) ---
 DEFAULT_LINK_PRESETS = {
@@ -1367,6 +1375,144 @@ def create_rhythmic_animatic(timeline_data, audio_path, measure_choice, out_w, o
 
 # --- NEW CREATIVE FUNCTIONS ---
 
+def _create_auto_trailer_impl(video_path, trailer_duration, clip_duration, analysis_method, transition_style, music_path, out_w, out_h, progress: Progress):
+    """Internal implementation of the auto-trailer creator."""
+    if not video_path: raise gr.Error("Please upload a source video.")
+    
+    source_duration = get_media_duration(video_path)
+    
+    if source_duration < trailer_duration:
+        gr.Warning(f"Source video is only {source_duration:.1f}s long. The trailer duration will be capped at the source video length.")
+        trailer_duration = source_duration
+
+    if clip_duration > trailer_duration:
+        new_clip_duration = trailer_duration / 2 if trailer_duration > 2 else trailer_duration
+        gr.Warning(f"Clip duration ({clip_duration}s) is longer than the trailer duration ({trailer_duration:.1f}s). Adjusting clip duration to {new_clip_duration:.1f}s.")
+        clip_duration = new_clip_duration
+
+
+    timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+    job_temp_dir = os.path.join(TEMP_DIR, f"trailer_{timestamp}")
+    os.makedirs(job_temp_dir, exist_ok=True)
+    
+    try:
+        # --- 1. Analyze Video to Find Interesting Clips ---
+        progress(0, desc="Analyzing video for high-motion scenes...")
+        
+        cap = cv2.VideoCapture(video_path)
+        fps = cap.get(cv2.CAP_PROP_FPS)
+        if fps == 0: fps = 30 # fallback
+        
+        chunk_duration_frames = int(clip_duration * fps)
+        video_total_frames = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
+        
+        chunk_scores = []
+        prev_frame = None
+        
+        frame_num = 0
+        while cap.isOpened():
+            ret, frame = cap.read()
+            if not ret: break
+
+            frame_skip = max(1, int(fps / 5)) # Analyze ~5 frames per second
+            if frame_num % frame_skip == 0:
+                gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
+                gray = cv2.GaussianBlur(gray, (21, 21), 0)
+
+                if prev_frame is not None:
+                    frame_delta = cv2.absdiff(prev_frame, gray)
+                    thresh = cv2.threshold(frame_delta, 25, 255, cv2.THRESH_BINARY)[1]
+                    motion_score = np.sum(thresh)
+                    
+                    chunk_index = frame_num // chunk_duration_frames
+                    while len(chunk_scores) <= chunk_index:
+                        chunk_scores.append({'start_time': (len(chunk_scores) * clip_duration), 'scores': []})
+                    chunk_scores[chunk_index]['scores'].append(motion_score)
+                
+                prev_frame = gray
+            
+            frame_num += 1
+            if frame_num % 100 == 0:
+                progress(0.2 * (frame_num / video_total_frames), desc=f"Analyzing frame {frame_num}/{video_total_frames}...")
+
+        cap.release()
+        
+        final_chunks = [{'start_time': chunk['start_time'], 'score': sum(chunk['scores']) / len(chunk['scores'])} for chunk in chunk_scores if chunk['scores']]
+
+        if not final_chunks: raise gr.Error("Could not analyze video for motion. Is the video very short or static?")
+
+        # --- 2. Select the Best Clips ---
+        progress(0.2, desc="Selecting the best clips...")
+        num_clips_to_select = max(1, int(trailer_duration / clip_duration))
+        selected_clips_info = sorted(sorted(final_chunks, key=lambda x: x['score'], reverse=True)[:num_clips_to_select], key=lambda x: x['start_time'])
+        
+        # --- 3. Extract and Process Clips ---
+        extracted_clips, out_w, out_h = [], int(out_w) - (int(out_w) % 2), int(out_h) - (int(out_h) % 2)
+        
+        for i, clip_info in enumerate(selected_clips_info):
+            progress(0.2 + (0.5 * (i / len(selected_clips_info))), desc=f"Extracting clip {i+1}/{len(selected_clips_info)}...")
+            output_clip_path = os.path.join(job_temp_dir, f"clip_{i:03d}.mp4")
+            vf_filter = f"scale={out_w}:{out_h}:force_original_aspect_ratio=decrease,pad={out_w}:{out_h}:(ow-iw)/2:(oh-ih)/2,setsar=1"
+            cmd = ["ffmpeg", "-y", "-ss", str(clip_info['start_time']), "-i", video_path, "-t", str(clip_duration), "-vf", vf_filter, "-an", "-c:v", "libx264", "-pix_fmt", "yuv420p", output_clip_path]
+            run_ffmpeg_command(cmd)
+            extracted_clips.append(output_clip_path)
+
+        # --- 4. Stitch Clips with Transitions ---
+        if not extracted_clips: raise gr.Error("Failed to extract any clips.")
+        progress(0.7, desc="Stitching clips together...")
+        
+        final_silent_path = os.path.join(job_temp_dir, "final_silent.mp4")
+
+        if transition_style == "None" or len(extracted_clips) == 1:
+            if len(extracted_clips) == 1:
+                shutil.copy(extracted_clips[0], final_silent_path)
+            else:
+                file_list_path = os.path.join(job_temp_dir, "files.txt")
+                with open(file_list_path, 'w', encoding='utf-8') as f:
+                    for path in extracted_clips: f.write(f"file '{os.path.abspath(path)}'\n")
+                run_ffmpeg_command(["ffmpeg", "-y", "-f", "concat", "-safe", "0", "-i", file_list_path, "-c", "copy", final_silent_path], "Concatenating clips...")
+        else:
+            # --- IMPROVEMENT: Build a single, efficient FFMPEG command ---
+            transition_duration = 0.5
+            cmd = ["ffmpeg", "-y"]
+            filter_complex = []
+            running_duration = 0
+            
+            for i, clip_path in enumerate(extracted_clips):
+                cmd.extend(["-i", clip_path])
+            
+            for i in range(len(extracted_clips) - 1):
+                input1 = f"[{i}:v]" if i == 0 else f"[v{i-1}]"
+                input2 = f"[{i+1}:v]"
+                output = f"[v{i}]"
+                offset = max(0, running_duration + clip_duration - transition_duration)
+                filter_complex.append(f"{input1}{input2}xfade=transition={transition_style.lower()}:duration={transition_duration}:offset={offset}{output}")
+                running_duration += clip_duration - transition_duration
+            
+            cmd.extend([
+                "-filter_complex", ";".join(filter_complex),
+                "-map", f"[v{len(extracted_clips)-2}]",
+                "-c:v", "libx264", "-pix_fmt", "yuv420p",
+                final_silent_path
+            ])
+            run_ffmpeg_command(cmd, "Applying transitions...")
+        
+        # --- 5. Mux Audio ---
+        progress(0.95, desc="Adding background music...")
+        final_output_path = os.path.join(TEMP_DIR, f"trailer_final_{timestamp}.mp4")
+        if music_path:
+            run_ffmpeg_command(["ffmpeg", "-y", "-i", final_silent_path, "-i", music_path, "-c:v", "copy", "-c:a", "aac", "-map", "0:v:0", "-map", "1:a:0", "-shortest", final_output_path], "Muxing audio")
+        else:
+            shutil.move(final_silent_path, final_output_path)
+        return final_output_path
+    finally:
+        if os.path.exists(job_temp_dir): shutil.rmtree(job_temp_dir)
+
+
+def auto_trailer_wrapper(video_path, trailer_duration, clip_duration, analysis_method, transition_style, music_path, out_w, out_h, progress=gr.Progress(track_tqdm=True)):
+    return _create_auto_trailer_impl(video_path, trailer_duration, clip_duration, analysis_method, transition_style, music_path, out_w, out_h, progress)
+
+
 def generate_waveform_video(video_path, style, size, position, color):
     if not video_path: raise gr.Error("Please upload a video first.")
     
@@ -1574,8 +1720,7 @@ def _create_automated_slideshow_impl(images, audio_path, kb_effect_style, transi
         looped_image_paths = [image_paths[i % len(image_paths)] for i in range(num_clips)]
         
         kb_clips = []
-        # <-- FIX: Use a proper progress bar that tracks the whole process -->
-        total_steps = num_clips * 2 # Roughly 1 step for creating, 1 for joining
+        total_steps = num_clips + 1 
         current_step = 0
 
         for i in range(num_clips):
@@ -1624,33 +1769,30 @@ def _create_automated_slideshow_impl(images, audio_path, kb_effect_style, transi
 
         if not kb_clips: raise gr.Error("No clips were generated. The audio may be too short or the rhythm settings too fast.")
         
-        # --- FIX: New iterative chaining logic ---
+        progress(current_step / total_steps, desc=f"Applying transitions...")
+        final_silent_path = os.path.join(job_temp_dir, "final_silent.mp4")
         if len(kb_clips) == 1:
-            final_silent_path = kb_clips[0]['path']
+            shutil.copy(kb_clips[0]['path'], final_silent_path)
         else:
             all_transitions = ["fade", "wipeleft", "wiperight", "wipeup", "wipedown", "slideleft", "slideright", "slideup", "slidedown", "dissolve"]
+            cmd = ["ffmpeg", "-y"]
+            filter_complex = []
+            running_duration = 0
             
-            # Start with the first clip
-            current_video = kb_clips[0]['path']
-            running_duration = kb_clips[0]['duration']
+            for i, clip in enumerate(kb_clips):
+                cmd.extend(["-i", clip['path']])
 
             for i in range(len(kb_clips) - 1):
-                progress(current_step / total_steps, desc=f"Applying transition {i+1}/{len(kb_clips)-1}")
-                next_clip = kb_clips[i+1]['path']
-                output_path = os.path.join(job_temp_dir, f"temp_{i}.mp4")
-                
+                input1 = f"[{i}:v]" if i == 0 else f"[v{i-1}]"
+                input2 = f"[{i+1}:v]"
+                output = f"[v{i}]"
                 transition = random.choice(all_transitions) if transition_style == "Random" else transition_style.lower()
-                offset = running_duration - transition_duration
-                
-                filter_complex = f"[0:v][1:v]xfade=transition={transition}:duration={transition_duration}:offset={offset}[v]"
-                cmd_transition = ["ffmpeg", "-y", "-i", current_video, "-i", next_clip, "-filter_complex", filter_complex, "-map", "[v]", "-c:v", "libx264", "-pix_fmt", "yuv420p", output_path]
-                run_ffmpeg_command(cmd_transition)
+                offset = running_duration + kb_clips[i]['duration'] - transition_duration
+                filter_complex.append(f"{input1}{input2}xfade=transition={transition}:duration={transition_duration}:offset={offset}{output}")
+                running_duration += kb_clips[i]['duration'] - transition_duration
 
-                current_video = output_path
-                running_duration += kb_clips[i+1]['duration'] - transition_duration
-                current_step += 1
-            
-            final_silent_path = current_video
+            cmd.extend(["-filter_complex", ";".join(filter_complex), "-map", f"[v{len(kb_clips)-2}]", "-c:v", "libx264", "-pix_fmt", "yuv420p", final_silent_path])
+            run_ffmpeg_command(cmd)
 
         progress(0.98, desc="Muxing final audio...")
         final_output_path = os.path.join(TEMP_DIR, f"slideshow_final_{timestamp}.mp4")
@@ -1827,10 +1969,68 @@ bling_js = """
         const a=document.createElement("script");a.setAttribute("src","https://cdn.jsdelivr.net/npm/canvas-confetti@1.9.2/dist/confetti.browser.min.js"),document.head.appendChild(a),a.onload=()=>{var e=confetti.create(null,{resize:!0,useWorker:!0});e({particleCount:150,spread:90,origin:{y:.6}})}
     }
 
+    // --- Audio Feedback with Delay and Echo ---
+    const skriptz_audio = {
+        context: null,
+        buffer: null,
+        isInitialized: false,
+    };
+
+    async function init_audio() {
+        if (skriptz_audio.isInitialized) return;
+        try {
+            skriptz_audio.context = new (window.AudioContext || window.webkitAudioContext)();
+            // --- FIX FOR BROWSER AUTOPLAY POLICY ---
+            if (skriptz_audio.context.state === 'suspended') {
+                await skriptz_audio.context.resume();
+            }
+            const response = await fetch('/file=finish_sound.mp3');
+            if (!response.ok) {
+                console.warn(`Finish sound not loaded: ${response.status} ${response.statusText}. Ensure 'finish_sound.mp3' is in the root directory and 'allowed_paths' is set in demo.launch().`);
+                skriptz_audio.isInitialized = true;
+                return;
+            }
+            const arrayBuffer = await response.arrayBuffer();
+            skriptz_audio.buffer = await skriptz_audio.context.decodeAudioData(arrayBuffer);
+        } catch (e) {
+            console.error('Error initializing or decoding audio:', e);
+        }
+        skriptz_audio.isInitialized = true;
+    }
+
+    async function play_finish_sound() {
+        if (!skriptz_audio.isInitialized || !skriptz_audio.context) {
+            await init_audio();
+        }
+        if (!skriptz_audio.buffer || !skriptz_audio.context) return;
+        
+        // --- FIX FOR BROWSER AUTOPLAY POLICY ---
+        if (skriptz_audio.context.state === 'suspended') {
+            await skriptz_audio.context.resume();
+        }
+        
+        const context = skriptz_audio.context;
+        const source = context.createBufferSource();
+        source.buffer = skriptz_audio.buffer;
+        
+        const delay = context.createDelay();
+        delay.delayTime.value = 0.25;
+        
+        const feedback = context.createGain();
+        feedback.gain.value = 0.4;
+
+        source.connect(context.destination);
+        source.connect(delay);
+        delay.connect(feedback);
+        feedback.connect(delay);
+        delay.connect(context.destination);
+        
+        source.start(0);
+    }
+    
     // --- Dynamic Page Title ---
     function update_title(tab_name) {
         if (tab_name) {
-            // Strip emoji from tab name for a cleaner title
             const clean_name = tab_name.replace(/[\\u{1F600}-\\u{1F64F}\\u{1F300}-\\u{1F5FF}\\u{1F680}-\\u{1F6FF}\\u{1F700}-\\u{1F77F}\\u{1F780}-\\u{1F7FF}\\u{1F800}-\\u{1F8FF}\\u{1F900}-\\u{1F9FF}\\u{1FA00}-\\u{1FA6F}\\u{1FA70}-\\u{1FAFF}\\u{2600}-\\u{26FF}\\u{2700}-\\u{27BF}]/gu, '').trim();
             document.title = `Skriptz - ${clean_name}`;
         } else {
@@ -1844,7 +2044,6 @@ bling_js = """
         if(text_area) {
             text_area.select();
             document.execCommand('copy');
-            // Show a temporary "Copied!" message
             const original_button = this.event.target;
             const original_text = original_button.innerText;
             original_button.innerText = 'Copied!';
@@ -1863,6 +2062,7 @@ bling_js = """
         show_overlay,
         hide_overlay,
         fire_confetti,
+        play_finish_sound,
         update_title,
         copy_to_clipboard,
         storyboard_get_time
@@ -1892,7 +2092,7 @@ with gr.Blocks(
     storyboard_get_time_js = "() => { return window.skriptz_bling.storyboard_get_time(); }"
     show_overlay_js = "() => { window.skriptz_bling.show_overlay('Working hard... this may take a moment!'); }"
     hide_overlay_js = "() => { window.skriptz_bling.hide_overlay(); }"
-    fire_confetti_js = "() => { window.skriptz_bling.fire_confetti(); }"
+    fire_confetti_and_sound_js = "() => { window.skriptz_bling.fire_confetti(); window.skriptz_bling.play_finish_sound(); }"
     copy_transcription_js = "() => { window.skriptz_bling.copy_to_clipboard('transcription_textbox'); }"
 
     with gr.Tabs(elem_id="main_tabs") as main_tabs:
@@ -1993,11 +2193,11 @@ with gr.Blocks(
             clear_assets_btn.click(lambda: ([], gr.update(value=None), None), outputs=[assets_state, asset_gallery, asset_preview_gallery])
             clear_timeline_btn.click(lambda: [], outputs=[timeline_state])
 
-            generate_animatic_btn.click(fn=create_animatic, inputs=[timeline_state, animatic_audio, animatic_out_w, animatic_out_h], outputs=animatic_output_video).then(fn=None, js=fire_confetti_js)
+            generate_animatic_btn.click(fn=create_animatic, inputs=[timeline_state, animatic_audio, animatic_out_w, animatic_out_h], outputs=animatic_output_video).then(fn=None, js=fire_confetti_and_sound_js)
             
             analyze_bpm_btn.click(detect_bpm, animatic_audio, bpm_display)
             
-            generate_rhythmic_btn.click(fn=create_rhythmic_animatic, inputs=[timeline_state, animatic_audio, measure_dropdown, animatic_out_w, animatic_out_h], outputs=animatic_output_video).then(fn=None, js=fire_confetti_js)
+            generate_rhythmic_btn.click(fn=create_rhythmic_animatic, inputs=[timeline_state, animatic_audio, measure_dropdown, animatic_out_w, animatic_out_h], outputs=animatic_output_video).then(fn=None, js=fire_confetti_and_sound_js)
 
             
         with gr.TabItem("🎨 Creative Suite", elem_id="creative_tab"):
@@ -2027,8 +2227,40 @@ with gr.Blocks(
                         fn=slideshow_wrapper,
                         inputs=[slideshow_input_images, slideshow_audio, slideshow_kb_effect, slideshow_transition, slideshow_rhythm, slideshow_out_w, slideshow_out_h],
                         outputs=slideshow_output_video
-                    ).then(fn=None, js=fire_confetti_js)
+                    ).then(fn=None, js=fire_confetti_and_sound_js)
                 
+                with gr.TabItem("🎞️ Auto-Trailer"):
+                    gr.Markdown("## Automatic Trailer Creator")
+                    gr.Info("Upload a long video, and this tool will find the most action-packed moments to create a short, dynamic trailer.")
+                    with gr.Row():
+                        with gr.Column(scale=2):
+                            trailer_input_video = gr.Video(label="Upload Source Video")
+                            trailer_music = gr.Audio(label="Add Background Music (Optional)", type="filepath")
+                            
+                            with gr.Accordion("⚙️ Trailer Settings", open=True):
+                                trailer_total_duration = gr.Slider(10, 120, 30, step=5, label="Total Trailer Length (s)")
+                                trailer_clip_duration = gr.Slider(1.0, 5.0, 2.0, step=0.5, label="Duration of Each Clip (s)")
+                                trailer_analysis_method = gr.Dropdown(["Motion"], value="Motion", label="Scene Analysis Method", info="Currently only motion detection is supported.")
+                                trailer_transition = gr.Dropdown(["None", "Fade", "WipeLeft", "WipeRight", "Dissolve", "SlideLeft", "SlideRight"], value="Fade", label="Transition Style")
+                            
+                            with gr.Row():
+                                trailer_out_w = gr.Number(label="Output Width", value=1920)
+                                trailer_out_h = gr.Number(label="Output Height", value=1080)
+                            
+                            trailer_generate_btn = gr.Button("🚀 Generate Trailer", variant="primary")
+                        with gr.Column(scale=3):
+                            trailer_output_video = gr.Video(label="Generated Trailer Video", show_download_button=True)
+
+                    trailer_generate_btn.click(
+                        fn=auto_trailer_wrapper,
+                        inputs=[
+                            trailer_input_video, trailer_total_duration, trailer_clip_duration, 
+                            trailer_analysis_method, trailer_transition, trailer_music,
+                            trailer_out_w, trailer_out_h
+                        ],
+                        outputs=trailer_output_video
+                    ).then(fn=None, js=fire_confetti_and_sound_js)
+
                 with gr.TabItem("🎵 Audio Waveform"):
                     gr.Markdown("## Generate & Overlay Audio Waveforms")
                     gr.Info("Upload a video with an audio track to generate a dynamic waveform visualization.")
@@ -2048,7 +2280,7 @@ with gr.Blocks(
                         fn=generate_waveform_video,
                         inputs=[waveform_input_video, waveform_style, waveform_size, waveform_position, waveform_color],
                         outputs=waveform_output_video
-                    )
+                    ).then(fn=None, js=fire_confetti_and_sound_js)
 
                 with gr.TabItem("🖼️ Picture-in-Picture"):
                     gr.Markdown("## Create Picture-in-Picture (PiP) & Image Overlays")
@@ -2070,7 +2302,7 @@ with gr.Blocks(
                         fn=create_pip_video,
                         inputs=[pip_main_video, pip_overlay_media, pip_position, pip_scale],
                         outputs=pip_output_video
-                    )
+                    ).then(fn=None, js=fire_confetti_and_sound_js)
                 
                 with gr.TabItem("😂 Meme Creator"):
                     gr.Markdown("## Classic Meme & Text Overlay Creator")
@@ -2096,7 +2328,7 @@ with gr.Blocks(
                         fn=create_meme,
                         inputs=[meme_input_image, meme_text, meme_position, meme_font, meme_font_size, meme_text_color, meme_outline_color],
                         outputs=meme_output_image
-                    )
+                    ).then(fn=None, js=fire_confetti_and_sound_js)
 
         with gr.TabItem("🖼️ Image Utilities", elem_id="image_tab"):
             gr.Markdown("## Tools for processing and generating single images.")
@@ -2114,7 +2346,7 @@ with gr.Blocks(
                             manip_apply_btn = gr.Button("🎨 Apply Manipulation", variant="primary")
                         with gr.Column():
                             manip_output_image = gr.Image(label="Output Image", interactive=True)
-                    manip_apply_btn.click(fn=manipulate_image, inputs=[manip_input_image, manip_operation_radio], outputs=manip_output_image)
+                    manip_apply_btn.click(fn=manipulate_image, inputs=[manip_input_image, manip_operation_radio], outputs=manip_output_image).then(fn=None, js=fire_confetti_and_sound_js)
 
                 with gr.TabItem("📹 Image to Video"):
                     gr.Markdown("### Create a short, looping video from a single static image.")
@@ -2126,7 +2358,7 @@ with gr.Blocks(
                             compile_i2v_btn = gr.Button("🎬 Create Looping Video", variant="primary")
                         with gr.Column():
                             output_video_i2v = gr.Video(label="Output Looping Video", interactive=True, show_download_button=True)
-                    compile_i2v_btn.click(image_to_looping_video, [input_image_i2v, duration_slider_i2v, input_audio_i2v], output_video_i2v)
+                    compile_i2v_btn.click(image_to_looping_video, [input_image_i2v, duration_slider_i2v, input_audio_i2v], output_video_i2v).then(fn=None, js=fire_confetti_and_sound_js)
                 
                 with gr.TabItem("🔎 Zoom Video"):
                     gr.Markdown("### Create a 'Ken Burns' style zoom/pan video from an image.")
@@ -2157,7 +2389,7 @@ with gr.Blocks(
                         fn=create_zoom_videos, 
                         inputs=[i2zv_input_images, i2zv_duration, i2zv_zoom_ratio, i2zv_zoom_dir, i2zv_combine, i2zv_audio], 
                         outputs=[i2zv_output_gallery, i2zv_output_video, i2zv_output_zip]
-                    )
+                    ).then(fn=None, js=fire_confetti_and_sound_js)
 
                 with gr.TabItem("✂️ Batch BG Remover"):
                     gr.Markdown("### Remove the background from a batch of images.")
@@ -2168,7 +2400,7 @@ with gr.Blocks(
                         with gr.Column():
                             output_gallery_bg = gr.Gallery(label="Images with Transparent Background", columns=4, object_fit="contain", height="auto")
                             output_zip_bg = gr.File(label="Download All as .zip", interactive=False)
-                    remove_bg_btn.click(remove_background_batch, input_images_bg, [output_gallery_bg, output_zip_bg])
+                    remove_bg_btn.click(remove_background_batch, input_images_bg, [output_gallery_bg, output_zip_bg]).then(fn=None, js=fire_confetti_and_sound_js)
                 
                 with gr.TabItem("🖋️ Batch Watermarker"):
                     gr.Markdown("### Apply a text watermark to a batch of images.")
@@ -2182,7 +2414,7 @@ with gr.Blocks(
                         with gr.Column():
                             output_gallery_wm = gr.Gallery(label="Watermarked Images", columns=4, object_fit="contain", height="auto")
                             output_zip_wm = gr.File(label="Download All as .zip", interactive=False)
-                    watermark_btn.click(apply_watermark_batch, [input_images_wm, watermark_text, watermark_pos, watermark_opacity], [output_gallery_wm, output_zip_wm])
+                    watermark_btn.click(apply_watermark_batch, [input_images_wm, watermark_text, watermark_pos, watermark_opacity], [output_gallery_wm, output_zip_wm]).then(fn=None, js=fire_confetti_and_sound_js)
                 
                 with gr.TabItem("↔️ Batch Resizer"):
                     gr.Markdown("### Convert, resize, and compress a batch of images.")
@@ -2208,7 +2440,7 @@ with gr.Blocks(
                         batch_resize_convert_images, 
                         [brc_input_images, brc_format, brc_quality, brc_enable_resize, brc_max_w, brc_max_h, brc_resize_mode], 
                         [brc_output_gallery, brc_output_zip]
-                    )
+                    ).then(fn=None, js=fire_confetti_and_sound_js)
 
                 with gr.TabItem("🤖 FLUX.1 API"):
                     gr.Markdown("### Generate an image using `FLUX.1` models via Gradio Client.")
@@ -2224,7 +2456,7 @@ with gr.Blocks(
                             flux_btn = gr.Button("☄️ Generate Image", variant="primary")
                         with gr.Column():
                             output_image_flux = gr.Image(label="Generated Image", interactive=True)
-                    flux_btn.click(call_flux_api, [prompt_input_flux, flux_model_dropdown, flux_width_slider, flux_height_slider, hf_token_input], output_image_flux)
+                    flux_btn.click(call_flux_api, [prompt_input_flux, flux_model_dropdown, flux_width_slider, flux_height_slider, hf_token_input], output_image_flux).then(fn=None, js=fire_confetti_and_sound_js)
         
         with gr.TabItem("🎥 Video Utilities", elem_id="video_tab"):
             gr.Markdown("## A collection of useful video tools.")
@@ -2239,7 +2471,7 @@ with gr.Blocks(
                                     extract_fl_btn = gr.Button("🎬 Extract Frames", variant="primary")
                                 with gr.Column():
                                     output_gallery_fl = gr.Gallery(label="Output Frames (First, Last)", columns=2, object_fit="contain", height="auto")
-                            extract_fl_btn.click(fn=extract_first_last_frame, inputs=input_video_fl, outputs=output_gallery_fl)
+                            extract_fl_btn.click(fn=extract_first_last_frame, inputs=input_video_fl, outputs=output_gallery_fl).then(fn=None, js=fire_confetti_and_sound_js)
 
                         with gr.TabItem("Video to Frames"):
                             gr.Markdown("### Extract all individual frames from a video file.")
@@ -2263,7 +2495,7 @@ with gr.Blocks(
                             input_video_v2f.upload(lambda v: f"{get_video_fps(v):.2f} FPS", input_video_v2f, v2f_fps_display)
                             v2f_resize.change(lambda x: [gr.update(interactive=x), gr.update(interactive=x)], v2f_resize, [v2f_width, v2f_height])
                             v2f_format.change(lambda x: gr.update(interactive=(x=="JPG")), v2f_format, v2f_jpg_quality)
-                            extract_v2f_btn.click(video_to_frames_extractor, [input_video_v2f, v2f_skip_rate, v2f_rotation, v2f_resize, v2f_width, v2f_height, v2f_format, v2f_jpg_quality], [output_gallery_v2f, output_zip_v2f])
+                            extract_v2f_btn.click(video_to_frames_extractor, [input_video_v2f, v2f_skip_rate, v2f_rotation, v2f_resize, v2f_width, v2f_height, v2f_format, v2f_jpg_quality], [output_gallery_v2f, output_zip_v2f]).then(fn=None, js=fire_confetti_and_sound_js)
                         
                         with gr.TabItem("Frames to Video"):
                             gr.Markdown("### Compile a sequence of image frames into a video file.")
@@ -2281,7 +2513,7 @@ with gr.Blocks(
                                 with gr.Column():
                                     output_video_f2v = gr.Video(label="Compiled Video", interactive=True, show_download_button=True)
                             f2v_resize.change(lambda x: [gr.update(interactive=x), gr.update(interactive=x)], f2v_resize, [f2v_width, f2v_height])
-                            compile_f2v_btn.click(create_video_from_frames, [input_frames_f2v, fps_slider_f2v, f2v_rotation, f2v_resize, f2v_width, f2v_height], output_video_f2v)
+                            compile_f2v_btn.click(create_video_from_frames, [input_frames_f2v, fps_slider_f2v, f2v_rotation, f2v_resize, f2v_width, f2v_height], output_video_f2v).then(fn=None, js=fire_confetti_and_sound_js)
                 
                 with gr.TabItem("✂️ Editing"):
                     with gr.Tabs():
@@ -2299,7 +2531,7 @@ with gr.Blocks(
                                 with gr.Column():
                                     vmanip_output_video = gr.Video(label="Output Video", interactive=True, show_download_button=True)
                             
-                            vmanip_apply_btn.click(fn=manipulate_video, inputs=[vmanip_input_video, vmanip_operation_radio], outputs=vmanip_output_video)
+                            vmanip_apply_btn.click(fn=manipulate_video, inputs=[vmanip_input_video, vmanip_operation_radio], outputs=vmanip_output_video).then(fn=None, js=fire_confetti_and_sound_js)
                         
                         with gr.TabItem("Ping-Pong"):
                             gr.Markdown("### Create a forward-then-reverse video loop (Boomerang).")
@@ -2310,7 +2542,7 @@ with gr.Blocks(
                                     pingpong_btn = gr.Button("🏓 Create Ping-Pong Video", variant="primary")
                                 with gr.Column():
                                     output_video_pingpong = gr.Video(label="Ping-Pong Video", interactive=True, show_download_button=True)
-                            pingpong_btn.click(fn=ping_pong_video, inputs=[input_video_pingpong, audio_option_pingpong], outputs=output_video_pingpong)
+                            pingpong_btn.click(fn=ping_pong_video, inputs=[input_video_pingpong, audio_option_pingpong], outputs=output_video_pingpong).then(fn=None, js=fire_confetti_and_sound_js)
                         
                         with gr.TabItem("Reverse"):
                             gr.Markdown("### Reverse a video clip.")
@@ -2321,7 +2553,7 @@ with gr.Blocks(
                                     reverse_btn = gr.Button("🔄 Reverse Video", variant="primary")
                                 with gr.Column():
                                     output_video_reverse = gr.Video(label="Reversed Video", interactive=True, show_download_button=True)
-                            reverse_btn.click(fn=reverse_video, inputs=[input_video_reverse, audio_option_reverse], outputs=output_video_reverse)
+                            reverse_btn.click(fn=reverse_video, inputs=[input_video_reverse, audio_option_reverse], outputs=output_video_reverse).then(fn=None, js=fire_confetti_and_sound_js)
                         
                         with gr.TabItem("Merger"):
                             gr.Markdown("### Simple Video Merger")
@@ -2336,7 +2568,7 @@ with gr.Blocks(
                                 fn=merge_videos,
                                 inputs=merger_input_videos,
                                 outputs=merger_output_video
-                            )
+                            ).then(fn=None, js=fire_confetti_and_sound_js)
 
                         with gr.TabItem("Trimmer"):
                             gr.Markdown("### Visually trim a video. Use the player to find a frame, then set it as the start or end point.")
@@ -2364,7 +2596,7 @@ with gr.Blocks(
                             set_end_btn.click(fn=None, js=get_current_time_js, outputs=[trim_end_time])
                             trim_start_time.change(fn=get_frame_from_time_wrapper, inputs=[input_video_trim, trim_start_time], outputs=[start_frame_img, trim_start_time_display])
                             trim_end_time.change(fn=get_frame_from_time_wrapper, inputs=[input_video_trim, trim_end_time], outputs=[end_frame_img, trim_end_time_display])
-                            trim_btn.click(fn=trim_video, inputs=[input_video_trim, trim_start_time, trim_end_time], outputs=output_video_trim)
+                            trim_btn.click(fn=trim_video, inputs=[input_video_trim, trim_start_time, trim_end_time], outputs=output_video_trim).then(fn=None, js=fire_confetti_and_sound_js)
                             input_video_trim.clear(fn=lambda: (None, "0.00", None, "0.00", 0, 0), outputs=[start_frame_img, trim_start_time_display, end_frame_img, trim_end_time_display, trim_start_time, trim_end_time])
                         
                         with gr.TabItem("Crop & Resize"):
@@ -2390,7 +2622,7 @@ with gr.Blocks(
                                     output_video_crop = gr.Video(label="Cropped Video", interactive=True, show_download_button=True)
                             crop_input_video.upload(fn=get_frame_at_time, inputs=crop_input_video, outputs=crop_preview_image)
                             crop_do_resize.change(lambda x: [gr.update(interactive=x), gr.update(interactive=x)], inputs=crop_do_resize, outputs=[crop_resize_w, crop_resize_h])
-                            crop_btn.click(fn=crop_video, inputs=[crop_input_video, crop_x, crop_y, crop_w, crop_h, crop_do_resize, crop_resize_w, crop_resize_h], outputs=output_video_crop)
+                            crop_btn.click(fn=crop_video, inputs=[crop_input_video, crop_x, crop_y, crop_w, crop_h, crop_do_resize, crop_resize_w, crop_resize_h], outputs=output_video_crop).then(fn=None, js=fire_confetti_and_sound_js)
                         
                         with gr.TabItem("Change Speed"):
                             gr.Markdown("### Create slow-motion or fast-forward videos.")
@@ -2401,7 +2633,7 @@ with gr.Blocks(
                                     speed_btn = gr.Button("🏃 Change Speed", variant="primary")
                                 with gr.Column():
                                     output_video_speed = gr.Video(label="Modified Video", interactive=True, show_download_button=True)
-                            speed_btn.click(fn=change_video_speed, inputs=[input_video_speed, speed_multiplier], outputs=output_video_speed)
+                            speed_btn.click(fn=change_video_speed, inputs=[input_video_speed, speed_multiplier], outputs=output_video_speed).then(fn=None, js=fire_confetti_and_sound_js)
                 
                 with gr.TabItem("🎨 Effects & Overlays"):
                     with gr.Tabs():
@@ -2416,7 +2648,7 @@ with gr.Blocks(
                                     fade_video_btn = gr.Button("✨ Apply Fade", variant="primary")
                                 with gr.Column():
                                     fade_output_video = gr.Video(label="Faded Video", interactive=True, show_download_button=True)
-                            fade_video_btn.click(apply_video_fade, [fade_input_video, fade_in_slider, fade_out_slider], fade_output_video)
+                            fade_video_btn.click(apply_video_fade, [fade_input_video, fade_in_slider, fade_out_slider], fade_output_video).then(fn=None, js=fire_confetti_and_sound_js)
                         
                         with gr.TabItem("BG Remover"):
                             gr.Markdown("## Video Background Remover")
@@ -2427,7 +2659,7 @@ with gr.Blocks(
                                     vbg_btn = gr.Button("✂️ Remove Video Background", variant="primary")
                                 with gr.Column():
                                     vbg_output_video = gr.Video(label="Output Video with Transparency (.webm)", interactive=True, show_download_button=True)
-                            vbg_btn.click(remove_video_background, vbg_input_video, vbg_output_video)
+                            vbg_btn.click(remove_video_background, vbg_input_video, vbg_output_video).then(fn=None, js=fire_confetti_and_sound_js)
                         
                         with gr.TabItem("Watermark"):
                             gr.Markdown("### Apply a text watermark to a video.")
@@ -2443,7 +2675,7 @@ with gr.Blocks(
                                     wm_btn = gr.Button("🖋️ Apply Watermark", variant="primary")
                                 with gr.Column():
                                     wm_output_video = gr.Video(label="Watermarked Video", interactive=True, show_download_button=True)
-                            wm_btn.click(apply_video_watermark, [wm_input_video, wm_text, wm_pos, wm_opacity, wm_size, wm_color], wm_output_video)
+                            wm_btn.click(apply_video_watermark, [wm_input_video, wm_text, wm_pos, wm_opacity, wm_size, wm_color], wm_output_video).then(fn=None, js=fire_confetti_and_sound_js)
 
                         with gr.TabItem("Create GIF"):
                             gr.Markdown("### Convert a video clip into a high-quality animated GIF.")
@@ -2456,7 +2688,7 @@ with gr.Blocks(
                                     gif_btn = gr.Button("🖼️ Create GIF", variant="primary")
                                 with gr.Column():
                                     output_gif = gr.Image(label="Output GIF", interactive=True)
-                            gif_btn.click(create_gif_from_video, [input_video_gif, gif_start_time, gif_end_time], output_gif)
+                            gif_btn.click(create_gif_from_video, [input_video_gif, gif_start_time, gif_end_time], output_gif).then(fn=None, js=fire_confetti_and_sound_js)
                 
                 with gr.TabItem("🔊 Audio & Transcription"):
                      with gr.Tabs():
@@ -2469,7 +2701,7 @@ with gr.Blocks(
                                     add_audio_btn = gr.Button("🎶 Add Audio", variant="primary")
                                 with gr.Column():
                                     output_video_audio = gr.Video(label="Final Video with Audio", interactive=True, show_download_button=True)
-                            add_audio_btn.click(add_audio_to_video, [input_video_audio, input_audio], output_video_audio)
+                            add_audio_btn.click(add_audio_to_video, [input_video_audio, input_audio], output_video_audio).then(fn=None, js=fire_confetti_and_sound_js)
                         
                         with gr.TabItem("Extract Audio"):
                             gr.Markdown("### Strip the audio track from a video file.")
@@ -2480,7 +2712,7 @@ with gr.Blocks(
                                     extract_audio_btn = gr.Button("🎵 Extract Audio", variant="primary")
                                 with gr.Column():
                                     extract_audio_output = gr.Audio(label="Extracted Audio", type="filepath")
-                            extract_audio_btn.click(extract_audio, [extract_audio_input_video, extract_audio_format], extract_audio_output)
+                            extract_audio_btn.click(extract_audio, [extract_audio_input_video, extract_audio_format], extract_audio_output).then(fn=None, js=fire_confetti_and_sound_js)
 
                         with gr.TabItem("Audio Editor"):
                             gr.Markdown("### Trim and Apply Fades to an Audio File")
@@ -2497,7 +2729,7 @@ with gr.Blocks(
                                     audio_trim_fade_btn = gr.Button("✂️ Process Audio", variant="primary")
                                 with gr.Column():
                                     audio_trim_output = gr.Audio(label="Processed Audio", type="filepath")
-                            audio_trim_fade_btn.click(trim_and_fade_audio, [audio_trim_input, audio_start_time, audio_end_time, audio_fade_in, audio_fade_out], audio_trim_output)
+                            audio_trim_fade_btn.click(trim_and_fade_audio, [audio_trim_input, audio_start_time, audio_end_time, audio_fade_in, audio_fade_out], audio_trim_output).then(fn=None, js=fire_confetti_and_sound_js)
                         
                         with gr.TabItem("Transcription", visible=(whisper is not None)):
                             gr.Markdown("## Transcribe Speech and Burn Subtitles")
@@ -2527,12 +2759,12 @@ with gr.Blocks(
                                 fn=transcribe_and_prep_burn, 
                                 inputs=[transcribe_input, transcribe_model], 
                                 outputs=[transcribe_text, transcribe_files, transcribed_video_path_state, burn_accordion]
-                            )
+                            ).then(fn=None, js=fire_confetti_and_sound_js)
                             burn_btn.click(
                                 fn=burn_subtitles,
                                 inputs=[transcribed_video_path_state, transcribe_files, burn_font_size, burn_font_color],
                                 outputs=burn_output_video
-                            )
+                            ).then(fn=None, js=fire_confetti_and_sound_js)
 
         with gr.TabItem("🧠 ControlNet", elem_id="controlnet_tab"):
             gr.Markdown("## ControlNet Preprocessing")
@@ -2546,7 +2778,7 @@ with gr.Blocks(
                             process_btn_cn = gr.Button("✨ Process Video", variant="primary")
                         with gr.Column():
                             output_video_cn = gr.Video(label="Output ControlNet Video", interactive=True, show_download_button=True)
-                    process_btn_cn.click(fn=process_video_with_detector, inputs=[input_video_cn, detector_dropdown_cn], outputs=output_video_cn)
+                    process_btn_cn.click(fn=process_video_with_detector, inputs=[input_video_cn, detector_dropdown_cn], outputs=output_video_cn).then(fn=None, js=fire_confetti_and_sound_js)
                 with gr.TabItem("Process Batch Images"):
                     gr.Markdown("### Generate ControlNet Maps from one or more images.")
                     with gr.Row():
@@ -2557,7 +2789,7 @@ with gr.Blocks(
                         with gr.Column():
                             output_gallery_cn = gr.Gallery(label="Output ControlNet Images", columns=4, object_fit="contain", height="auto")
                             output_zip_cn = gr.File(label="Download All as .zip", interactive=False)
-                    process_btn_img.click(fn=process_batch_images_with_detector, inputs=[input_images_cn, detector_dropdown_img], outputs=[output_gallery_cn, output_zip_cn])
+                    process_btn_img.click(fn=process_batch_images_with_detector, inputs=[input_images_cn, detector_dropdown_img], outputs=[output_gallery_cn, output_zip_cn]).then(fn=None, js=fire_confetti_and_sound_js)
         
         with gr.TabItem("🗜️ Converter", elem_id="converter_tab"):
             gr.Markdown("## Universal Video Converter & Compressor")
@@ -2579,7 +2811,7 @@ with gr.Blocks(
                         conv_acodec = gr.Dropdown(["copy", "aac", "opus"], value="copy", label="Audio Codec", info="'copy' is fastest and preserves quality.")
                         conv_abitrate = gr.Dropdown([96, 128, 192, 256, 320], value=192, label="Audio Bitrate (kbps)", interactive=False)
                     conv_acodec.change(lambda x: gr.update(interactive=(x != "copy")), conv_acodec, conv_abitrate)
-            conv_btn.click(fn=convert_compress_video, inputs=[conv_input_video, conv_format, conv_vcodec, conv_crf, conv_scale, conv_acodec, conv_abitrate], outputs=conv_output_video)
+            conv_btn.click(fn=convert_compress_video, inputs=[conv_input_video, conv_format, conv_vcodec, conv_crf, conv_scale, conv_acodec, conv_abitrate], outputs=conv_output_video).then(fn=None, js=fire_confetti_and_sound_js)
 
         with gr.TabItem("🔗 Transfer", elem_id="transfer_tab"):
             gr.Markdown("## Image & Link Transfer Utility")
@@ -2627,4 +2859,6 @@ if __name__ == "__main__":
         try: shutil.rmtree(TEMP_DIR)
         except OSError as e: print(f"Error removing temp directory {TEMP_DIR}: {e}")
     os.makedirs(TEMP_DIR, exist_ok=True)
-    demo.launch(inbrowser=True)
+    # --- THIS IS THE MAIN FIX ---
+    # Add allowed_paths to let Gradio serve the sound file from the current directory.
+    demo.launch(inbrowser=True, allowed_paths=["."])