Update main.py

main.py

@@ -1,4 +1,4 @@
-from fastapi import FastAPI, Query
+from fastapi import FastAPI, Query, File, UploadFile, HTTPException
 from fastapi.responses import JSONResponse
 from fastapi.staticfiles import StaticFiles
 import os
@@ -9,14 +9,12 @@ import cloudinary.uploader
 from cloudinary.utils import cloudinary_url
 
 VIDEO_DIR = "compressed_videos"
-os.makedirs(VIDEO_DIR, exist_ok=True)  
+os.makedirs(VIDEO_DIR, exist_ok=True)
 
 app = FastAPI()
 
-
 app.mount("/videos", StaticFiles(directory=VIDEO_DIR), name="videos")
 
-
 @app.get("/compress_video")
 async def compress_video(
     video_url: str = Query(..., description="URL of the video"),
@@ -30,25 +28,36 @@ async def compress_video(
         file_name = f"{uuid.uuid4()}.mkv"
         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)
-            .global_args("-threads", "0") 
+            .global_args("-threads", "0")
             .overwrite_output()
             .run()
-
         )
 
-        
         file_url = f"/videos/{file_name}"
-
-        return JSONResponse(content={"compressed_video_url": "https://akane710-imagecompress.hf.space"+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)
 
+@app.post("/upload_video")
+async def upload_video(file: UploadFile = File(...)):
+    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:
+            buffer.write(file.file.read())
+        
+        file_url = f"/videos/{new_filename}"
+        return JSONResponse(content={"uploaded_video_url": "https://akane710-imagecompress.hf.space" + file_url})
+    
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
 
 cloudinary.config(
     cloud_name=os.getenv("cloudname"),
@@ -57,7 +66,6 @@ cloudinary.config(
     secure=True
 )
 
-
 @app.get("/compress_image")
 async def compress_image(image_url: str = Query(..., description="URL of the image to compress")):
     try:
@@ -71,7 +79,7 @@ async def compress_image(image_url: str = Query(..., description="URL of the ima
 
     except Exception as e:
         return JSONResponse(content={"error": str(e)}, status_code=500)
-        
+
 @app.get("/")
 def root():
-    return "بتعمل شنو هنا يا زول"
+    return "بتعمل شنو هنا يا زول"