diff --git a/workflows/preview_server/preview_server.py b/workflows/preview_server/preview_server.py index b9f68ca..6938e48 100644 --- a/workflows/preview_server/preview_server.py +++ b/workflows/preview_server/preview_server.py @@ -10,13 +10,13 @@ import os import json import random from typing import Dict, Any, Optional +import mimetypes from fastapi import FastAPI, File, UploadFile, HTTPException, Form from fastapi.responses import FileResponse import requests from jsonpath_ng import jsonpath, parse from moviepy.editor import ImageClip -import magic app = FastAPI() lock = asyncio.Lock() # Create an asyncio lock @@ -129,7 +129,7 @@ async def preview_request(request: str = Form(...), input_file: UploadFile = Fil with open(temp_filename, "wb") as buffer: shutil.copyfileobj(input_file.file, buffer) - mime_type = magic.from_file(temp_filename, mime=True) + mime_type, _ = mimetypes.guess_type(temp_filename) if 'video' in mime_type: input_filename = f"{temp_filename}.mp4" diff --git a/workflows/preview_server/server_requirements.txt b/workflows/preview_server/server_requirements.txt index c90bd64..9a5e083 100644 --- a/workflows/preview_server/server_requirements.txt +++ b/workflows/preview_server/server_requirements.txt @@ -4,5 +4,4 @@ fastapi~=0.110.0 python-multipart~=0.0.9 pydantic~=2.6.4 jsonpath-ng~=1.6.1 -moviepy -python-magic \ No newline at end of file +moviepy \ No newline at end of file