works on asr

GooeyAI · Sep 20, 2023 · becb7c9 · becb7c9
1 parent af6048b
commit becb7c9
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/daras_ai_v2/glossary.py b/daras_ai_v2/glossary.py
@@ -13,7 +13,7 @@
 # ================================ Glossary UI ================================
 def glossary_input(
     label="##### Glossary\nUpload a google sheet, csv, or xlsx file.",
-    key="glossary_url",
+    key="glossary_documents",
 ):
     from daras_ai_v2.doc_search_settings_widgets import document_uploader
 

diff --git a/recipes/asr.py b/recipes/asr.py
@@ -38,7 +38,7 @@ class RequestModel(BaseModel):
         selected_model: typing.Literal[tuple(e.name for e in AsrModels)] | None
         language: str | None
         google_translate_target: str | None
-        glossary_url: str | None
+        glossary_documents: list[str] | None
         output_format: typing.Literal[tuple(e.name for e in AsrOutputFormat)] | None
 
     class ResponseModel(BaseModel):
@@ -146,6 +146,9 @@ def run(self, state: dict):
                 source_language=forced_asr_languages.get(
                     selected_model, request.language
                 ),
+                glossary_url=request.glossary_documents[0]
+                if request.glossary_documents
+                else "",
             )
         else:
             # Save the raw ASR text for details view