old gradio code

app.py

@@ -1,18 +1,13 @@
 import os
 os.environ["OMP_NUM_THREADS"] = "1"
-
+import gradio as gr
 import cv2
 import shutil
 import uuid
-import subprocess
-import numpy as np
 import insightface
 from insightface.app import FaceAnalysis
 from huggingface_hub import hf_hub_download
-from fastapi import FastAPI, UploadFile, File
-from fastapi.responses import StreamingResponse
-import io
-import gradio as gr
+import subprocess
 
 # -------------------------------------------------
 # Paths
@@ -54,8 +49,8 @@ for f in buffalo_files:
 # -------------------------------------------------
 # Initialize face analysis and swapper
 # -------------------------------------------------
-app_face = FaceAnalysis(name="buffalo_l", root=MODELS_DIR, providers=['CPUExecutionProvider'])
-app_face.prepare(ctx_id=0, det_size=(640, 640))
+app = FaceAnalysis(name="buffalo_l", root=MODELS_DIR, providers=['CPUExecutionProvider'])
+app.prepare(ctx_id=0, det_size=(640, 640))
 swapper = insightface.model_zoo.get_model(inswapper_path, providers=['CPUExecutionProvider'])
 
 # -------------------------------------------------
@@ -76,108 +71,74 @@ ensure_codeformer()
 # -------------------------------------------------
 # Pipeline Function
 # -------------------------------------------------
+# -------------------------------------------------
+# Pipeline Function (defaults applied)
+# -------------------------------------------------
 def face_swap_and_enhance(src_img, tgt_img):
     try:
         src_bgr = cv2.cvtColor(src_img, cv2.COLOR_RGB2BGR)
         tgt_bgr = cv2.cvtColor(tgt_img, cv2.COLOR_RGB2BGR)
 
-        src_faces = app_face.get(src_bgr)
-        tgt_faces = app_face.get(tgt_bgr)
+        src_faces = app.get(src_bgr)
+        tgt_faces = app.get(tgt_bgr)
         if not src_faces or not tgt_faces:
-            return None, "❌ Face not detected in one of the images."
+            return None, None, "❌ Face not detected in one of the images."
 
-        # clean dirs
         shutil.rmtree(UPLOAD_DIR, ignore_errors=True)
         shutil.rmtree(RESULT_DIR, ignore_errors=True)
         os.makedirs(UPLOAD_DIR, exist_ok=True)
         os.makedirs(RESULT_DIR, exist_ok=True)
 
-        # save swapped
+        # ---------------- Save swapped face ----------------
         unique_name = f"swapped_{uuid.uuid4().hex[:8]}.jpg"
         swapped_path = os.path.join(UPLOAD_DIR, unique_name)
         swapped_bgr = swapper.get(tgt_bgr, tgt_faces[0], src_faces[0])
         cv2.imwrite(swapped_path, swapped_bgr)
 
-        # run CodeFormer
+        # ---------------- Run CodeFormer ----------------
         cmd = f"python {CODEFORMER_PATH} -w 0.7 --input_path {swapped_path} --output_path {RESULT_DIR} --bg_upsampler realesrgan --face_upsample"
         result = subprocess.run(cmd, shell=True, capture_output=True, text=True)
         if result.returncode != 0:
-            return None, f"❌ CodeFormer failed:\n{result.stderr}"
+            return None, None, f"❌ CodeFormer failed:\n{result.stderr}"
 
+        # ---------------- Locate final result ----------------
         final_results_dir = os.path.join(RESULT_DIR, "final_results")
         if not os.path.exists(final_results_dir):
-            return None, "❌ CodeFormer did not produce final results."
+            return None, None, "❌ CodeFormer did not produce final results."
 
         final_files = [f for f in os.listdir(final_results_dir) if f.endswith(".png")]
         if not final_files:
-            return None, "❌ No enhanced image found in final results."
+            return None, None, "❌ No enhanced image found in final results."
 
         final_path = os.path.join(final_results_dir, final_files[0])
         final_img = cv2.cvtColor(cv2.imread(final_path), cv2.COLOR_BGR2RGB)
 
-        return final_img, None
+        return final_img, final_path, ""
 
     except Exception as e:
-        return None, f"❌ Error: {str(e)}"
-
-# -------------------------------------------------
-# FastAPI REST API
-# -------------------------------------------------
-api = FastAPI()
-# Add this to your app.py in the FastAPI section
-@api.get("/api/health")
-async def health_check():
-    return {"status": "healthy", "message": "FaceSwap API is running"} 
-
-@api.post("/api/faceswap")
-async def faceswap(src_img: UploadFile = File(...), tgt_img: UploadFile = File(...)):
-    """
-    REST API endpoint for face swapping.
-    Accepts source and target images as file uploads.
-    Returns enhanced face-swapped image as PNG.
-    """
-    try:
-        # Read images from uploads
-        src = cv2.imdecode(np.frombuffer(await src_img.read(), np.uint8), cv2.IMREAD_COLOR)
-        tgt = cv2.imdecode(np.frombuffer(await tgt_img.read(), np.uint8), cv2.IMREAD_COLOR)
-
-        # Run face swap + CodeFormer enhancement
-        result_img, err = face_swap_and_enhance(
-            cv2.cvtColor(src, cv2.COLOR_BGR2RGB),
-            cv2.cvtColor(tgt, cv2.COLOR_BGR2RGB)
-        )
+        return None, None, f"❌ Error: {str(e)}"
 
-        if result_img is None:
-            return {"error": err}
 
-        # Encode as PNG and return
-        _, buffer = cv2.imencode(".png", cv2.cvtColor(result_img, cv2.COLOR_RGB2BGR))
-        return StreamingResponse(io.BytesIO(buffer.tobytes()), media_type="image/png")
-
-    except Exception as e:
-        return {"error": f"Exception: {str(e)}"}
 # -------------------------------------------------
-# Gradio UI
+# Gradio Interface (simplified)
 # -------------------------------------------------
 with gr.Blocks() as demo:
-    gr.Markdown("### Face Swap + CodeFormer Enhancement")
+    gr.Markdown("Face Swap")
 
     with gr.Row():
         src_input = gr.Image(type="numpy", label="Upload Your Face")
         tgt_input = gr.Image(type="numpy", label="Upload Target Image")
 
     btn = gr.Button("Swap Face")
+
     output_img = gr.Image(type="numpy", label="Enhanced Output")
+    download = gr.File(label="⬇️ Download Enhanced Image")
     error_box = gr.Textbox(label="Logs / Errors", interactive=False)
 
     def process(src, tgt):
-        img, err = face_swap_and_enhance(src, tgt)
-        return img, err
-
-    btn.click(process, [src_input, tgt_input], [output_img, error_box])
+        img, path, err = face_swap_and_enhance(src, tgt)
+        return img, path, err
 
-# Mount FastAPI into Gradio
-demo.app = api
+    btn.click(process, [src_input, tgt_input], [output_img, download, error_box])
 
-# Launch (only once)
-demo.launch(server_name="0.0.0.0", server_port=7860)
+demo.launch(share=True)