Update app.py

app.py

@@ -253,7 +253,7 @@ def get_mp3(text: str, voice: str, audio_model: str, api_key: str = None) -> byt
 
 from functools import wraps
 
-def conditional_llm(model, api_base=None):
+def conditional_llm(model, api_base=None, api_key=None):
     """
     Conditionally apply the @llm decorator based on the api_base parameter.
     If api_base is provided, it applies the @llm decorator with api_base.
@@ -263,7 +263,7 @@ def conditional_llm(model, api_base=None):
         if api_base:
             return llm(model=model, api_base=api_base)(func)
         else:
-            return llm(model=model)(func)
+            return llm(model=model, api_key=api_key)(func)
     return decorator
 
 def generate_audio(
@@ -295,8 +295,10 @@ def generate_audio(
 
     # Configure the LLM based on selected model and api_base
     @retry(retry=retry_if_exception_type(ValidationError))
-    @conditional_llm(model=text_model, api_base=api_base)
-    def generate_dialogue(text: str, intro_instructions: str, text_instructions: str, scratch_pad_instructions: str, prelude_dialog: str, podcast_dialog_instructions: str) -> Dialogue:
+    @conditional_llm(model=text_model, api_base=api_base, api_key=openai_api_key)
+    def generate_dialogue(text: str, intro_instructions: str, text_instructions: str, scratch_pad_instructions: str, 
+                          prelude_dialog: str, podcast_dialog_instructions: str,
+                          ) -> Dialogue:
         """
         {intro_instructions}