Update app.py

app.py

@@ -16,32 +16,31 @@ def swap_face_api(source_img, target_img, doFaceEnhancer):
         api_key = get_random_api_key()
         client = Client("tuan2308/face-swap")
 
-        # Конвертируем PIL изображения в байты, затем в base64
-        source_bytes = BytesIO()
-        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")
-        target_b64 = base64.b64encode(target_bytes.getvalue()).decode("utf-8")
-
+        # Кодируем PIL изображения в base64, используя gr.processing_utils
+        source_b64 = gr.processing_utils.encode_pil_to_base64(source_img)
+        target_b64 = gr.processing_utils.encode_pil_to_base64(target_img)
 
         result = client.predict(
-            source_file=source_b64,  # Передаем base64 строку
-            target_file=target_b64,  # Передаем base64 строку
+            source_file=source_b64,
+            target_file=target_b64,
             doFaceEnhancer=doFaceEnhancer,
             api_name="/predict",
-            api_key=api_key # Передаем api_key в predict
+            api_key=api_key #  Передаем api_key в predict
         )
 
-        # Конвертируем результат из байтов в PIL Image
-        output_image = Image.open(BytesIO(result))
+        # Декодируем результат из base64 в PIL Image
+        result_decoded = base64.b64decode(result) # Декодируем результат
+        output_image = Image.open(BytesIO(result_decoded))
         return output_image
 
     except Exception as e:
         print(f"Ошибка при вызове API: {e}")
         return None
 
+    except Exception as e:
+        print(f"Ошибка при вызове API: {e}")
+        return None
+
 
 iface = gr.Interface(
     fn=swap_face_api,