Update main.py

main.py

@@ -7,9 +7,11 @@ from huggingface_hub import create_repo, upload_file
 app = Flask(__name__)
 
 # Função para chamar a API de hospedagem de imagens
-def host_image(image_url):
-    api_url = "https://wosocial.bubbleapps.io/version-test/api/1.1/wf/save?file=" + image_url
-    response = requests.get(api_url)
+def host_image(endpoint, image_path):
+    api_url = "https://wosocial.bubbleapps.io/version-test/api/1.1/wf/save"
+    image_url = f"{endpoint}/file={image_path}"
+    files = {'file': open(image_url, 'rb')}
+    response = requests.post(api_url, files=files)
     if response.status_code == 200:
         return response.json()["response"]["result"]
     else:
@@ -93,11 +95,16 @@ def faceswapper():
     user_photo = request.args.get('user_photo', default='')
     result_photo = request.args.get('result_photo', default='')
 
-    # Formar a URL completa
-    full_url = endpoint + "/file=" + result_photo
+    # Chamar a API Gradio
+    client = Client(endpoint, upload_files=True)
+    result_path = client.predict(
+        user_photo,
+        result_photo,
+        api_name="/predict"
+    )
 
     # Hospedar a imagem e obter a URL
-    hosted_url = host_image(full_url)
+    hosted_url = host_image(endpoint, result_path)
 
     if hosted_url:
         return jsonify({"result_url": hosted_url})