more usererror handling

recipes/DeforumSD.py

@@ -9,6 +9,7 @@
 from bots.models import Workflow
 from daras_ai_v2.base import BasePage
 from daras_ai_v2.enum_selector_widget import enum_selector
+from daras_ai_v2.exceptions import UserError
 from daras_ai_v2.gpu_server import call_celery_task_outfile
 from daras_ai_v2.loom_video_widget import youtube_video
 from daras_ai_v2.safety_checker import safety_checker
@@ -481,4 +482,4 @@ def run(self, state: dict):
         except RuntimeError as e:
             msg = "\n\n".join(e.args).lower()
             if "key frame string not correctly formatted" in msg:
-                raise st.UserError(str(e)) from e
+                raise UserError(str(e)) from e

routers/root.py

@@ -1,6 +1,5 @@
 import datetime
 import os.path
-import subprocess
 import tempfile
 import typing
 from time import time
@@ -34,7 +33,7 @@
 from daras_ai_v2.bots import request_json
 from daras_ai_v2.copy_to_clipboard_button_widget import copy_to_clipboard_scripts
 from daras_ai_v2.db import FIREBASE_SESSION_COOKIE
-from daras_ai_v2.exceptions import ffmpeg
+from daras_ai_v2.exceptions import ffmpeg, UserError
 from daras_ai_v2.manage_api_keys_widget import manage_api_keys
 from daras_ai_v2.meta_content import build_meta_tags, raw_build_meta_tags
 from daras_ai_v2.meta_preview_url import meta_preview_url
@@ -187,13 +186,16 @@ def file_upload(request: Request, form_data: FormData = Depends(request_form_fil
         ) as infile:
             infile.write(data)
             infile.flush()
-            if not check_wav_audio_format(infile.name):
-                with tempfile.NamedTemporaryFile(suffix=".wav") as outfile:
-                    ffmpeg("-i", infile.name, *FFMPEG_WAV_ARGS, outfile.name)
-
-                    filename += ".wav"
-                    content_type = "audio/wav"
-                    data = outfile.read()
+            try:
+                if not check_wav_audio_format(infile.name):
+                    with tempfile.NamedTemporaryFile(suffix=".wav") as outfile:
+                        ffmpeg("-i", infile.name, *FFMPEG_WAV_ARGS, outfile.name)
+
+                        filename += ".wav"
+                        content_type = "audio/wav"
+                        data = outfile.read()
+            except UserError as e:
+                return Response(content=str(e), status_code=400)
 
     if content_type.startswith("image/"):
         with Image(blob=data) as img: