Backup: server-side OAuth version before client-side experiment

app_server_oauth.py

@@ -0,0 +1,775 @@
+import gradio as gr
+import subprocess
+import os
+import tempfile
+import shutil
+import logging
+import time
+from pathlib import Path
+try:
+    from native_drive_picker import GoogleDrivePickerManager, get_native_picker_instructions, GOOGLE_DRIVE_AVAILABLE
+except ImportError:
+    GOOGLE_DRIVE_AVAILABLE = False
+    GoogleDrivePickerManager = None
+    def get_native_picker_instructions():
+        return "Google Drive integration not available in this environment."
+
+# Set up logging
+logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
+logger = logging.getLogger(__name__)
+
+def process_video_trim(video_file, start_time, end_time):
+    """Process video trimming using the trim-convert.sh script"""
+    logger.info(f"🎬 Starting trim process: file={video_file}, start={start_time}, end={end_time}")
+    
+    if not video_file or start_time is None or end_time is None:
+        error_msg = "Please provide video file and both start/end times"
+        logger.error(f"❌ {error_msg}")
+        return None, None, None, error_msg
+    
+    try:
+        # start_time and end_time are now numbers (seconds) from sliders
+        start_seconds = float(start_time)
+        end_seconds = float(end_time)
+        
+        logger.info(f"📊 Parsed times: start={start_seconds}s, end={end_seconds}s")
+        
+        if start_seconds >= end_seconds:
+            error_msg = "Start time must be less than end time"
+            logger.error(f"❌ {error_msg}")
+            return None, None, None, error_msg
+        
+        # Check if input file exists
+        if not os.path.exists(video_file):
+            error_msg = f"Input video file not found: {video_file}"
+            logger.error(f"❌ {error_msg}")
+            return None, None, None, error_msg
+        
+        # Create temporary directory for output
+        temp_dir = tempfile.mkdtemp()
+        logger.info(f"📁 Created temp directory: {temp_dir}")
+        
+        input_path = video_file
+        
+        # Get the base filename without extension
+        base_name = Path(input_path).stem
+        output_prefix = os.path.join(temp_dir, f"{base_name}_trimmed")
+        
+        # The script will create these files based on the prefix
+        output_video = f"{output_prefix}.mp4"
+        output_audio = f"{output_prefix}.aac"
+        
+        logger.info(f"📤 Output files will be: video={output_video}, audio={output_audio}")
+        
+        # Check if trim-convert.sh script exists
+        script_path = "./trim-convert.sh"
+        if not os.path.exists(script_path):
+            error_msg = f"trim-convert.sh script not found at: {script_path}"
+            logger.error(f"❌ {error_msg}")
+            return None, None, None, error_msg
+        
+        # Convert seconds to HH:MM:SS format for the script
+        def seconds_to_time(seconds):
+            hours = int(seconds // 3600)
+            minutes = int((seconds % 3600) // 60)
+            secs = seconds % 60
+            return f"{hours:02d}:{minutes:02d}:{secs:06.3f}"
+        
+        start_time_str = seconds_to_time(start_seconds)
+        end_time_str = seconds_to_time(end_seconds)
+        
+        logger.info(f"🕒 Converted times: start={start_time_str}, end={end_time_str}")
+        
+        # Call the trim-convert.sh script with proper format
+        cmd = [
+            "bash", script_path,
+            "-s", start_time_str,
+            "-e", end_time_str, 
+            "-o", output_prefix,
+            input_path
+        ]
+        
+        logger.info(f"🚀 Running command: {' '.join(cmd)}")
+        
+        result = subprocess.run(cmd, capture_output=True, text=True, cwd='.')
+        
+        logger.info(f"📋 Command finished with return code: {result.returncode}")
+        logger.info(f"📤 STDOUT: {result.stdout}")
+        if result.stderr:
+            logger.warning(f"⚠️  STDERR: {result.stderr}")
+        
+        if result.returncode == 0:
+            # Check if files were created
+            video_exists = os.path.exists(output_video)
+            audio_exists = os.path.exists(output_audio)
+            
+            logger.info(f"📁 File check: video_exists={video_exists}, audio_exists={audio_exists}")
+            
+            if video_exists and audio_exists:
+                video_size = os.path.getsize(output_video)
+                audio_size = os.path.getsize(output_audio)
+                logger.info(f"📊 File sizes: video={video_size} bytes, audio={audio_size} bytes")
+                
+                # Check if video file is valid and convert for better web compatibility
+                try:
+                    test_duration = get_video_duration(output_video)
+                    logger.info(f"✅ Output video duration: {test_duration} seconds")
+                    if test_duration == 0:
+                        logger.warning("⚠️ Output video duration is 0, may have encoding issues")
+                    
+                    # Check if trimmed video is web-compatible, if not, convert only the headers
+                    display_video = output_video  # Start with original
+                    
+                    # Quick check if video might have compatibility issues
+                    try:
+                        # Test if ffprobe can read the file properly
+                        probe_cmd = ["ffprobe", "-v", "quiet", "-print_format", "json", "-show_format", output_video]
+                        probe_result = subprocess.run(probe_cmd, capture_output=True, text=True)
+                        
+                        if probe_result.returncode == 0:
+                            import json
+                            probe_data = json.loads(probe_result.stdout)
+                            format_info = probe_data.get('format', {})
+                            
+                            # Check if it needs web optimization
+                            needs_conversion = False
+                            
+                            # If the file has issues or isn't web-optimized, do a quick fix
+                            if needs_conversion or True:  # Always do quick web optimization for now
+                                web_video_path = os.path.join(temp_dir, f"{base_name}_web.mp4")
+                                
+                                # Quick web compatibility fix - just fix headers and ensure proper format
+                                web_convert_cmd = [
+                                    "ffmpeg", "-y", "-i", output_video,
+                                    "-c", "copy",  # Copy streams (fast)
+                                    "-movflags", "+faststart",  # Optimize for web
+                                    "-f", "mp4",  # Ensure MP4 format
+                                    web_video_path
+                                ]
+                                
+                                logger.info(f"🌐 Quick web optimization (stream copy)...")
+                                web_result = subprocess.run(web_convert_cmd, capture_output=True, text=True)
+                                
+                                if web_result.returncode == 0 and os.path.exists(web_video_path):
+                                    web_size = os.path.getsize(web_video_path)
+                                    logger.info(f"✅ Web-optimized video: {web_video_path} ({web_size} bytes)")
+                                    display_video = web_video_path
+                                    
+                                    # Verify the optimized video
+                                    web_duration = get_video_duration(web_video_path)
+                                    logger.info(f"🎬 Optimized video duration: {web_duration} seconds")
+                                else:
+                                    logger.warning(f"⚠️ Quick optimization failed: {web_result.stderr}")
+                                    logger.info("Using original trimmed video")
+                        else:
+                            logger.warning("⚠️ Could not analyze trimmed video, using as-is")
+                            
+                    except Exception as e:
+                        logger.warning(f"⚠️ Video analysis failed: {e}, using original")
+                        
+                except Exception as e:
+                    logger.warning(f"⚠️ Could not verify output video: {e}")
+                    display_video = output_video
+                
+                # Create MP3 version for audio player (better browser compatibility)
+                timestamp = str(int(time.time() * 1000))
+                temp_audio_dir = os.path.dirname(output_audio)
+                audio_player_file = os.path.join(temp_audio_dir, f"player_audio_{timestamp}.mp3")
+                
+                # Convert AAC to MP3 for better browser support
+                convert_cmd = [
+                    "ffmpeg", "-y", "-i", output_audio, 
+                    "-codec:a", "libmp3lame", "-b:a", "128k",
+                    audio_player_file
+                ]
+                
+                logger.info(f"🔄 Converting audio for player: {' '.join(convert_cmd)}")
+                convert_result = subprocess.run(convert_cmd, capture_output=True, text=True)
+                
+                if convert_result.returncode == 0 and os.path.exists(audio_player_file):
+                    logger.info(f"🎵 Created MP3 audio player file: {audio_player_file}")
+                    logger.info(f"📊 Audio player file size: {os.path.getsize(audio_player_file)} bytes")
+                else:
+                    logger.warning(f"⚠️ MP3 conversion failed, using original AAC file")
+                    audio_player_file = output_audio
+                
+                success_msg = f"✅ Successfully trimmed video from {start_seconds:.1f}s to {end_seconds:.1f}s"
+                
+                # No automatic upload - will be done manually after trimming
+                
+                logger.info(success_msg)
+                return display_video, audio_player_file, output_audio, success_msg, output_video, output_audio
+            else:
+                error_msg = f"❌ Output files not created.\n\nScript STDOUT:\n{result.stdout}\n\nScript STDERR:\n{result.stderr}\n\nExpected files:\nVideo: {output_video}\nAudio: {output_audio}"
+                logger.error(error_msg)
+                return None, None, None, error_msg, None, None
+        else:
+            error_msg = f"❌ trim-convert.sh failed with return code {result.returncode}\n\nCommand run:\n{' '.join(cmd)}\n\nSTDOUT:\n{result.stdout}\n\nSTDERR:\n{result.stderr}"
+            logger.error(error_msg)
+            return None, None, None, error_msg
+            
+    except Exception as e:
+        error_msg = f"❌ Unexpected error: {str(e)}"
+        logger.exception(error_msg)
+        return None, None, None, error_msg
+
+def get_video_duration(video_file):
+    """Get video duration in seconds"""
+    if not video_file:
+        return 0
+    
+    try:
+        logger.info(f"📺 Getting duration for: {video_file}")
+        
+        # Use ffprobe to get video duration
+        cmd = [
+            "ffprobe", "-v", "quiet", "-print_format", "json", 
+            "-show_format", "-show_streams", video_file
+        ]
+        result = subprocess.run(cmd, capture_output=True, text=True)
+        
+        if result.returncode == 0:
+            import json
+            data = json.loads(result.stdout)
+            duration = float(data['format']['duration'])
+            logger.info(f"⏱️ Video duration: {duration} seconds")
+            return duration
+        else:
+            logger.warning(f"⚠️ Could not get duration: {result.stderr}")
+            return 0
+    except Exception as e:
+        logger.exception(f"❌ Error getting video duration: {e}")
+        return 0
+
+def format_time(seconds):
+    """Format seconds to mm:ss"""
+    if seconds is None:
+        return "0:00"
+    minutes = int(seconds // 60)
+    secs = int(seconds % 60)
+    return f"{minutes}:{secs:02d}"
+
+def get_video_info(video_file):
+    """Get video duration and basic info"""
+    if not video_file:
+        return "No video uploaded", 0, 0, 0
+    
+    logger.info(f"📹 Processing video upload: {video_file}")
+    
+    duration = get_video_duration(video_file)
+    if duration > 0:
+        minutes = int(duration // 60)
+        seconds = int(duration % 60)
+        info = f"📹 Video loaded! Duration: {minutes}:{seconds:02d} ({duration:.1f}s)"
+        logger.info(f"✅ {info}")
+        return info, duration, 0, duration
+    else:
+        info = "📹 Video loaded! (Could not determine duration)"
+        logger.warning(f"⚠️ {info}")
+        return info, 100, 0, 100
+
+# Native Google Drive picker functions
+def open_file_picker(drive_manager):
+    """Open Google Drive for file selection (full access)"""
+    if not drive_manager or not drive_manager.is_available():
+        return "❌ Google Drive not available"
+    
+    instructions = drive_manager.open_drive_picker("file")
+    return instructions
+
+def open_folder_picker(drive_manager):
+    """Open Google Drive for folder selection"""
+    if not drive_manager or not drive_manager.is_available():
+        return "❌ Google Drive not available"
+    
+    instructions = drive_manager.open_drive_picker("folder")
+    return instructions
+
+def download_from_drive_url(drive_manager, drive_url, custom_filename=""):
+    """Download video from Google Drive URL"""
+    if not drive_manager or not drive_manager.is_available():
+        return None, "❌ Google Drive not available"
+    
+    if not drive_url or not drive_url.strip():
+        return None, "⚠️ Please paste a Google Drive link"
+    
+    filename = custom_filename.strip() if custom_filename.strip() else None
+    return drive_manager.download_file_from_url(drive_url, filename)
+
+def download_from_google_drive(file_id, file_display, drive_manager):
+    """Download selected file from Google Drive"""
+    if not file_id or not drive_manager or not drive_manager.is_available():
+        return None, "❌ No file selected or Google Drive unavailable"
+    
+    try:
+        # Extract filename from display string
+        filename = file_display.split(' (')[0] if file_display else f"video_{file_id}.mp4"
+        
+        logger.info(f"📥 Downloading {filename} from Google Drive...")
+        local_path = drive_manager.download_file(file_id, filename)
+        
+        if local_path and os.path.exists(local_path):
+            return local_path, f"✅ Downloaded: {filename}"
+        else:
+            return None, "❌ Download failed"
+    except Exception as e:
+        logger.error(f"Error downloading from Google Drive: {e}")
+        return None, f"❌ Download error: {str(e)}"
+
+# Initialize Google Drive manager
+try:
+    if GOOGLE_DRIVE_AVAILABLE and GoogleDrivePickerManager:
+        # Check if running on HF Space and use secrets
+        oauth_json = os.getenv('OAUTH_CREDENTIALS_JSON')
+        oauth_token = os.getenv('OAUTH_TOKEN_PICKLE')
+        logger.info(f"🔍 Checking for OAuth secrets... Credentials: {oauth_json is not None}, Token: {oauth_token is not None}")
+        
+        if oauth_json:
+            logger.info(f"📝 OAuth credentials length: {len(oauth_json)} characters")
+            # Write the credentials to a temporary file
+            with open('oauth_credentials.json', 'w') as f:
+                f.write(oauth_json)
+            logger.info("✅ OAuth credentials loaded from HF secret and written to file")
+            
+            # Verify file was created
+            if os.path.exists('oauth_credentials.json'):
+                file_size = os.path.getsize('oauth_credentials.json')
+                logger.info(f"✅ oauth_credentials.json created successfully ({file_size} bytes)")
+            else:
+                logger.error("❌ Failed to create oauth_credentials.json file")
+        else:
+            logger.info("ℹ️ No OAuth credentials secret found - checking for local file")
+            if os.path.exists('oauth_credentials.json'):
+                logger.info("✅ Using local oauth_credentials.json file")
+            else:
+                logger.warning("⚠️ No OAuth credentials available (neither secret nor local file)")
+        
+        if oauth_token:
+            import base64
+            logger.info(f"📝 OAuth token length: {len(oauth_token)} characters")
+            # Smart detection: try base64 decode, fallback to raw
+            try:
+                try:
+                    # Try base64 decode first (for HF Spaces secrets)
+                    token_data = base64.b64decode(oauth_token, validate=True)
+                    logger.info("✅ Detected and decoded base64-encoded token")
+                except Exception:
+                    # If base64 fails, treat as raw binary (shouldn't happen in env vars)
+                    logger.info("ℹ️ Not base64, treating as raw token data")
+                    token_data = oauth_token.encode('latin1')
+                
+                with open('oauth_token.pickle', 'wb') as f:
+                    f.write(token_data)
+                logger.info("✅ OAuth token written to file")
+                
+                # Verify file was created
+                if os.path.exists('oauth_token.pickle'):
+                    file_size = os.path.getsize('oauth_token.pickle')
+                    logger.info(f"✅ oauth_token.pickle created successfully ({file_size} bytes)")
+                else:
+                    logger.error("❌ Failed to create oauth_token.pickle file")
+            except Exception as e:
+                logger.error(f"❌ Failed to process OAuth token: {e}")
+                logger.info("💡 Tip: Encode your token with: base64 -i oauth_token.pickle")
+        else:
+            logger.info("ℹ️ No OAuth token secret found - checking for local file")
+            if os.path.exists('oauth_token.pickle'):
+                logger.info("✅ Using local oauth_token.pickle file")
+            else:
+                logger.warning("⚠️ No OAuth token available (neither secret nor local file)")
+        
+        # Set environment variable to disable browser for HF Spaces only if we don't have a token
+        if oauth_json and not oauth_token:
+            os.environ['GOOGLE_DRIVE_HEADLESS'] = 'true'
+            logger.info("🌐 Set headless mode for HF Spaces (no token available)")
+        elif oauth_token:
+            logger.info("🎉 Using pre-authenticated token - browser not needed!")
+        
+        drive_manager = GoogleDrivePickerManager()
+        drive_available = drive_manager.is_available()
+    else:
+        drive_manager = None
+        drive_available = False
+except Exception as e:
+    logger.warning(f"Google Drive initialization failed: {e}")
+    drive_manager = None
+    drive_available = False
+
+# Create the Gradio interface with custom CSS and JS
+custom_css = """
+.video-container video {
+    width: 100%;
+    max-height: 400px;
+}
+.slider-container {
+    margin: 10px 0;
+}
+.drive-section {
+    border: 1px solid #e0e0e0;
+    padding: 15px;
+    border-radius: 8px;
+    margin: 10px 0;
+}
+"""
+
+custom_js = """
+function seekVideo(slider_value, video_id) {
+    const video = document.querySelector('#' + video_id + ' video');
+    if (video && !isNaN(slider_value)) {
+        video.currentTime = slider_value;
+    }
+    return slider_value;
+}
+"""
+
+with gr.Blocks(title="Video Trimmer Tool", theme=gr.themes.Soft(), css=custom_css) as demo:
+    gr.Markdown("""
+    # 🎬 Video Trimmer Demo
+    Upload an MP4 video, set trim points, and generate trimmed video + audio files.
+    """)
+    
+    # Native Google Drive picker section
+    if drive_available:
+        user_email = drive_manager.get_user_info() if drive_manager else "Unknown"
+        with gr.Group():
+            gr.Markdown("### 🔗 Google Drive Integration (Native Picker)")
+            gr.Markdown(f"**👤 Signed in as:** {user_email}")
+            
+            # Video picker section
+            with gr.Row():
+                with gr.Column(scale=2):
+                    gr.Markdown("#### 📁 Load Any File from Google Drive")
+                    
+                    open_picker_btn = gr.Button(
+                        "🌍 Browse Your Entire Google Drive",
+                        variant="primary",
+                        size="lg"
+                    )
+                    
+                    picker_instructions = gr.Textbox(
+                        label="📝 Instructions",
+                        value="Click the button above to open your full Google Drive - browse any folder!",
+                        interactive=False,
+                        lines=6
+                    )
+                    
+                    drive_url_input = gr.Textbox(
+                        label="🔗 Paste Any Google Drive File Link",
+                        placeholder="https://drive.google.com/file/d/FILE_ID/view...",
+                        info="Works with any file type - videos, docs, etc. from any folder"
+                    )
+                    
+                    custom_filename_input = gr.Textbox(
+                        label="🏷️ Custom Filename (Optional)",
+                        placeholder="my_video.mp4"
+                    )
+                    
+                    download_from_url_btn = gr.Button(
+                        "📥 Download Video from Link",
+                        variant="secondary"
+                    )
+                
+                with gr.Column(scale=1):
+                    drive_status = gr.Textbox(
+                        label="📊 Status",
+                        value="✅ Ready to pick from Google Drive",
+                        interactive=False
+                    )
+            
+            # Simplified note
+            gr.Markdown("🚀 **Upload to Google Drive will be available after video trimming.**")
+    else:
+        with gr.Group():
+            gr.Markdown("### 🔗 Google Drive Integration")
+            if not GOOGLE_DRIVE_AVAILABLE:
+                gr.Markdown("**⚠️ Google Drive libraries not installed.**")
+                gr.Markdown("Install with: `pip install google-api-python-client google-auth google-auth-oauthlib`")
+            else:
+                gr.Markdown("**⚠️ Setup needed:** Create oauth_credentials.json file")
+            
+            with gr.Accordion("📋 Setup Instructions", open=False):
+                gr.Markdown(get_native_picker_instructions())
+    
+    with gr.Row():
+        with gr.Column(scale=2):
+            # Video upload and display
+            video_input = gr.File(
+                label="📁 Upload MP4 Video",
+                file_types=[".mp4", ".mov", ".avi", ".mkv"],
+                type="filepath"
+            )
+            
+            video_player = gr.Video(
+                label="🎥 Video Player",
+                show_label=True,
+                elem_id="main_video_player",
+                elem_classes=["video-container"]
+            )
+            
+            video_info = gr.Textbox(
+                label="📊 Video Info",
+                interactive=False,
+                value="Upload a video to see information"
+            )
+        
+        with gr.Column(scale=1):
+            # Trim controls
+            gr.Markdown("### ✂️ Trim Settings")
+            gr.Markdown("**🎯 Drag sliders to set trim points:**")
+            
+            with gr.Group():
+                gr.Markdown("**🎯 Scrub to find start point:**")
+                start_slider = gr.Slider(
+                    minimum=0,
+                    maximum=100,
+                    value=0,
+                    step=0.1,
+                    label="⏯️ Start Time (scrub video)",
+                    info="Drag to seek video and set start position",
+                    elem_classes=["slider-container"]
+                )
+                
+                start_time_display = gr.Textbox(
+                    label="⏯️ Start Time",
+                    value="0:00",
+                    interactive=False,
+                    info="Current start time"
+                )
+            
+            with gr.Group():
+                gr.Markdown("**🎯 Scrub to find end point:**")
+                end_slider = gr.Slider(
+                    minimum=0,
+                    maximum=100,
+                    value=100,
+                    step=0.1,
+                    label="⏹️ End Time (scrub video)",
+                    info="Drag to seek video and set end position",
+                    elem_classes=["slider-container"]
+                )
+                
+                end_time_display = gr.Textbox(
+                    label="⏹️ End Time",
+                    value="1:40",
+                    interactive=False,
+                    info="Current end time"
+                )
+            
+            trim_btn = gr.Button(
+                "✂️ Trim Video",
+                variant="primary",
+                size="lg"
+            )
+            
+            # Note about manual upload
+            gr.Markdown("📝 **Note:** Upload options will appear after trimming is complete.")
+            
+            status_msg = gr.Textbox(
+                label="📝 Status",
+                interactive=False,
+                value="Ready to trim..."
+            )
+    
+    # Output section
+    gr.Markdown("### 📤 Output Files")
+    
+    with gr.Row():
+        with gr.Column():
+            output_video = gr.Video(
+                label="🎬 Trimmed Video",
+                show_label=True
+            )
+        
+        with gr.Column():
+            output_audio_player = gr.Audio(
+                label="🎵 Play Extracted Audio",
+                show_label=True,
+                type="filepath"
+            )
+            
+            output_audio_download = gr.File(
+                label="💾 Download Audio (AAC)",
+                show_label=True
+            )
+    
+    # Post-processing upload section (appears after trimming)
+    if drive_available:
+        with gr.Group(visible=False) as post_upload_section:
+            gr.Markdown("### 🚀 Upload Trimmed Files to Google Drive")
+            
+            with gr.Row():
+                with gr.Column(scale=2):
+                    post_open_folder_btn = gr.Button(
+                        "🌍 Choose Google Drive Upload Folder",
+                        variant="primary"
+                    )
+                    
+                    post_folder_instructions = gr.Textbox(
+                        label="📝 Folder Instructions",
+                        value="Click button above to choose where to upload your trimmed files",
+                        interactive=False,
+                        lines=4
+                    )
+                    
+                    post_upload_folder_url = gr.Textbox(
+                        label="📁 Upload Folder Link",
+                        placeholder="https://drive.google.com/drive/folders/FOLDER_ID...",
+                        info="Leave empty to upload to My Drive root"
+                    )
+                    
+                    post_upload_btn = gr.Button(
+                        "📤 Upload Files to Google Drive",
+                        variant="secondary",
+                        size="lg"
+                    )
+                
+                with gr.Column(scale=1):
+                    post_upload_status = gr.Textbox(
+                        label="📊 Upload Status",
+                        value="Ready to upload",
+                        interactive=False
+                    )
+        
+        # Hidden state to store file paths for post-upload
+        trimmed_video_path = gr.State(None)
+        trimmed_audio_path = gr.State(None)
+    
+    # Event handlers
+    def update_video_and_sliders(video_file):
+        info, duration, start_val, end_val = get_video_info(video_file)
+        return (
+            video_file,  # video_player
+            info,  # video_info
+            gr.Slider(minimum=0, maximum=duration, value=0, step=0.1),  # start_slider
+            gr.Slider(minimum=0, maximum=duration, value=duration, step=0.1),  # end_slider
+            "0:00",  # start_time_display
+            format_time(duration)  # end_time_display
+        )
+    
+    def update_start_display(start_val):
+        return format_time(start_val)
+    
+    def update_end_display(end_val):
+        return format_time(end_val)
+    
+    video_input.change(
+        fn=update_video_and_sliders,
+        inputs=[video_input],
+        outputs=[video_player, video_info, start_slider, end_slider, start_time_display, end_time_display]
+    )
+    
+    def update_start_and_seek(start_val):
+        return format_time(start_val)
+    
+    def update_end_and_seek(end_val):
+        return format_time(end_val)
+    
+    start_slider.change(
+        fn=update_start_and_seek,
+        inputs=[start_slider],
+        outputs=[start_time_display]
+    )
+    
+    end_slider.change(
+        fn=update_end_and_seek,
+        inputs=[end_slider],
+        outputs=[end_time_display]
+    )
+    
+    # Google Drive native picker event handlers
+    if drive_available:
+        # Open file picker (full Google Drive access)
+        open_picker_btn.click(
+            fn=lambda: open_file_picker(drive_manager),
+            outputs=[picker_instructions]
+        )
+        
+        # Download from URL
+        download_from_url_btn.click(
+            fn=lambda url, filename: download_from_drive_url(drive_manager, url, filename),
+            inputs=[drive_url_input, custom_filename_input],
+            outputs=[video_input, drive_status]
+        ).then(
+            fn=update_video_and_sliders,
+            inputs=[video_input],
+            outputs=[video_player, video_info, start_slider, end_slider, start_time_display, end_time_display]
+        )
+        
+        # No pre-upload handlers needed
+        
+        # Post-upload event handlers
+        post_open_folder_btn.click(
+            fn=lambda: open_folder_picker(drive_manager),
+            outputs=[post_folder_instructions]
+        )
+        
+        def post_upload_files(video_path, audio_path, folder_url):
+            if not video_path or not audio_path:
+                return "❌ No files to upload"
+            
+            try:
+                folder_url_clean = folder_url.strip() if folder_url and folder_url.strip() else None
+                
+                video_success, video_result = drive_manager.upload_file_to_folder(video_path, folder_url_clean)
+                audio_success, audio_result = drive_manager.upload_file_to_folder(audio_path, folder_url_clean)
+                
+                if video_success and audio_success:
+                    return f"✅ Files uploaded successfully:\n• {video_result}\n• {audio_result}"
+                elif video_success:
+                    return f"✅ {video_result}\n❌ Audio upload failed: {audio_result}"
+                elif audio_success:
+                    return f"✅ {audio_result}\n❌ Video upload failed: {video_result}"
+                else:
+                    return f"❌ Upload failed:\n• Video: {video_result}\n• Audio: {audio_result}"
+                    
+            except Exception as e:
+                return f"❌ Upload error: {str(e)}"
+        
+        post_upload_btn.click(
+            fn=post_upload_files,
+            inputs=[trimmed_video_path, trimmed_audio_path, post_upload_folder_url],
+            outputs=[post_upload_status]
+        )
+    
+    # Trim button handler with Google Drive upload support
+    if drive_available:
+        # Simplified trim function that shows upload section after completion
+        def trim_and_show_upload(video_file, start_time, end_time):
+            result = process_video_trim(video_file, start_time, end_time)
+            display_video, audio_player, audio_download, status, orig_video, orig_audio = result
+            
+            # Show post-upload section if trimming was successful
+            show_upload = orig_video is not None and orig_audio is not None
+            
+            return (
+                display_video, audio_player, audio_download, status,  # Original outputs
+                orig_video, orig_audio,  # Store paths for post-upload
+                gr.Group(visible=show_upload)  # Show/hide upload section
+            )
+        
+        trim_btn.click(
+            fn=trim_and_show_upload,
+            inputs=[video_input, start_slider, end_slider],
+            outputs=[output_video, output_audio_player, output_audio_download, status_msg, 
+                    trimmed_video_path, trimmed_audio_path, post_upload_section]
+        )
+    else:
+        # No Google Drive available - simple trim only
+        def simple_trim(video_file, start_time, end_time):
+            result = process_video_trim(video_file, start_time, end_time)
+            return result[:4]  # Return only the first 4 outputs
+        
+        trim_btn.click(
+            fn=simple_trim,
+            inputs=[video_input, start_slider, end_slider],
+            outputs=[output_video, output_audio_player, output_audio_download, status_msg]
+        )
+
+if __name__ == "__main__":
+    demo.launch(
+        server_name="0.0.0.0",
+        server_port=None,  # Auto-find available port
+        share=False,
+        show_error=True,
+        debug=True
+    )