Update app.py

app.py

@@ -30,7 +30,7 @@ class PodcastGenerator:
     def __init__(self):
         pass
 
-    async def generate_script(self, prompt: str, language: str, api_key: str, file_obj=None, progress=None) -> Dict:
+    async def generate_script(self, prompt: str, language: str,  file_obj=None, progress=None) -> Dict:
         example = """
 {
     "topic": "AGI",
@@ -209,14 +209,14 @@ Return the result strictly as a JSON object in the format:
             
         return output_filename
 
-    async def generate_podcast(self, input_text: str, language: str, speaker1: str, speaker2: str, api_key: str, file_obj=None, progress=None) -> str:
+    async def generate_podcast(self, input_text: str, language: str, speaker1: str, speaker2: str, file_obj=None, progress=None) -> str:
         try:
             if progress:
                 progress(0.1, "Starting podcast generation...")
                 
             # Set overall timeout for the entire process
             return await asyncio.wait_for(
-                self._generate_podcast_internal(input_text, language, speaker1, speaker2, api_key, file_obj, progress),
+                self._generate_podcast_internal(input_text, language, speaker1, speaker2, file_obj, progress),
                 timeout=600  # 10 minutes total timeout
             )
         except asyncio.TimeoutError:
@@ -224,11 +224,11 @@ Return the result strictly as a JSON object in the format:
         except Exception as e:
             raise Exception(f"Error generating podcast: {str(e)}")
     
-    async def _generate_podcast_internal(self, input_text: str, language: str, speaker1: str, speaker2: str, api_key: str, file_obj=None, progress=None) -> str:
+    async def _generate_podcast_internal(self, input_text: str, language: str, speaker1: str, speaker2: str, file_obj=None, progress=None) -> str:
         if progress:
             progress(0.2, "Generating podcast script...")
             
-        podcast_json = await self.generate_script(input_text, language, api_key, file_obj, progress)
+        podcast_json = await self.generate_script(input_text, language,  file_obj, progress)
         
         if progress:
             progress(0.5, "Converting text to speech...")
@@ -281,7 +281,7 @@ Return the result strictly as a JSON object in the format:
         combined_audio = await self.combine_audio_files(audio_files, progress)
         return combined_audio
 
-async def process_input(input_text: str, input_file, language: str, speaker1: str, speaker2: str, api_key: str = "", progress=None) -> str:
+async def process_input(input_text: str, input_file, language: str, speaker1: str, speaker2: str,  progress=None) -> str:
     start_time = time.time()
 
     voice_names = {
@@ -305,7 +305,7 @@ async def process_input(input_text: str, input_file, language: str, speaker1: st
         api_key = ""  # No API key needed for local model
 
         podcast_generator = PodcastGenerator()
-        podcast = await podcast_generator.generate_podcast(input_text, language, speaker1, speaker2, api_key, input_file, progress)
+        podcast = await podcast_generator.generate_podcast(input_text, language, speaker1, speaker2, input_file, progress)
 
         end_time = time.time()
         print(f"Total podcast generation time: {end_time - start_time:.2f} seconds")
@@ -322,7 +322,7 @@ async def process_input(input_text: str, input_file, language: str, speaker1: st
             raise Exception(f"Error: {error_msg}")
 
 # Gradio UI
-def generate_podcast_gradio(input_text, input_file, language, speaker1, speaker2, api_key, progress=gr.Progress()):
+def generate_podcast_gradio(input_text, input_file, language, speaker1, speaker2,  progress=gr.Progress()):
     # Handle the file if uploaded
     file_obj = None
     if input_file is not None:
@@ -339,7 +339,6 @@ def generate_podcast_gradio(input_text, input_file, language, speaker1, speaker2
         language, 
         speaker1, 
         speaker2, 
-        api_key,
         progress_callback
     ))