Update app.py

app.py

@@ -1,27 +1,18 @@
-import logging
-from telegram import Update
-from telegram.ext import (
-    Application,
-    CommandHandler,
-    MessageHandler,
-    filters,
-    ContextTypes,
-)
-import torch
+from fastapi import FastAPI, HTTPException
 from PIL import Image
-import requests
+import torch
 from io import BytesIO
+import requests
+from pydantic import BaseModel
 
-# تنظیمات لاگینگ
-logging.basicConfig(
-    format="%(asctime)s - %(name)s - %(levelname)s - %(message)s", level=logging.INFO
-)
-logger = logging.getLogger(__name__)
+app = FastAPI()
 
-# توکن ربات تلگرام
-TOKEN = "7496239368:AAHkCBvlhqCA760FxkTfy1znu0YlYI16cOw"  # توکنی که از BotFather گرفتی رو اینجا بذار
+# مدل درخواست ورودی
+class ImageRequest(BaseModel):
+    image_url: str
+    style: str = "face_paint_512_v2"
 
-# بارگذاری مدل‌ها بر اساس استایل
+# بارگذاری مدل‌ها
 def load_model(style_name):
     model = torch.hub.load(
         "bryandlee/animegan2-pytorch:main",
@@ -32,85 +23,36 @@ def load_model(style_name):
     return model
 
 # پردازش تصویر
-def animegan2_transform(input_img, style_name="face_paint_512_v2"):
-    if isinstance(input_img, str):
-        input_img = Image.open(BytesIO(requests.get(input_img).content)).convert("RGB")
-    elif isinstance(input_img, Image.Image):
-        input_img = input_img.convert("RGB")
-    else:
-        raise ValueError("فرمت تصویر ورودی صحیح نیست!")
-    
-    input_img = input_img.resize((460, 460))
-    model = load_model(style_name)
-    face2paint_func = torch.hub.load(
-        "bryandlee/animegan2-pytorch:main",
-        "face2paint",
-        size=1024,
-        verbose=False
-    )
-    output_img = face2paint_func(model, input_img)
-    return output_img
-
-# دستور شروع ربات
-async def start(update: Update, context: ContextTypes.DEFAULT_TYPE):
-    await update.message.reply_text(
-        "سلام! من ربات تبدیل عکس به انیمه هستم. یه عکس بفرست تا به استایل انیمه تبدیلش کنم!\n"
-        "برای انتخاب استایل، یکی از این‌ها رو بنویس:\n"
-        "- face_paint_512_v1\n- face_paint_512_v2\n- paprika\n- celeba_distill"
-    )
-
-# مدیریت دریافت عکس
-async def handle_photo(update: Update, context: ContextTypes.DEFAULT_TYPE):
-    style = context.user_data.get("style", "face_paint_512_v2")  # استایل پیش‌فرض
-    await update.message.reply_text("در حال پردازش تصویر... کمی صبر کن!")
-    
-    # دریافت عکس از پیام
-    photo_file = await update.message.photo[-1].get_file()
-    photo_bytes = await photo_file.download_as_bytearray()
-    input_img = Image.open(BytesIO(photo_bytes)).convert("RGB")
-    
+def animegan2_transform(image_url, style_name):
     try:
-        # پردازش تصویر با مدل
-        output_img = animegan2_transform(input_img, style)
-        
-        # ذخیره تصویر خروجی
+        input_img = Image.open(BytesIO(requests.get(image_url).content)).convert("RGB")
+        input_img = input_img.resize((460, 460))
+        model = load_model(style_name)
+        face2paint_func = torch.hub.load(
+            "bryandlee/animegan2-pytorch:main",
+            "face2paint",
+            size=1024,
+            verbose=False
+        )
+        output_img = face2paint_func(model, input_img)
         output_buffer = BytesIO()
         output_img.save(output_buffer, format="PNG")
         output_buffer.seek(0)
-        
-        # ارسال تصویر به کاربر
-        await update.message.reply_photo(photo=output_buffer, caption="تصویر انیمه شما!")
+        return output_buffer
     except Exception as e:
-        await update.message.reply_text(f"خطایی رخ داد: {str(e)}")
+        raise HTTPException(status_code=500, detail=str(e))
 
-# مدیریت انتخاب استایل
-async def set_style(update: Update, context: ContextTypes.DEFAULT_TYPE):
-    style = update.message.text
+# مسیر API برای پردازش تصویر
+@app.post("/process_image")
+async def process_image(request: ImageRequest):
     valid_styles = ["face_paint_512_v1", "face_paint_512_v2", "paprika", "celeba_distill"]
-    if style in valid_styles:
-        context.user_data["style"] = style
-        await update.message.reply_text(f"استایل به {style} تنظیم شد. حالا یه عکس بفرست!")
-    else:
-        await update.message.reply_text("استایل نامعتبر! یکی از استایل‌های زیر رو انتخاب کن:\n" + "\n".join(valid_styles))
-
-# مدیریت خطاها
-async def error_handler(update: Update, context: ContextTypes.DEFAULT_TYPE):
-    logger.error(f"Update {update} caused error {context.error}")
-    await update.message.reply_text("یه خطایی پیش اومد. دوباره امتحان کن!")
-
-# تابع اصلی
-def main():
-    # ساخت اپلیکیشن ربات
-    application = Application.builder().token(TOKEN).build()
-
-    # ثبت دستورات
-    application.add_handler(CommandHandler("start", start))
-    application.add_handler(MessageHandler(filters.PHOTO, handle_photo))
-    application.add_handler(MessageHandler(filters.TEXT & ~filters.COMMAND, set_style))
-    application.add_error_handler(error_handler)
-
-    # شروع ربات
-    application.run_polling()
+    if request.style not in valid_styles:
+        raise HTTPException(status_code=400, detail="استایل نامعتبر!")
+    
+    output_buffer = animegan2_transform(request.image_url, request.style)
+    return {"image": output_buffer.getvalue(), "content_type": "image/png"}
 
-if __name__ == "__main__":
-    main()
+# برای تست ساده
+@app.get("/")
+async def root():
+    return {"message": "API برای تبدیل تصویر به انیمه"}