Update app.py

app.py

@@ -1,7 +1,7 @@
 import os
 import subprocess
 import uuid
-from fastapi import FastAPI, UploadFile, File, Header
+from fastapi import FastAPI, UploadFile, File, Header, HTTPException
 from fastapi.responses import FileResponse
 import uvicorn
 
@@ -9,43 +9,55 @@ app = FastAPI()
 API_KEY = "MySecretWatermarkKey123"
 
 @app.post("/watermark")
-async def add_watermark(file: UploadFile = File(...), watermark_text: str = "@YourGroup", api_key: str = Header(None)):
+async def add_watermark(
+    file: UploadFile = File(...), 
+    watermark_text: str = "@YourGroup", 
+    api_key: str = Header(None)
+):
     if api_key != API_KEY: 
-        return {"error": "Unauthorized"}
+        raise HTTPException(status_code=403, detail="Unauthorized")
     
-    # ফাইলের এক্সটেনশন চেক
-    ext = os.path.splitext(file.filename)[1].lower()
-    input_path = f"in_{uuid.uuid4()}{ext}"
-    output_path = f"out_{uuid.uuid4()}.mp4" # সব আউটপুট mp4 তে নেওয়া ভালো
+    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())
     
-    # ভিডিওর জন্য কমান্ড (থাম্বনেইল এবং মেটাডেটা ঠিক রাখার জন্য)
+    # FFmpeg কমান্ড: 
+    # -movflags +faststart থাম্বনেইল এবং প্লে-ব্যাকের জন্য সবথেকে জরুরি
+    # -pix_fmt yuv420p ভিডিও প্লেয়ারে চালানোর জন্য স্ট্যান্ডার্ড
     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", 
+        "-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, media_type="video/mp4")
+        
+        # ফাইলটি চেক করে ফেরত পাঠানো
+        if os.path.exists(output_path) and os.path.getsize(output_path) > 0:
+            return FileResponse(
+                output_path, 
+                media_type="video/mp4", 
+                filename="watermarked.mp4",
+                headers={"Content-Disposition": "attachment; filename=watermarked.mp4"}
+            )
+        else:
+            raise HTTPException(status_code=500, detail="Processing failed")
+            
     except Exception as e:
-        return {"error": str(e)}
-    finally:
-        # ফাইল ডিলিট করা
-        if os.path.exists(input_path): os.remove(input_path)
-        # output_path এরর এড়াতে এখানে রিমুভ করব না, FastAPI এটা হ্যান্ডেল করবে
+        raise HTTPException(status_code=500, detail=str(e))
+    
+    # ফাইল ডিলিট করার লজিক এখানে দেব না কারণ FileResponse এর জন্য ফাইলটি ওপেন থাকতে হয়
+    # Hugging Face ফাইলগুলো অটোমেটিক ডিলিট করে দেয়, তাই ভয়ের কিছু নেই।
 
 if __name__ == "__main__":
-    # Hugging Face পোর্ট চেক
     port = int(os.environ.get("PORT", 7860))
     uvicorn.run(app, host="0.0.0.0", port=port)