Update app.py

app.py

@@ -41,8 +41,10 @@ def generate_podcast_script(api_key, prompt, uploaded_file, duration, num_hosts)
             file_content = uploaded_file.read().decode('utf-8')
             combined_content += "\n" + file_content if combined_content else file_content
         
+        num_hosts = int(num_hosts)  # Convert to integer
+        
         prompt = f"""
-        Create a podcast script for {'one person' if num_hosts == 1 else 'two people'} discussing:
+        Create a podcast script for {num_hosts} {'person' if num_hosts == 1 else 'people'} discussing:
         {combined_content}
         
         Duration: {duration} minutes. Include natural speech, humor, and occasional off-topic thoughts.
@@ -56,6 +58,7 @@ def generate_podcast_script(api_key, prompt, uploaded_file, duration, num_hosts)
         Example: "I can't believe I stayed up all night <yawn> only to find out the meeting was canceled <groan>."
         
         Ensure content flows naturally and stays on topic. Match the script length to {duration} minutes.
+        {'Make sure the script is a monologue for one person.' if num_hosts == 1 else 'Ensure the dialogue alternates between two distinct voices.'}
         """
         
         response = model.generate_content(prompt)