Update main.py

main.py

@@ -4,6 +4,8 @@ import cloudinary
 import cloudinary.uploader
 from cloudinary.utils import cloudinary_url
 import os
+import ffmpeg
+import tempfile
 
 app = FastAPI()
 
@@ -15,55 +17,42 @@ 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:
-        upload_result = cloudinary.uploader.upload(image_url)
-        compressed_url, options = cloudinary_url(
-            upload_result["public_id"],
-            fetch_format="auto",
-            quality="auto:low"
-        )
-        return JSONResponse(content={"compressed_image_url": compressed_url})
-
-    except Exception as e:
-        return JSONResponse(content={"error": str(e)}, status_code=500)
-
-# Video compression and format conversion endpoint
+# 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"),
-    quality: str = Query("auto:low", description="Compression quality (e.g., auto:low, auto:good)"),
-    resolution: str = Query(None, description="Resolution (e.g., 720p, 480p)"),
+    resolution: str = Query("265x", description="Resolution (e.g., 720x480, 265x265)"),
     format: str = Query("mkv", description="Output video format (e.g., mp4, mkv)"),
-    codec: str = Query("h265", description="Video codec (e.g., h264, h265)")
+    crf: int = Query(28, description="Constant Rate Factor (CRF) for compression (0-51, lower is better quality)")
 ):
     try:
-        # Upload the video to Cloudinary
-        upload_result = cloudinary.uploader.upload_large(
-            video_url, resource_type="video"
-        )
+        # Create a temporary file to store the downloaded video
+        with tempfile.NamedTemporaryFile(suffix=f".{format}", delete=False) as temp_file:
+            temp_file_path = temp_file.name
 
-        # Build the transformation options
-        transformation = {
-            "fetch_format": format,  # Set the output format
-            "quality": quality,      # Set compression quality
-            "video_codec": codec     # Specify the video codec
-        }
+        # Download the video from the URL using FFmpeg
+        (
+            ffmpeg
+            .input(video_url)
+            .output(temp_file_path, vf=f"scale={resolution}", crf=crf, vcodec="libx265", format=format)
+            .overwrite_output()
+            .run()
+        )
 
-        # Add resolution only if specified
-        if resolution:
-            transformation["width"] = resolution.split("p")[0]
-            transformation["crop"] = "scale"  # Ensures scaling to specified resolution
+        # Upload the compressed video to Cloudinary
+        upload_result = cloudinary.uploader.upload_large(
+            temp_file_path, resource_type="video"
+        )
 
-        # Generate the compressed and converted video URL
+        # Generate the URL for the compressed video
         compressed_video_url, options = cloudinary_url(
             upload_result["public_id"],
-            resource_type="video",
-            **transformation
+            resource_type="video"
         )
 
+        # Clean up the temporary file
+        os.remove(temp_file_path)
+
         return JSONResponse(content={"compressed_video_url": compressed_video_url})
 
     except Exception as e:
@@ -72,4 +61,4 @@ async def compress_video(
 # Root endpoint
 @app.get("/")
 def root():
-    return "بتعمل شنو هنا يا زول"
+    return "بتعمل شنو هنا يا زول"