Update app.py

app.py

@@ -3,27 +3,27 @@ import subprocess
 import uuid
 import httpx
 import asyncio
+import shutil
 from fastapi import FastAPI, BackgroundTasks, UploadFile, File, Form, Header, HTTPException
 from telegram import Bot
+from telegram.request import HTTPXRequest
 
 app = FastAPI()
 
 SECRET_KEY = "MySecretWatermarkKey123"
 BOT_TOKEN = os.environ.get("BOT_TOKEN")
 
-async def process_video_task(file_bytes, watermark: str, chat_id: int, original_msg_id: int):
+async def process_video_task(input_path: str, watermark: str, chat_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"
+    
+    # আপলোড টাইমআউট ফিক্স (১০ মিনিট সময় দেওয়া হয়েছে)
+    t_request = HTTPXRequest(connect_timeout=600, read_timeout=600, write_timeout=600)
+    bot = Bot(token=BOT_TOKEN, request=t_request)
+    
+    output_path = f"out_{uuid.uuid4()}.mp4"
     
     try:
-        # ফাইল সেভ করা
-        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",
@@ -32,19 +32,24 @@ async def process_video_task(file_bytes, watermark: str, chat_id: int, original_
             "-movflags", "+faststart+frag_keyframe+empty_moov",
             output_path
         ]
-        subprocess.run(cmd, check=True)
+        
+        # প্রসেসিং রান করা
+        process = await asyncio.create_subprocess_exec(*cmd)
+        await process.wait()
 
-        # সরাসরি Hugging Face থেকে আপলোড
-        with open(output_path, "rb") as v:
-            await bot.send_video(
-                chat_id=chat_id, 
-                video=v, 
-                caption=f"✅ {watermark}",
-                supports_streaming=True
-            )
+        # সরাসরি ফাইল থেকে স্ট্রিম করে টেলিগ্রামে আপলোড
+        if os.path.exists(output_path):
+            with open(output_path, "rb") as v:
+                await bot.send_video(
+                    chat_id=chat_id, 
+                    video=v, 
+                    caption=f"✅ {watermark}",
+                    supports_streaming=True
+                )
     except Exception as e:
-        print(f"Error: {e}")
+        print(f"Detailed Error: {e}")
     finally:
+        # সব ফাইল পরিষ্কার করা
         if os.path.exists(input_path): os.remove(input_path)
         if os.path.exists(output_path): os.remove(output_path)
 
@@ -59,10 +64,15 @@ async def trigger_watermark(
     if api_key != SECRET_KEY:
         raise HTTPException(status_code=403, detail="Unauthorized")
     
-    # ফাইলটি মেমরিতে নিয়ে ব্যাকগ্রাউন্ডে পাঠানো
-    file_content = await file.read()
-    background_tasks.add_task(process_video_task, file_content, watermark_text, chat_id, 0)
-    return {"status": "Success"}
+    unique_id = uuid.uuid4()
+    input_path = f"in_{unique_id}.mp4"
+    
+    # মেমরি বাঁচাতে ফাইলটি সরাসরি ডিস্কে রাইট করা
+    with open(input_path, "wb") as buffer:
+        shutil.copyfileobj(file.file, buffer)
+    
+    background_tasks.add_task(process_video_task, input_path, watermark_text, chat_id)
+    return {"status": "Processing queued"}
 
 if __name__ == "__main__":
     import uvicorn