Update app.py

app.py

@@ -3,39 +3,27 @@ import subprocess
 import uuid
 import httpx
 import asyncio
-from fastapi import FastAPI, BackgroundTasks, Body, Header, HTTPException
+from fastapi import FastAPI, BackgroundTasks, UploadFile, File, Form, Header, HTTPException
 from telegram import Bot
 
 app = FastAPI()
 
-# ১. সরাসরি কোডের ভেতরে সিক্রেট কী (বটের কোডের সাথে মিল থাকতে হবে)
 SECRET_KEY = "MySecretWatermarkKey123"
-
-# ২. বটের টোকেন Hugging Face Space এর 'Secrets' থেকে নিবে
 BOT_TOKEN = os.environ.get("BOT_TOKEN")
 
-async def process_video_task(video_url: str, watermark: str, chat_id: int):
-    """ব্যাকগ্রাউন্ডে ভিডিও ডাউনলোড, এডিট ও আপলোড করার ফাংশন"""
-    if not BOT_TOKEN:
-        print("Error: BOT_TOKEN not found in Secrets!")
-        return
-
+async def process_video_task(file_bytes, watermark: str, chat_id: int, original_msg_id: int):
+    if not BOT_TOKEN: return
     bot = Bot(token=BOT_TOKEN)
     unique_id = uuid.uuid4()
     input_path = f"in_{unique_id}.mp4"
     output_path = f"out_{unique_id}.mp4"
     
     try:
-        # টেলিগ্রাম থেকে সরাসরি ভিডিও ডাউনলোড
-        async with httpx.AsyncClient() as client:
-            response = await client.get(video_url, timeout=600.0)
-            if response.status_code == 200:
-                with open(input_path, "wb") as f:
-                    f.write(response.content)
-            else:
-                return
+        # ফাইল সেভ করা
+        with open(input_path, "wb") as f:
+            f.write(file_bytes)
 
-        # FFmpeg ওয়াটারমার্ক ও মেটাডেটা ফিক্স কমান্ড
+        # FFmpeg এডিটিং
         cmd = [
             "ffmpeg", "-i", input_path,
             "-vf", f"drawtext=text='{watermark}':x=w-tw-20:y=h-th-20:fontsize=30:fontcolor=white@0.6",
@@ -46,42 +34,36 @@ async def process_video_task(video_url: str, watermark: str, chat_id: int):
         ]
         subprocess.run(cmd, check=True)
 
-        # ভিডিও আপলোড
-        with open(output_path, "rb") as video_file:
+        # সরাসরি Hugging Face থেকে আপলোড
+        with open(output_path, "rb") as v:
             await bot.send_video(
                 chat_id=chat_id, 
-                video=video_file, 
+                video=v, 
                 caption=f"✅ {watermark}",
                 supports_streaming=True
             )
-            
     except Exception as e:
-        print(f"Background Process Error: {e}")
+        print(f"Error: {e}")
     finally:
-        # টেম্পোরারি ফাইল পরিষ্কার
         if os.path.exists(input_path): os.remove(input_path)
         if os.path.exists(output_path): os.remove(output_path)
 
 @app.post("/watermark")
 async def trigger_watermark(
-    background_tasks: BackgroundTasks, 
-    api_key: str = Header(None), 
-    data: dict = Body(...)
+    background_tasks: BackgroundTasks,
+    file: UploadFile = File(...),
+    watermark_text: str = Form(...),
+    chat_id: int = Form(...),
+    api_key: str = Header(None)
 ):
-    # কোডের ভেতরকার কী-এর সাথে হেডার চেক
     if api_key != SECRET_KEY:
         raise HTTPException(status_code=403, detail="Unauthorized")
     
-    # প্রসেসিং ব্যাকগ্রাউন্ডে পাঠিয়ে দেওয়া হলো
-    background_tasks.add_task(
-        process_video_task, 
-        data["video_url"], 
-        data["watermark_text"], 
-        data["chat_id"]
-    )
-    return {"status": "Task queued successfully"}
+    # ফাইলটি মেমরিতে নিয়ে ব্যাকগ্রাউন্ডে পাঠানো
+    file_content = await file.read()
+    background_tasks.add_task(process_video_task, file_content, watermark_text, chat_id, 0)
+    return {"status": "Success"}
 
 if __name__ == "__main__":
     import uvicorn
-    port = int(os.environ.get("PORT", 7860))
-    uvicorn.run(app, host="0.0.0.0", port=port)
+    uvicorn.run(app, host="0.0.0.0", port=int(os.environ.get("PORT", 7860)))