storytold/storyteller-ml
2
0
Fork 0
mirror of https://github.com/storytold/storyteller-ml.git synced 2026-10-24 08:36:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

preview server: disable cookie check

Browse Source
This commit is contained in:
Kasisnu
2024-04-22 19:52:48 +05:30
parent 49e6910b41
commit a9ca66d9fb
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
workflows/preview_server/preview_server.py
+6 -1
View File
@@ -99,6 +99,8 @@ allowed_keys = ["style", "positive_prompt", "negative_prompt"]
def can_use_vst(cookies): 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.""" """Check if the user is authenticated and if the 'video_style_transfer' feature flag is enabled."""
try: try:
response = requests.get(AUTH_URL, cookies=cookies) 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) positive_prompt = request.get("positive_prompt", None)
negative_prompt = request.get("negative_prompt", None) negative_prompt = request.get("negative_prompt", None)
mime_type = input_file.content_type mime_type = input_file.content_type
if not mime_type:
mime_type, _ = mimetypes.guess_type(input_file.filename)
input_filename_ext = '' input_filename_ext = ''
if 'video' in mime_type: if 'video' in mime_type:
input_filename_ext = ".mp4" 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) clip = clip.set_duration(clip_duration)
input_filename = f"temp_files/{uuid.uuid4()}.mp4" input_filename = f"temp_files/{uuid.uuid4()}.mp4"
clip.write_videofile(input_filename, fps=fps) clip.write_videofile(input_filename, fps=fps)
else:
input_filename = temp_filename
except Exception as e: except Exception as e:
raise HTTPException(status_code=500, detail=f"Failed to save file: {str(e)}") raise HTTPException(status_code=500, detail=f"Failed to save file: {str(e)}")