Update main.py

main.py

@@ -6,6 +6,16 @@ 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_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)
+    if response.status_code == 200:
+        return response.json()["response"]["result"]
+    else:
+        return None
+        
 @app.route('/run', methods=['POST'])
 def run_model():
       # Obter parâmetros da consulta da URL
@@ -92,7 +102,13 @@ def faceswapper():
         api_name="/predict"
     )
 
-    return jsonify(result)
+    # Hospedar a imagem e obter a URL
+    hosted_url = host_image(result_path)
+
+    if hosted_url:
+        return jsonify({"result_url": hosted_url})
+    else:
+        return jsonify({"error": "Falha ao hospedar a imagem."}), 500
 
 @app.route('/train', methods=['POST'])
 def answer():