Update app.py

app.py

@@ -43,9 +43,8 @@ async def generate_conversation(script):
     combined = concatenate_audioclips(audio_clips)
 
     # Create temporary file for the combined output
-    temp_output_file = tempfile.NamedTemporaryFile(suffix='.mp3', delete=False)
-    output_filename = temp_output_file.name
-
+    output_filename = tempfile.NamedTemporaryFile(suffix='.mp3', delete=False).name
+    
     # Save the combined file
     combined.write_audiofile(output_filename)
     print(f"Combined audio saved as: {output_filename}")
@@ -58,7 +57,7 @@ async def generate_conversation(script):
     return output_filename
 
 # Function to generate podcast based on user input
-def generate_podcast(topic, seed):
+async def generate_podcast(topic, seed=42):
     system_instructions = '''[SYSTEM] You are an educational podcast generator. You have to create a podcast between Alice and Bob that gives an overview of the topic given by the user.
   Please provide the script in the following JSON format:
     {
@@ -72,7 +71,7 @@ def generate_podcast(topic, seed):
     Be concise.
     '''
 
-    text = f" Topic: {topic}"
+    text = f" Topic: {topic} json:"
     formatted_prompt = system_instructions + text
     stream = Client.text_generation(formatted_prompt, max_new_tokens=1024, seed=seed, stream=True, details=True, return_full_text=False)
     
@@ -80,43 +79,37 @@ def generate_podcast(topic, seed):
     for response in stream:
         if not response.token.text == "</s>":
             generated_script += response.token.text
+    
+    print("Generated Script:"+generated_script)
+
+    # Check if the generated_script is empty or not valid JSON
+    if not generated_script or not generated_script.strip().startswith('{'):
+        raise ValueError("Failed to generate a valid script.")
+
 
-    # Generate the podcast
     script_json = json.loads(generated_script)  # Use the generated script as input
-    output_filename = asyncio.run(generate_conversation(script_json))
+    output_filename = await generate_conversation(script_json)
     print("Output File:"+output_filename)
 
     # Read the generated audio file
-    with open(output_filename, "rb") as f:
-        audio_bytes = f.read()
-
-    # Clean up the final output temporary file
-    os.remove(output_filename)
-    print(f"Deleted temporary file: {output_filename}")
-
-    return audio_bytes
+    return output_filename
 
+# Gradio Interface
 DESCRIPTION = """ # <center><b>PODGEN 📻</b></center>
         ### <center>Generate a podcast on any topic</center>
         ### <center>Use the Power of llms to understand any topic better</center>
         """
-
 with gr.Blocks(css="#important{display: none;}") as demo:    
     gr.Markdown(DESCRIPTION)
     with gr.Row():
         input = gr.Textbox(label="Topic", placeholder="Enter a topic")
-        output = gr.Audio(label="Podgen", type="filepath",
-                        interactive=False,
-                        autoplay=True,
-                        elem_classes="audio")
+        output = gr.Audio(label="Podgen", type="filepath", interactive=False, autoplay=True, elem_classes="audio")
+        
         gr.Interface(
-            batch=True,
-            max_batch_size=10, 
             fn=generate_podcast, 
             inputs=[input],
-            outputs=[output],)  
-        
-    
+            outputs=[output],
+        )  
 
 if __name__ == "__main__":
     demo.queue(max_size=200).launch()