Update app.py

app.py

@@ -3,8 +3,11 @@ import os
 import sys
 import base64
 import uuid
+from fastapi import FastAPI
 
-# --- Purana Code (Koi Change Nahi) ---
+# -------------------------------------------------------------------
+#  PART 1: CORE LOGIC (Yeh bilkul perfect hai, ismein koi change nahi)
+# -------------------------------------------------------------------
 sys.path.append('roop')
 from roop.core import run as roop_run
 
@@ -18,30 +21,33 @@ def face_swap_logic(source_path, target_path):
         "--output", output_path, "--execution-provider", "cpu",
     ]
     sys.argv = args
-    print(f"Running roop with args: {sys.argv}")
-
+    
     try:
         roop_run()
     except SystemExit: pass
     except Exception as e:
-        print(f"Error during face swap: {e}")
+        print(f"Face swap fail hua. Error: {str(e)}")
         return None
 
     if not os.path.exists(output_path):
         return None
-        
+    
     return output_path
 
-# --- Gradio UI (Yeh Web Users ke liye hai) ---
-with gr.Blocks(theme=gr.themes.Soft()) as app: # Yahan 'app' ek Gradio Blocks object hai
+# -------------------------------------------------------------------
+#  PART 2: GRADIO UI DEFINITION (Aapka Web Interface)
+# -------------------------------------------------------------------
+with gr.Blocks(theme=gr.themes.Soft()) as gradio_ui:
     gr.Markdown("# 🎭 Roop Face Swap AI")
     with gr.Tabs():
         with gr.TabItem("🖼️ Swap on Image"):
+            # ... (UI ka baaki code bilkul same)
             with gr.Row():
                 source_face_img = gr.Image(label="Source Face", type="filepath")
                 target_image = gr.Image(label="Target Image", type="filepath")
             submit_btn_image = gr.Button("Swap Face on Image")
             result_image = gr.Image(label="Result Image")
+
         with gr.TabItem("🎬 Swap on Video"):
             with gr.Row():
                 source_face_vid = gr.Image(label="Source Face", type="filepath")
@@ -49,16 +55,15 @@ with gr.Blocks(theme=gr.themes.Soft()) as app: # Yahan 'app' ek Gradio Blocks ob
             submit_btn_video = gr.Button("Swap Face on Video")
             result_video = gr.Video(label="Result Video")
 
-    def web_ui_swap(source, target):
-        return face_swap_logic(source, target)
-
-    submit_btn_image.click(fn=web_ui_swap, inputs=[source_face_img, target_image], outputs=result_image)
-    submit_btn_video.click(fn=web_ui_swap, inputs=[source_face_vid, target_video], outputs=result_video)
+    # UI ke buttons ko logic se jodna
+    submit_btn_image.click(fn=face_swap_logic, inputs=[source_face_img, target_image], outputs=result_image)
+    submit_btn_video.click(fn=face_swap_logic, inputs=[source_face_vid, target_video], outputs=result_video)
 
-
-# --- Naya API Code (Yeh Aapke Android App ke liye hai) ---
-# Hum FastAPI ko alag se banane ke bajaye, seedha Gradio ke chalte hue server (app.app) mein hi apna API endpoint jod denge.
-# Yeh sabse saaf aur sahi tarika hai.
+# -------------------------------------------------------------------
+#  PART 3: FASTAPI APP DEFINITION (Aapke Android App ke liye)
+# -------------------------------------------------------------------
+# Yahan hum main app banayenge jo Hugging Face chalaayega
+app = FastAPI()
 
 def save_base64_to_file(b64_string, extension):
     filename = f"/tmp/{uuid.uuid4().hex}.{extension}"
@@ -66,7 +71,8 @@ def save_base64_to_file(b64_string, extension):
         f.write(base64.b64decode(b64_string))
     return filename
 
-@app.post("/api/swap-face") # Yahan 'app' Gradio ka Blocks object hai
+# API ka darwaza
+@app.post("/api/swap-face")
 async def api_face_swap(payload: dict):
     source_b64 = payload.get("source_face_b64")
     target_b64 = payload.get("target_media_b64")
@@ -87,10 +93,14 @@ async def api_face_swap(payload: dict):
         os.remove(source_file_path)
         os.remove(target_file_path)
         os.remove(result_path)
-
         return {"result_b64": result_b64}
     else:
         return {"error": "Face swap fail hua."}
 
-# AAKHRI AUR SABSE ZAROORI LINE - APP KO LAUNCH KARNA
-app.launch()
+# -------------------------------------------------------------------
+#  PART 4: MOUNTING (Dono ko Jodna)
+# -------------------------------------------------------------------
+# Yeh aakhri line Gradio UI (gradio_ui) ko FastAPI app (app) ke upar mount karti hai
+app = gr.mount_gradio_app(app, gradio_ui, path="/")
+
+# Ismein app.launch() ki zaroorat nahi hoti. Hugging Face khud 'app' object ko chala leta hai.