Update app.py

app.py

@@ -69,6 +69,8 @@ print(f"📁 Base directory: {BASE_DIR}")
 print(f"📂 Temp directory: {TEMP_DIR}")
 print(f"📤 Output directory: {OUTPUT_DIR}")
 print(f"🎯 Assets directory: {ASSETS_DIR}")
+print(f"📁 Convert directory: {CONVERT_DIR}")
+
 try:
     from SwitcherAI.utilities import set_temp_directory, get_temp_directory_info
     
@@ -89,6 +91,7 @@ try:
 except ImportError as e:
     print(f"⚠️ Could not import SwitcherAI utilities: {e}")
     print("🔄 Using default temp directory behavior")
+
 # Download required model files
 def download_required_models():
     """Download required model files if not present"""
@@ -219,6 +222,56 @@ def safe_copy_file(source, destination):
         print(f"❌ Copy error: {e}")
         return False
 
+def handle_batch_file_upload(files):
+    """Handle multiple file uploads for batch mode"""
+    if not files:
+        return "📁 No files uploaded"
+    
+    # Clear existing files in convert directory
+    for existing_file in CONVERT_DIR.glob("*"):
+        if existing_file.is_file():
+            existing_file.unlink()
+    
+    uploaded_count = 0
+    failed_count = 0
+    
+    for file in files:
+        try:
+            if file is None:
+                continue
+            
+            # Get the original filename
+            original_name = Path(file.name).name if hasattr(file, 'name') else f"video_{uploaded_count}.mp4"
+            
+            # Copy file to convert directory
+            dest_path = CONVERT_DIR / original_name
+            shutil.copy2(file, dest_path)
+            
+            if dest_path.exists() and dest_path.stat().st_size > 0:
+                file_size = dest_path.stat().st_size / (1024 * 1024)  # MB
+                print(f"✅ Uploaded: {original_name} ({file_size:.1f}MB)")
+                uploaded_count += 1
+            else:
+                print(f"❌ Failed to upload: {original_name}")
+                failed_count += 1
+                
+        except Exception as e:
+            print(f"❌ Error uploading file: {e}")
+            failed_count += 1
+    
+    status_msg = f"📦 Batch Upload Complete:\n✅ Uploaded: {uploaded_count} files\n"
+    if failed_count > 0:
+        status_msg += f"❌ Failed: {failed_count} files\n"
+    
+    # List uploaded files
+    uploaded_files = [f.name for f in CONVERT_DIR.glob("*.mp4")] + [f.name for f in CONVERT_DIR.glob("*.avi")] + [f.name for f in CONVERT_DIR.glob("*.mov")]
+    if uploaded_files:
+        status_msg += f"📁 Files ready for processing:\n" + "\n".join([f"  • {f}" for f in uploaded_files[:10]])
+        if len(uploaded_files) > 10:
+            status_msg += f"\n  ... and {len(uploaded_files) - 10} more"
+    
+    return status_msg
+
 def resize_video(input_path, output_path, fps=30):
     """Resize/process video with fallback"""
     if not MOVIEPY_AVAILABLE:
@@ -265,6 +318,16 @@ def cleanup_temp_files():
     except Exception as e:
         print(f"⚠️ Cleanup error: {e}")
 
+def cleanup_convert_files():
+    """Clean up convert directory files"""
+    try:
+        for file in CONVERT_DIR.glob("*"):
+            if file.is_file():
+                file.unlink()
+        print("🧹 Convert directory cleaned")
+    except Exception as e:
+        print(f"⚠️ Convert cleanup error: {e}")
+
 def create_batch_zip():
     """Create zip file of all output files"""
     try:
