Update app.py

app.py

@@ -1,41 +1,87 @@
 import os
 import subprocess
 import uuid
-from fastapi import FastAPI, UploadFile, File, Header, HTTPException
-from fastapi.responses import FileResponse
-import uvicorn
+import httpx
+import asyncio
+from fastapi import FastAPI, BackgroundTasks, Body, Header, HTTPException
+from telegram import Bot
 
 app = FastAPI()
-API_KEY = "MySecretWatermarkKey123"
 
-@app.post("/watermark")
-async def add_watermark(file: UploadFile = File(...), watermark_text: str = "@YourGroup", api_key: str = Header(None)):
-    if api_key != API_KEY: raise HTTPException(status_code=403, detail="Unauthorized")
-    
+# ১. সরাসরি কোডের ভেতরে সিক্রেট কী (বটের কোডের সাথে মিল থাকতে হবে)
+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
+
+    bot = Bot(token=BOT_TOKEN)
     unique_id = uuid.uuid4()
     input_path = f"in_{unique_id}.mp4"
     output_path = f"out_{unique_id}.mp4"
     
-    with open(input_path, "wb") as buffer: buffer.write(await file.read())
-    
-    # নতুন কমান্ড: এতে থাম্বনেইল সরাসরি ভিডিও ফাইলের ভেতরে এম্বেড হয়ে যাবে
-    cmd = [
-        "ffmpeg", "-i", input_path,
-        "-vf", f"drawtext=text='{watermark_text}':x=w-tw-20:y=h-th-20:fontsize=30:fontcolor=white@0.6",
-        "-c:v", "libx264", "-profile:v", "main", "-pix_fmt", "yuv420p", 
-        "-preset", "ultrafast", "-crf", "23", "-c:a", "copy",
-        "-movflags", "+faststart+frag_keyframe+empty_moov",
-        "-metadata:s:v:0", "handler_name=VideoHandler",
-        output_path
-    ]
-    
     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
+
+        # 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",
+            "-c:v", "libx264", "-profile:v", "main", "-pix_fmt", "yuv420p", 
+            "-preset", "ultrafast", "-crf", "23", "-c:a", "copy", 
+            "-movflags", "+faststart+frag_keyframe+empty_moov",
+            output_path
+        ]
         subprocess.run(cmd, check=True)
-        return FileResponse(output_path, media_type="video/mp4", filename="watermarked.mp4")
+
+        # ভিডিও আপলোড
+        with open(output_path, "rb") as video_file:
+            await bot.send_video(
+                chat_id=chat_id, 
+                video=video_file, 
+                caption=f"✅ {watermark}",
+                supports_streaming=True
+            )
+            
     except Exception as e:
-        raise HTTPException(status_code=500, detail=str(e))
+        print(f"Background Process 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(...)
+):
+    # কোডের ভেতরকার কী-এর সাথে হেডার চেক
+    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"}
 
 if __name__ == "__main__":
-    uvicorn.run(app, host="0.0.0.0", port=int(os.environ.get("PORT", 7860)))
+    import uvicorn
+    port = int(os.environ.get("PORT", 7860))
+    uvicorn.run(app, host="0.0.0.0", port=port)