Update main.py

main.py

@@ -7,46 +7,49 @@ import uuid
 import cloudinary
 import cloudinary.uploader
 from cloudinary.utils import cloudinary_url
-# Define the directory to store videos
+
 VIDEO_DIR = "compressed_videos"
-os.makedirs(VIDEO_DIR, exist_ok=True)  # Ensure the directory exists
+os.makedirs(VIDEO_DIR, exist_ok=True)  
 
 app = FastAPI()
 
-# Mount the static file directory to serve videos
+
 app.mount("/videos", StaticFiles(directory=VIDEO_DIR), name="videos")
 
-# Video compression and format conversion endpoint using FFmpeg
+
 @app.get("/compress_video")
 async def compress_video(
-    video_url: str = Query(..., description="URL of the video to compress and convert"),
-    resolution: str = Query("720x480", description="Resolution (e.g., 1280x720, 640x360)"),
-    format: str = Query("mp4", description="Output video format (e.g., mp4, mkv)"),
+    video_url: str = Query(..., description="URL of the video"),
+    resolution: str = Query("1280x720", description="Resolution (e.g., 1280x720, 640x360)"),
+    format: str = Query("matroska", description="Output video format (e.g., mp4, mkv)"),
+    codec: str = Query("libx264", description="264 faster but bigger size,265 slower but smaller size"),
+    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:
-        # Generate a unique filename
-        file_name = f"{uuid.uuid4()}.{format}"
+        file_name = f"{uuid.uuid4()}.mkv"
         file_path = os.path.join(VIDEO_DIR, file_name)
 
-        # Download and compress the video using FFmpeg
+       
         (
             ffmpeg
             .input(video_url)
-            .output(file_path, vf=f"scale={resolution}", crf=crf, vcodec="libx265", f=format)
+            .output(file_path, vf=f"scale={resolution}", crf=crf, vcodec={codec}, preset={preset}, f=format)
+            .global_args("-threads", "0") 
             .overwrite_output()
             .run()
+
         )
 
-        # Generate direct URL to the file
+        
         file_url = f"/videos/{file_name}"
 
-        return JSONResponse(content={"compressed_video_url": file_url})
+        return JSONResponse(content={"compressed_video_url": "https://akane710-imagecompress.hf.space"+file_url})
 
     except Exception as e:
         return JSONResponse(content={"error": str(e)}, status_code=500)
 
-# Cloudinary configuration
+
 cloudinary.config(
     cloud_name=os.getenv("cloudname"),
     api_key=os.getenv("key"),
@@ -54,7 +57,7 @@ cloudinary.config(
     secure=True
 )
 
-# Image compression endpoint
+
 @app.get("/compress_image")
 async def compress_image(image_url: str = Query(..., description="URL of the image to compress")):
     try:
@@ -71,4 +74,4 @@ async def compress_image(image_url: str = Query(..., description="URL of the ima
         
 @app.get("/")
 def root():
-    return "بتعمل شنو هنا يا زول"
+    return "بتعمل شنو هنا يا زول"