Update main.py

main.py

@@ -25,13 +25,13 @@ async def compress_video(
     crf: int = Query(28, description="Constant Rate Factor (CRF) for compression (0-51, lower is better quality)")
 ):
     try:
-        file_name = f"{uuid.uuid4()}.mkv"
+        file_name = f"{uuid.uuid4()}.{format}"
         file_path = os.path.join(VIDEO_DIR, file_name)
 
         (
             ffmpeg
             .input(video_url)
-            .output(file_path, vf=f"scale={resolution}", crf=crf, vcodec=codec, preset=preset, f=format)
+            .output(file_path, crf=crf, vcodec=codec, preset=preset, s=resolution, f=format)
             .global_args("-threads", "0")
             .overwrite_output()
             .run()
@@ -52,15 +52,30 @@ async def upload_video(
     preset: str = Query("fast", description="ultrafast, superfast, fast, medium, slow"),
     crf: int = Query(28, description="Constant Rate Factor (CRF) for compression (0-51, lower is better quality)")):
     try:
-        file_ext = file.filename.split(".")[-1]
-        new_filename = f"{uuid.uuid4()}.{file_ext}"
-        file_path = os.path.join(VIDEO_DIR, new_filename)
-        
-        with open(file_path, "wb") as buffer:
+        original_filename = file.filename
+        file_ext = original_filename.split(".")[-1]
+        temp_filename = f"{uuid.uuid4()}.{file_ext}"
+        temp_filepath = os.path.join(VIDEO_DIR, temp_filename)
+
+        with open(temp_filepath, "wb") as buffer:
             buffer.write(file.file.read())
-        
-        file_url = f"/videos/{new_filename}"
-        return JSONResponse(content={"uploaded_video_url": "https://akane710-imagecompress.hf.space" + file_url})
+
+        compressed_filename = f"{uuid.uuid4()}.{format}"
+        compressed_filepath = os.path.join(VIDEO_DIR, compressed_filename)
+
+        (
+            ffmpeg
+            .input(temp_filepath)
+            .output(compressed_filepath, crf=crf, vcodec=codec, preset=preset, s=resolution, f=format)
+            .global_args("-threads", "0")
+            .overwrite_output()
+            .run()
+        )
+
+        os.remove(temp_filepath)  # Clean up the original uploaded file
+
+        file_url = f"/videos/{compressed_filename}"
+        return JSONResponse(content={"compressed_video_url": "https://akane710-imagecompress.hf.space" + file_url})
     
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))