Update app.py

app.py

@@ -12,26 +12,34 @@ API_KEY = "MySecretWatermarkKey123"
 async def add_watermark(file: UploadFile = File(...), watermark_text: str = "@YourGroup", api_key: str = Header(None)):
     if api_key != API_KEY: return {"error": "Unauthorized"}
     
-    input_path = f"in_{uuid.uuid4()}.mp4"
-    output_path = f"out_{uuid.uuid4()}.mp4"
+    ext = os.path.splitext(file.filename)[1].lower() # ফাইল ফরম্যাট চেক
+    input_path = f"in_{uuid.uuid4()}{ext}"
+    output_path = f"out_{uuid.uuid4()}{ext}"
     
     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", "-c:a", "copy",
-        output_path
-    ]
+    if ext in [".mp4", ".mov", ".mkv"]:
+        # ভিডিওর জন্য কমান্ড (থাম্বনেইল ঠিক রাখতে -movflags +faststart)
+        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", "-c:a", "copy", "-movflags", "+faststart",
+            output_path
+        ]
+    else:
+        # ছবির জন্য কমান্ড
+        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",
+            output_path
+        ]
     
     try:
         subprocess.run(cmd, check=True)
-        return FileResponse(output_path, media_type="video/mp4")
+        return FileResponse(output_path)
     finally:
         if os.path.exists(input_path): os.remove(input_path)
 
 if __name__ == "__main__":
-    # Hugging Face এ পোর্ট ৭৯৬০ বা এনভায়রনমেন্ট পোর্ট ব্যবহার করা ভালো
-    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)))