Update app.py

app.py

@@ -2,6 +2,8 @@ import gradio as gr
 from gradio_client import Client, handle_file
 import os
 import random
+from io import BytesIO
+from PIL import Image
 
 KEYS = os.getenv("KEYS").split(",")
 
@@ -11,31 +13,35 @@ def get_random_api_key():
 def swap_face_api(source_img, target_img, doFaceEnhancer):
     try:
         api_key = get_random_api_key()
-
         client = Client("tuan2308/face-swap")
 
-        source_filename = f"source_{random.randint(1000, 9999)}.jpg"
-        target_filename = f"target_{random.randint(1000, 9999)}.jpg"
+        # Конвертируем PIL изображения в байты
+        source_bytes = BytesIO()
+        source_img.save(source_bytes, format="JPEG")  # или PNG, в зависимости от формата
+        source_bytes.seek(0)
 
-        source_img.save(source_filename)
-        target_img.save(target_filename)
+        target_bytes = BytesIO()
+        target_img.save(target_bytes, format="JPEG") # или PNG, в зависимости от формата
+        target_bytes.seek(0)
 
         result = client.predict(
-            source_file=handle_file(source_filename),
-            target_file=handle_file(target_filename),
+            source_file=handle_file(source_bytes.read(), filename="source.jpg"), #  Добавляем имя файла
+            target_file=handle_file(target_bytes.read(), filename="target.jpg"), #  Добавляем имя файла
             doFaceEnhancer=doFaceEnhancer,
             api_name="/predict",
             api_key=api_key
         )
 
-        os.remove(source_filename)
-        os.remove(target_filename)
+        # Конвертируем результат из байтов обратно в PIL Image
+        output_image = Image.open(BytesIO(result))
+
+        return output_image
 
-        return result
     except Exception as e:
         print(f"Ошибка при вызове API: {e}")
         return None
 
+
 iface = gr.Interface(
     fn=swap_face_api,
     inputs=[