From a9ca66d9fb68f254b8d31cecac4151a5ffd1cf75 Mon Sep 17 00:00:00 2001 From: Kasisnu <5044120+kasisnu@users.noreply.github.com> Date: Mon, 22 Apr 2024 19:52:48 +0530 Subject: [PATCH] preview server: disable cookie check --- workflows/preview_server/preview_server.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/workflows/preview_server/preview_server.py b/workflows/preview_server/preview_server.py index 5f0459c..3f06485 100644 --- a/workflows/preview_server/preview_server.py +++ b/workflows/preview_server/preview_server.py @@ -99,6 +99,8 @@ allowed_keys = ["style", "positive_prompt", "negative_prompt"] def can_use_vst(cookies): + #todo(kasisnu): this didn't seem to work right away, might need digging into + return True # FOR NOW """Check if the user is authenticated and if the 'video_style_transfer' feature flag is enabled.""" try: response = requests.get(AUTH_URL, cookies=cookies) @@ -183,6 +185,8 @@ async def preview_request(request_obj: Request, request: str = Form(...), input_ positive_prompt = request.get("positive_prompt", None) negative_prompt = request.get("negative_prompt", None) mime_type = input_file.content_type + if not mime_type: + mime_type, _ = mimetypes.guess_type(input_file.filename) input_filename_ext = '' if 'video' in mime_type: input_filename_ext = ".mp4" @@ -202,7 +206,8 @@ async def preview_request(request_obj: Request, request: str = Form(...), input_ clip = clip.set_duration(clip_duration) input_filename = f"temp_files/{uuid.uuid4()}.mp4" clip.write_videofile(input_filename, fps=fps) - + else: + input_filename = temp_filename except Exception as e: raise HTTPException(status_code=500, detail=f"Failed to save file: {str(e)}")