@@ -423,7 +486,7 @@ def run_batch_processing(source_image, frame_processor, face_analyser_direction,
     video_files = list(CONVERT_DIR.glob("*.mp4")) + list(CONVERT_DIR.glob("*.avi")) + list(CONVERT_DIR.glob("*.mov"))
     
     if not video_files:
-        yield None, f"📁 No video files found in {CONVERT_DIR}"
+        yield None, f"📁 No video files found in Convert folder.\nPlease upload videos using the file input above."
         return
     
     temp_source = TEMP_DIR / 'source-image.jpg'
@@ -432,7 +495,7 @@ def run_batch_processing(source_image, frame_processor, face_analyser_direction,
         return
     
     source_name = Path(source_image).stem
-    cli_output = f"📊 Processing {len(video_files)} videos\n🎯 Source: {source_name}\n\n"
+    cli_output = f"📊 Processing {len(video_files)} videos in batch mode\n🎯 Source: {source_name}\n\n"
     yield None, cli_output
     
     successful = 0
@@ -586,6 +649,7 @@ def cancel_processing():
 def reset_interface():
     """Reset interface to defaults"""
     cleanup_temp_files()
+    cleanup_convert_files()
     return (
         None,  # source_image
         None,  # target_video
@@ -622,6 +686,33 @@ def handle_action_button(button_text, *inputs):
     else:
         return inputs + ("", button_text)
 
+def toggle_batch_mode(use_folder_mode):
+    """Handle batch mode toggle"""
+    if use_folder_mode:
+        return gr.update(
+            label="📁 Target Videos (Drag multiple files here)",
+            file_count="multiple",
+            file_types=["video"]
+        )
+    else:
+        return gr.update(
+            label="Target Video (Video to modify)", 
+            file_count="single",
+            file_types=["video"]
+        )
+
+def handle_file_upload(files, use_folder_mode):
+    """Handle file uploads - single or multiple"""
+    if use_folder_mode and files:
+        # Handle batch upload
+        status = handle_batch_file_upload(files)
+        return status
+    elif not use_folder_mode and files:
+        # Single file mode - just return status
+        return f"✅ Single video uploaded: {Path(files.name).name if hasattr(files, 'name') else 'video file'}"
+    else:
+        return "📁 No files uploaded"
+
 # Initialize GPU detection
 AVAILABLE_GPUS = get_available_gpus()
 print(f"🖥️ Available GPUs: {AVAILABLE_GPUS}")
@@ -661,11 +752,25 @@ def create_interface():
                         file_count="single"
                     )
                     
+                    # Batch mode toggle
+                    use_folder_mode = gr.Checkbox(
+                        label="📁 Batch Mode (Process multiple videos)",
+                        value=False
+                    )
+                    
                     target_video = gr.File(
                         label="Target Video (Video to modify)",
                         file_types=["video"],
                         file_count="single"
                     )
+                    
+                    # Upload status display
+                    upload_status = gr.Textbox(
+                        label="Upload Status",
+                        value="Ready to upload files...",
+                        interactive=False,
+                        lines=3
+                    )
                 
                 with gr.Group(elem_classes="control-panel"):
                     gr.HTML('<div class="section-header">🎮 Controls</div>')
@@ -766,11 +871,6 @@ def create_interface():
                             skip_audio = gr.Checkbox(label="🔇 Skip Audio", value=False)
                         
                         with gr.Column():
-                            use_folder_mode = gr.Checkbox(
-                                label="📁 Batch Mode (Convert folder)",
-                                value=False
-                            )
-                            
                             keep_fps = gr.Checkbox(label="🎬 Keep Original FPS", value=True)
         
         # Event handlers
@@ -781,11 +881,17 @@ def create_interface():
         )
         
         use_folder_mode.change(
-            lambda x: gr.update(visible=not x),
+            toggle_batch_mode,
             inputs=[use_folder_mode],
             outputs=[target_video]
         )
         
+        target_video.upload(
+            handle_file_upload,
+            inputs=[target_video, use_folder_mode],
+            outputs=[upload_status]
+        )
+        
         start_button.click(
             handle_processing,
             inputs=[