added app.py

app.py

@@ -71,8 +71,8 @@ inswapper_path = download_models()
 # -------------------------------------------------
 providers = ['CUDAExecutionProvider', 'CPUExecutionProvider']
 logger.info(f"Initializing FaceAnalysis with providers: {providers}")
-app = FaceAnalysis(name="buffalo_l", root=MODELS_DIR, providers=providers)
-app.prepare(ctx_id=0, det_size=(640, 640))
+face_analysis_app = FaceAnalysis(name="buffalo_l", root=MODELS_DIR, providers=providers)
+face_analysis_app.prepare(ctx_id=0, det_size=(640, 640))
 swapper = insightface.model_zoo.get_model(inswapper_path, providers=providers)
 logger.info("FaceAnalysis and swapper initialized")
 
@@ -132,8 +132,8 @@ def face_swap_and_enhance(src_img, tgt_img):
             tgt_bgr = cv2.cvtColor(tgt_img, cv2.COLOR_RGB2BGR)
 
             logger.info("Detecting faces...")
-            src_faces = app.get(src_bgr)
-            tgt_faces = app.get(tgt_bgr)
+            src_faces = face_analysis_app.get(src_bgr)
+            tgt_faces = face_analysis_app.get(tgt_bgr)
             if not src_faces or not tgt_faces:
                 logger.error("Face not detected in one of the images")
                 return None, None, "❌ Face not detected in one of the images"
@@ -201,17 +201,17 @@ with gr.Blocks() as demo:
 # -------------------------------------------------
 # FastAPI App
 # -------------------------------------------------
-app = FastAPI()
+fastapi_app = FastAPI()
 
-@app.get("/")
+@fastapi_app.get("/")
 def root():
     return RedirectResponse("/gradio")
 
-@app.get("/health")
+@fastapi_app.get("/health")
 async def health():
     return {"status": "healthy"}
 
-@app.post("/source")
+@fastapi_app.post("/source")
 async def upload_source(image: UploadFile = File(...)):
     logger.info(f"Uploading source image: {image.filename}")
     contents = await image.read()
@@ -224,7 +224,7 @@ async def upload_source(image: UploadFile = File(...)):
     logger.info(f"Source image uploaded with ID: {str(result.inserted_id)}")
     return {"source_id": str(result.inserted_id)}
 
-@app.post("/target")
+@fastapi_app.post("/target")
 async def upload_target(image: UploadFile = File(...)):
     logger.info(f"Uploading target image: {image.filename}")
     contents = await image.read()
@@ -241,7 +241,7 @@ class FaceSwapRequest(BaseModel):
     source_id: str
     target_id: str
 
-@app.post("/faceswap")
+@fastapi_app.post("/faceswap")
 async def perform_faceswap(request: FaceSwapRequest):
     logger.info(f"Starting face swap for source_id: {request.source_id}, target_id: {request.target_id}")
     source_doc = await source_images_collection.find_one({"_id": ObjectId(request.source_id)})
@@ -287,7 +287,7 @@ async def perform_faceswap(request: FaceSwapRequest):
     logger.info(f"Face swap result stored with ID: {str(result.inserted_id)}")
     return {"result_id": str(result.inserted_id)}
 
-@app.get("/download/{result_id}")
+@fastapi_app.get("/download/{result_id}")
 async def download_result(result_id: str):
     logger.info(f"Downloading result: {result_id}")
     doc = await results_collection.find_one({"_id": ObjectId(result_id)})
@@ -301,8 +301,8 @@ async def download_result(result_id: str):
     )
 
 # Mount Gradio at /gradio
-app = mount_gradio_app(app, demo, path="/gradio")
+fastapi_app = mount_gradio_app(fastapi_app, demo, path="/gradio")
 
 # Run the app with Uvicorn
 if __name__ == "__main__":
-    uvicorn.run(app, host="0.0.0.0", port=7860)
+    uvicorn.run(fastapi_app, host="0.0.0.0", port=7860)

test.py

@@ -0,0 +1,33 @@
+# import requests
+
+# response = requests.get("https://logicgoinfotechspaces-faceswap.hf.space/health")
+# print(response.status_code)  # Should be 200
+# print(response.json())  # Should print: {'status': 'healthy'}
+
+# import requests
+
+# with open("./source.jpeg", "rb") as f:
+#     files = {"image": ("./source.jpeg", f, "image/jpeg")}
+#     response = requests.post("https://logicgoinfotechspaces-faceswap.hf.space/source", files=files)
+#     print(response.status_code)  # Should be 200
+#     print(response.json())  # Should print: {'source_id': '<id>'}
+#     source_id = response.json().get("source_id")
+
+# import requests
+
+# with open("./target.jpeg", "rb") as f:
+#     files = {"image": ("./target.jpeg", f, "image/jpeg")}
+#     response = requests.post("https://logicgoinfotechspaces-faceswap.hf.space/target", files=files)
+#     print(response.status_code)  # Should be 200
+#     print(response.json())  # Should print: {'target_id': '<id>'}
+#     target_id = response.json().get("target_id")
+
+import requests
+
+source_id = "68bff17eb3b2e06c24cafa22"  # Replace with actual ID
+target_id = "68bff1a1b3b2e06c24cafa23"  # Replace with actual ID
+payload = {"source_id": source_id, "target_id": target_id}
+response = requests.post("https://logicgoinfotechspaces-faceswap.hf.space/faceswap", json=payload)
+print(response.status_code)  # Should be 200
+print(response.json())  # Should print: {'result_id': '<id>'}
+result_id = response.json().get("result_id")