Update app.py

app.py

@@ -3,6 +3,11 @@ import torch
 from PIL import Image
 import requests
 from io import BytesIO
+from fastapi import FastAPI, UploadFile, Form
+from fastapi.middleware.wsgi import WSGIMiddleware
+from starlette.responses import JSONResponse
+import base64
+import io
 
 # بارگذاری مدل‌ها بر اساس استایل انتخابی
 def load_model(style_name):
@@ -23,7 +28,7 @@ def animegan2_transform(input_img, style_name):
     else:
         raise ValueError("فرمت تصویر ورودی صحیح نیست!")
     
-    # تغییر رزولوشن به 1024 برای کیفیت بالا
+    # تغییر رزولوشن به 460 برای کیفیت بالا
     input_img = input_img.resize((460, 460))
 
     # بارگذاری مدل انتخابی
@@ -41,7 +46,25 @@ def animegan2_transform(input_img, style_name):
     output_img = face2paint_func(model, input_img)
     return output_img
 
-# رابط کاربری Gradio
+# ------------------ API با FastAPI ------------------
+api = FastAPI()
+
+@api.post("/api/animegan")
+async def animegan_api(
+    file: UploadFile,
+    style: str = Form("face_paint_512_v2")
+):
+    img_bytes = await file.read()
+    input_img = Image.open(BytesIO(img_bytes)).convert("RGB")
+    output_img = animegan2_transform(input_img, style)
+
+    # تبدیل به Base64
+    buf = io.BytesIO()
+    output_img.save(buf, format="PNG")
+    img_b64 = base64.b64encode(buf.getvalue()).decode("utf-8")
+    return JSONResponse({"image": img_b64})
+
+# ------------------ Gradio UI ------------------
 iface = gr.Interface(
     fn=animegan2_transform,
     inputs=[
@@ -57,5 +80,9 @@ iface = gr.Interface(
     description="تصاویر خود را با کیفیت بالا و استایل‌های مختلف به انیمه تبدیل کنید."
 )
 
+# ترکیب Gradio با FastAPI
+api.mount("/", WSGIMiddleware(iface.app))
+
 if __name__ == "__main__":
-    iface.launch()
+    import uvicorn
+    uvicorn.run(api, host="0.0.0.0", port=7860)