Update app.py

app.py

@@ -1,11 +1,8 @@
+import gradio as gr
 import torch
 from PIL import Image
 import requests
 from io import BytesIO
-from fastapi import FastAPI, UploadFile, Form
-from starlette.responses import JSONResponse
-import base64
-import io
 
 # بارگذاری مدل‌ها بر اساس استایل انتخابی
 def load_model(style_name):
@@ -25,8 +22,8 @@ def animegan2_transform(input_img, style_name):
         input_img = input_img.convert("RGB")
     else:
         raise ValueError("فرمت تصویر ورودی صحیح نیست!")
-
-    # تغییر رزولوشن به 460 برای کیفیت بالا
+    
+    # تغییر رزولوشن به 1024 برای کیفیت بالا
     input_img = input_img.resize((460, 460))
 
     # بارگذاری مدل انتخابی
@@ -44,24 +41,21 @@ def animegan2_transform(input_img, style_name):
     output_img = face2paint_func(model, input_img)
     return output_img
 
-# ------------------ FastAPI API ------------------
-app = FastAPI()
-
-@app.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
+iface = gr.Interface(
+    fn=animegan2_transform,
+    inputs=[
+        gr.Image(type="pil", label="آپلود تصویر یا وارد کردن URL"),
+        gr.Dropdown(
+            ["face_paint_512_v1", "face_paint_512_v2", "paprika", "celeba_distill"],
+            value="face_paint_512_v2",
+            label="انتخاب استایل"
+        )
+    ],
+    outputs=gr.Image(type="pil", label="تصویر انیمه (کیفیت بالا)"),
+    title="AnimeGANv2 - تبدیل تصویر به انیمه با انتخاب استایل",
+    description="تصاویر خود را با کیفیت بالا و استایل‌های مختلف به انیمه تبدیل کنید."
+)
 
 if __name__ == "__main__":
-    import uvicorn
-    uvicorn.run(app, host="0.0.0.0", port=7860)
+    iface.launch()