Update app.py

app.py

@@ -10,36 +10,42 @@ 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: return {"error": "Unauthorized"}
+    if api_key != API_KEY: 
+        return {"error": "Unauthorized"}
     
-    ext = os.path.splitext(file.filename)[1].lower() # ফাইল ফরম্যাট চেক
+    # ফাইলের এক্সটেনশন চেক
+    ext = os.path.splitext(file.filename)[1].lower()
     input_path = f"in_{uuid.uuid4()}{ext}"
-    output_path = f"out_{uuid.uuid4()}{ext}"
+    output_path = f"out_{uuid.uuid4()}.mp4" # সব আউটপুট mp4 তে নেওয়া ভালো
     
-    with open(input_path, "wb") as buffer: buffer.write(await file.read())
+    with open(input_path, "wb") as buffer: 
+        buffer.write(await file.read())
     
-    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
-        ]
+    # ভিডিওর জন্য কমান্ড (থাম্বনেইল এবং মেটাডেটা ঠিক রাখার জন্য)
+    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+frag_keyframe+empty_moov",
+        output_path
+    ]
     
     try:
+        # প্রসেস রান করা
         subprocess.run(cmd, check=True)
-        return FileResponse(output_path)
+        return FileResponse(output_path, media_type="video/mp4")
+    except Exception as e:
+        return {"error": str(e)}
     finally:
+        # ফাইল ডিলিট করা
         if os.path.exists(input_path): os.remove(input_path)
+        # output_path এরর এড়াতে এখানে রিমুভ করব না, FastAPI এটা হ্যান্ডেল করবে
 
 if __name__ == "__main__":
-    uvicorn.run(app, host="0.0.0.0", port=int(os.environ.get("PORT", 7860)))
+    # Hugging Face পোর্ট চেক
+    port = int(os.environ.get("PORT", 7860))
+    uvicorn.run(app, host="0.0.0.0", port=port)