Update app.py

app.py

@@ -9,55 +9,32 @@ app = FastAPI()
 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: 
-        raise HTTPException(status_code=403, detail="Unauthorized")
+async def add_watermark(file: UploadFile = File(...), watermark_text: str = "@YourGroup", api_key: str = Header(None)):
+    if api_key != API_KEY: raise HTTPException(status_code=403, detail="Unauthorized")
     
     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())
+    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", 
+        "-preset", "ultrafast", "-crf", "23", "-c:a", "copy", 
         "-movflags", "+faststart+frag_keyframe+empty_moov",
         output_path
     ]
     
     try:
-        # প্রসেসিং শুরু
         subprocess.run(cmd, check=True)
-        
-        # ফাইলটি চেক করে ফেরত পাঠানো
-        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")
-            
+        return FileResponse(output_path, media_type="video/mp4", filename="watermarked.mp4")
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))
-    
-    # ফাইল ডিলিট করার লজিক এখানে দেব না কারণ FileResponse এর জন্য ফাইলটি ওপেন থাকতে হয়
-    # Hugging Face ফাইলগুলো অটোমেটিক ডিলিট করে দেয়, তাই ভয়ের কিছু নেই।
+    finally:
+        if os.path.exists(input_path): os.remove(input_path)
 
 if __name__ == "__main__":
-    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)))