Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import roop.globals
 import roop.core
 import os
+import shutil
 
 # ---------------------------
 # GLOBAL CONFIG (RUNS ONCE)
@@ -25,6 +26,25 @@ roop.globals.max_memory = 8  # limit memory for stability
 
 print("🔥 Roop initialized (CPU optimized)")
 
+def get_valid_image_path(file_obj, suffix=""):
+    if file_obj is None:
+        return None
+    
+    # Check if API sent string, dict, or file object
+    if isinstance(file_obj, str):
+        path = file_obj
+    elif isinstance(file_obj, dict) and 'path' in file_obj:
+        path = file_obj['path']
+    else:
+        path = getattr(file_obj, 'name', str(file_obj))
+        
+    # Agari extension missing hai API ki wajah se, usko .png dedo! (Bug Fixed Here)
+    if not any(path.lower().endswith(ext) for ext in ['.jpg', '.jpeg', '.png', '.bmp', '.webp']):
+        new_path = path + suffix + ".png"
+        shutil.copy(path, new_path)
+        return new_path
+    return path
+
 # ---------------------------
 # MAIN FUNCTION
 # ---------------------------
@@ -32,13 +52,9 @@ def swap_face(source_file, target_file, enable_enhancer):
     if source_file is None or target_file is None:
         return None
 
-    source_path = source_file if isinstance(source_file, str) else source_file.name
-    target_path = target_file if isinstance(target_file, str) else target_file.name
-
-    # Allow only images
-    _, target_ext = os.path.splitext(target_path)
-    if target_ext.lower() not in ['.jpg', '.jpeg', '.png', '.bmp', '.webp']:
-        raise gr.Error("❌ Only image files allowed (CPU space limitation).")
+    # Ab humein farq nahi padta kon kaisa image bhej raha hai!
+    source_path = get_valid_image_path(source_file, "_src")
+    target_path = get_valid_image_path(target_file, "_tgt")
 
     output_path = os.path.join(os.getcwd(), "output.png")
 
@@ -55,7 +71,7 @@ def swap_face(source_file, target_file, enable_enhancer):
     roop.globals.target_path = target_path
     roop.globals.output_path = output_path
 
-    # Processors config (NO RESET BUG)
+    # Processors config
     processors = ['face_swapper']
     if enable_enhancer:
         processors.append('face_enhancer')
@@ -99,7 +115,7 @@ with gr.Blocks(title="AI Guruji Face Swap") as demo:
             submit_btn = gr.Button("🚀 Swap Face", variant="primary")
 
         with gr.Column():
-            result_image = gr.Image(label="Result")
+            result_image = gr.Image(label="Result", type="filepath")
 
     submit_btn.click(
         fn=swap_face,
@@ -112,4 +128,4 @@ with gr.Blocks(title="AI Guruji Face Swap") as demo:
 # ---------------------------
 if __name__ == "__main__":
     demo.queue(max_size=10)  # Prevent overload
-    demo.launch()
+    demo.launch()