Update app.py

app.py

@@ -13,7 +13,7 @@ app = FastAPI()
 SECRET_KEY = "MySecretWatermarkKey123"
 BOT_TOKEN = os.environ.get("BOT_TOKEN")
 
-# গ্লোবাল বট কানেকশন
+# হাই স্পিড কানেকশন পুল
 t_request = HTTPXRequest(connect_timeout=30, read_timeout=60, write_timeout=600)
 bot_instance = Bot(token=BOT_TOKEN, request=t_request) if BOT_TOKEN else None
 
@@ -30,44 +30,52 @@ async def add_watermark(
     unique_id = uuid.uuid4()
     input_path = f"in_{unique_id}.mp4"
     output_path = f"out_{unique_id}.mp4"
+    thumb_path = f"thumb_{unique_id}.jpg"
     
     try:
-        # ১. ফাইলটি ডিস্কে সেভ করা
+        # ১. ফাইল সেভ
         with open(input_path, "wb") as buffer:
             shutil.copyfileobj(file.file, buffer)
         
-        # ২. FFmpeg দিয়ে প্রসেস করা
+        # ২. সুপার ফাস্ট FFmpeg (Speed optimized)
+        # crf 28 এবং preset superfast ভিডিও এডিটিং দ্রুত করবে
         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",
+            "-vf", f"drawtext=text='{watermark_text}':x=w-tw-20:y=h-th-20:fontsize=30:fontcolor=white@0.5",
+            "-c:v", "libx264", "-preset", "superfast", "-crf", "28", "-c:a", "copy", 
+            "-movflags", "+faststart",
             output_path
         ]
         process = await asyncio.create_subprocess_exec(*cmd)
         await process.wait()
 
-        # ৩. Hugging Face থেকে আপলোড করার চেষ্টা
+        # ৩. থাম্বনেইল জেনারেট করা (ভিডিওর ২য় সেকেন্ড থেকে)
+        thumb_cmd = ["ffmpeg", "-i", output_path, "-ss", "00:00:02", "-vframes", "1", "-y", thumb_path]
+        thumb_proc = await asyncio.create_subprocess_exec(*thumb_cmd)
+        await thumb_proc.wait()
+
+        # ৪. আপলোড করার চেষ্টা (থাম্বনেইল সহ)
         if os.path.exists(output_path):
             try:
-                with open(output_path, "rb") as v:
+                with open(output_path, "rb") as v, open(thumb_path, "rb") as t:
                     await bot_instance.send_video(
-                        chat_id=chat_id, video=v, 
+                        chat_id=chat_id, 
+                        video=v, 
+                        thumbnail=t, # থাম্বনেইল সেট করা হলো
                         caption=f"✅ {watermark_text}", 
                         supports_streaming=True
                     )
-                return JSONResponse({"status": "success", "message": "Uploaded by HF"})
+                return JSONResponse({"status": "success"})
             except Exception as e:
-                print(f"HF Upload Failed, sending back to Main Bot: {e}")
-                # আপলোড ফেইল হলে ভিডিওটি মেইন বটকে রিটার্ন করবে
-                return FileResponse(output_path, media_type="video/mp4", filename="watermarked.mp4")
+                # আপলোড ফেইল হলে মেইন বটকে ভিডিও এবং থাম্বনেইল ফেরত পাঠানো (Rare case)
+                return FileResponse(output_path, media_type="video/mp4")
         
     except Exception as e:
         return JSONResponse({"status": "error", "message": str(e)}, status_code=500)
     finally:
-        if os.path.exists(input_path): os.remove(input_path)
-        # output_path রিমুভ করা হবে না যদি FileResponse দেওয়া হয় (FastAPI নিজে হ্যান্ডেল করবে)
+        # ক্লিনআপ
+        for path in [input_path, thumb_path]:
+            if os.path.exists(path): os.remove(path)
 
 if __name__ == "__main__":
     import uvicorn