Update main.py

main.py

@@ -6,10 +6,10 @@ from huggingface_hub import create_repo, upload_file
 
 app = Flask(__name__)
 
-def host_image(image_path):
-    api_url = "https://wosocial.bubbleapps.io/version-test/api/1.1/wf/save"
-    files = {'file': open(image_path, 'rb')}
-    response = requests.post(api_url, files=files)
+# 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)
     if response.status_code == 200:
         return response.json()["response"]["result"]
     else:
@@ -92,17 +92,12 @@ def faceswapper():
     endpoint = request.args.get('endpoint', default='https://pierroromeu-faceswapper.hf.space/--replicas/u42x7/')
     user_photo = request.args.get('user_photo', default='')
     result_photo = request.args.get('result_photo', default='')
-  
-    # Chamar a API Gradio
-    client = Client(endpoint, upload_files=True)
-    result_path = client.predict(
-        user_photo,
-        result_photo,
-        api_name="/predict"
-    )
+
+    # Formar a URL completa
+    full_url = endpoint + "/file=" + result_photo
 
     # Hospedar a imagem e obter a URL
-    hosted_url = host_image(result_path)
+    hosted_url = host_image(full_url)
 
     if hosted_url:
         return jsonify({"result_url": hosted_url})