Update app.py

app.py

@@ -1,9 +1,10 @@
 import gradio as gr
-from gradio_client import Client, handle_file
+from gradio_client import Client
 import os
 import random
 from io import BytesIO
 from PIL import Image
+import base64
 
 KEYS = os.getenv("KEYS").split(",")
 
@@ -15,26 +16,26 @@ def swap_face_api(source_img, target_img, doFaceEnhancer):
         api_key = get_random_api_key()
         client = Client("tuan2308/face-swap")
 
-        # Конвертируем PIL изображения в байты
+        # Конвертируем PIL изображения в байты, затем в base64
         source_bytes = BytesIO()
-        source_img.save(source_bytes, format="JPEG")  # или PNG, в зависимости от формата
-        source_bytes.seek(0)
+        source_img.save(source_bytes, format="JPEG")
+        source_b64 = base64.b64encode(source_bytes.getvalue()).decode("utf-8")
 
         target_bytes = BytesIO()
-        target_img.save(target_bytes, format="JPEG") # или PNG, в зависимости от формата
-        target_bytes.seek(0)
+        target_img.save(target_bytes, format="JPEG")
+        target_b64 = base64.b64encode(target_bytes.getvalue()).decode("utf-8")
+
 
         result = client.predict(
-            source_file=handle_file(source_bytes.read(), filename="source.jpg"), #  Добавляем имя файла
-            target_file=handle_file(target_bytes.read(), filename="target.jpg"), #  Добавляем имя файла
+            source_file=source_b64,  # Передаем base64 строку
+            target_file=target_b64,  # Передаем base64 строку
             doFaceEnhancer=doFaceEnhancer,
             api_name="/predict",
-            api_key=api_key
+            api_key=api_key # Передаем api_key в predict
         )
 
-        # Конвертируем результат из байтов обратно в PIL Image
+        # Конвертируем результат из байтов в PIL Image
         output_image = Image.open(BytesIO(result))
-
         return output_image
 
     except Exception as e: