Update app.py

app.py

@@ -6,6 +6,7 @@ import shutil
 import subprocess
 import time
 import argparse
+import zipfile
 
 # --- LOAD ENGINE ---
 try:
@@ -21,12 +22,8 @@ except ImportError:
 
 # --- CSS & THEME CONFIGURATION ---
 custom_css = """
-body {
-    background-color: #0b0f19;
-}
-.gradio-container {
-    background-color: #0b0f19 !important;
-}
+body { background-color: #0b0f19; }
+.gradio-container { background-color: #0b0f19 !important; }
 h1.title-header {
     color: #00ff41;
     font-family: 'Courier New', Courier, monospace;
@@ -61,6 +58,28 @@ shinyy_theme = gr.themes.Soft(
     button_primary_text_color="#000000",
 )
 
+# --- UTILITY: ZIP PROJECT ---
+def zip_project_directory():
+    """Zips the current working directory for download."""
+    zip_filename = "project_backup.zip"
+    # Files/Folders to ignore to keep zip clean/small
+    ignore_folders = {'.git', '__pycache__', 'venv', 'env', '.ipynb_checkpoints'}
+    
+    with zipfile.ZipFile(zip_filename, 'w', zipfile.ZIP_DEFLATED) as zipf:
+        for root, dirs, files in os.walk('.'):
+            # Remove ignored directories from traversal
+            dirs[:] = [d for d in dirs if d not in ignore_folders]
+            
+            for file in files:
+                if file == zip_filename: continue # Don't zip the zip itself
+                
+                file_path = os.path.join(root, file)
+                # Archive name is the path relative to current directory
+                arcname = os.path.relpath(file_path, '.')
+                zipf.write(file_path, arcname)
+                
+    return zip_filename
+
 # --- LOGIC FUNCTIONS ---
 
 def swap_single_photo(src_img, src_idx, dst_img, dst_idx, include_hair, progress=gr.Progress(track_tqdm=True)):
@@ -79,30 +98,25 @@ def swap_single_photo(src_img, src_idx, dst_img, dst_idx, include_hair, progress
         os.makedirs(os.path.dirname(dst_path), exist_ok=True)
         os.makedirs(os.path.dirname(output_path), exist_ok=True)
         
-        # Save images
         cv2.imwrite(src_path, cv2.cvtColor(src_img, cv2.COLOR_RGB2BGR))
         cv2.imwrite(dst_path, cv2.cvtColor(dst_img, cv2.COLOR_RGB2BGR))
         
         progress(0.5, desc="Swapping faces" + (" & hair" if include_hair else ""))
         
-        # --- CRITICAL UPDATE: PASSING THE HAIR ARGUMENT ---
+        # --- ENGINE CALL ---
         result = swapper.swap_faces(src_path, int(src_idx), dst_path, int(dst_idx), swap_hair=include_hair)
         
         cv2.imwrite(output_path, result)
         log += f"Swapped{' (with Hair)' if include_hair else ''} and saved.\n"
         
         progress(1, desc="Done")
-        
-        # Convert BGR back to RGB for Gradio display
         result_rgb = cv2.cvtColor(result, cv2.COLOR_BGR2RGB)
         return result_rgb, log
     except Exception as e:
         log += f"Error: {e}\n"
         return None, log
 
-# --- REMAINING LOGIC FUNCTIONS (Standard) ---
 def swap_single_src_multi_dst(src_img, dst_imgs, dst_indices, progress=gr.Progress(track_tqdm=True)):
-    # Standard implementation (hair swap disabled for batch stability)
     log = ""
     results = []
     src_dir = "SingleSrcMultiDst/src"
@@ -129,7 +143,6 @@ def swap_single_src_multi_dst(src_img, dst_imgs, dst_indices, progress=gr.Progre
         dst_path = os.path.join(dst_dir, f"data_dst_{j}.jpg")
         output_path = os.path.join(output_dir, f"output_swapped_{j}.jpg")
         cv2.imwrite(dst_path, cv2.cvtColor(dst_img, cv2.COLOR_RGB2BGR))
-        
         try:
             dst_idx = dst_indices_list[j] if j < len(dst_indices_list) else 1
             result = swapper.swap_faces(src_path, 1, dst_path, int(dst_idx))
@@ -347,7 +360,7 @@ with gr.Blocks(theme=shinyy_theme, css=custom_css, title="Shinyy's Face Swapper"
     gr.Markdown("""<h1 class="title-header">s h i n y y ' s &nbsp;&nbsp; Face Swapping Tool</h1>""")
     
     with gr.Tabs():
-        # TAB 1: Single Photo (UPDATED)
+        # TAB 1: Single Photo
         with gr.TabItem("📸 Single Photo"):
             with gr.Row():
                 with gr.Column(variant="panel"):
@@ -360,7 +373,6 @@ with gr.Blocks(theme=shinyy_theme, css=custom_css, title="Shinyy's Face Swapper"
                     sp_dst = gr.Image(label="Target Image", type="numpy", height=300)
                     sp_dst_idx = gr.Number(value=1, label="Face Index", precision=0)
             
-            # --- NEW CHECKBOX ---
             with gr.Row():
                 sp_hair = gr.Checkbox(label="Swap Hair (Beta)", value=False, 
                                       info="Warps source hair onto target.")
@@ -451,6 +463,16 @@ with gr.Blocks(theme=shinyy_theme, css=custom_css, title="Shinyy's Face Swapper"
                     va_out = gr.Video(label="Result")
                     va_btn.click(swap_video_all_faces, inputs=[va_src, va_target, va_count, vid_del, vid_audio, vid_drive], outputs=[va_out, vid_log])
 
+        # TAB 7: DOWNLOAD
+        with gr.TabItem("💾 Download Source Code"):
+            gr.Markdown("### Backup & Export")
+            gr.Markdown("Click the button below to zip the entire project folder (excluding models and cache) so you can run it on your local PC.")
+            
+            dl_btn = gr.Button("📦 Zip & Download Project", variant="primary")
+            dl_file = gr.File(label="Download Zip")
+            
+            dl_btn.click(zip_project_directory, inputs=[], outputs=[dl_file])
+
     gr.Markdown("---")
     gr.Markdown("### s h i n y y ' s   S u i t e   v 2 . 1")