Update app.py

app.py

@@ -1,10 +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 fastapi.middleware.wsgi import WSGIMiddleware
 from starlette.responses import JSONResponse
 import base64
 import io
@@ -27,7 +25,7 @@ def animegan2_transform(input_img, style_name):
         input_img = input_img.convert("RGB")
     else:
         raise ValueError("فرمت تصویر ورودی صحیح نیست!")
-    
+
     # تغییر رزولوشن به 460 برای کیفیت بالا
     input_img = input_img.resize((460, 460))
 
@@ -46,10 +44,10 @@ def animegan2_transform(input_img, style_name):
     output_img = face2paint_func(model, input_img)
     return output_img
 
-# ------------------ API با FastAPI ------------------
-api = FastAPI()
+# ------------------ FastAPI API ------------------
+app = FastAPI()
 
-@api.post("/api/animegan")
+@app.post("/api/animegan")
 async def animegan_api(
     file: UploadFile,
     style: str = Form("face_paint_512_v2")
@@ -64,25 +62,6 @@ async def animegan_api(
     img_b64 = base64.b64encode(buf.getvalue()).decode("utf-8")
     return JSONResponse({"image": img_b64})
 
-# ------------------ Gradio UI ------------------
-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="تصاویر خود را با کیفیت بالا و استایل‌های مختلف به انیمه تبدیل کنید."
-)
-
-# ترکیب Gradio با FastAPI
-api.mount("/", WSGIMiddleware(iface.app))
-
 if __name__ == "__main__":
     import uvicorn
-    uvicorn.run(api, host="0.0.0.0", port=7860)
+    uvicorn.run(app, host="0.0.0.0", port=7860)