Update app.py

app.py

@@ -1,11 +1,13 @@
 from fastapi import FastAPI, File, UploadFile, HTTPException
 from fastapi.responses import StreamingResponse
 from ./image_enhancer import EnhancementMethod, Enhancer
+from vidoe_enhancer import VideoEnhancer
 from pydantic import BaseModel
 from PIL import Image
 from io import BytesIO
 import base64
 import numpy as np
+import magic
 
 class EnhancementRequest(BaseModel):
     method: EnhancementMethod = EnhancementMethod.gfpgan
@@ -23,7 +25,7 @@ app = FastAPI()
 def greet_json():
     return {"Initializing GlamApp Enhancer"}
 
-@app.post("/enhance")
+@app.post("/enhance/image/")
 async def enhance_image(
     file: UploadFile = File(...),
     request: EnhancementRequest = EnhancementRequest()
@@ -48,4 +50,25 @@ async def enhance_image(
         return StreamingResponse(img_byte_arr, media_type="image/png")
 
     except Exception as e:
-        raise HTTPException(status_code=500, detail=str(e))
+        raise HTTPException(status_code=500, detail=str(e))
+        
+@app.post("/enhance/video/")
+async def enhance_video(file: UploadFile = File(...)):
+    enhancer = VideoEnhancer()
+    file_header = await file.read(1024)
+    file.file.seek(0)
+    mime = magic.Magic(mime=True)
+    file_mime_type = mime.from_buffer(file_header)
+    
+    accepted_mime_types = [
+        'video/mp4',
+        'video/mpeg',
+        'video/x-msvideo',
+        'video/quicktime',
+        'video/x-matroska',
+        'video/webm'
+    ]
+    
+    if file_mime_type not in accepted_mime_types:
+        raise HTTPException(status_code=400, detail="Invalid file type. Please upload a video file.")
+    return await enhancer.stream_enhanced_video(file)