Update app.py

app.py

@@ -12,6 +12,8 @@ import logging
 from dash.exceptions import PreventUpdate
 import pandas as pd
 import time
+import os
+from pydub import AudioSegment
 
 # Set up logging
 logging.basicConfig(level=logging.INFO)
@@ -20,11 +22,10 @@ logger = logging.getLogger(__name__)
 # Initialize Dash app
 app = dash.Dash(__name__, external_stylesheets=[dbc.themes.BOOTSTRAP])
 
-# Initialize Gemini AI
-genai.configure(api_key='YOUR_GEMINI_API_KEY')
+# Initialize Gemini AI with environment variable
+genai.configure(api_key=os.environ.get('GEMINI_API_KEY'))
 
-def generate_podcast_script(api_key, content, duration, num_hosts):
-    genai.configure(api_key=api_key)
+def generate_podcast_script(content, duration, num_hosts):
     model = genai.GenerativeModel('gemini-2.5-pro-preview-03-25')
     
     if num_hosts == 1:
@@ -70,7 +71,7 @@ async def text_to_speech(text, voice):
     audio.seek(0)
     return audio.read()
 
-async def render_podcast(api_key, script, voice1, voice2, num_hosts):
+async def render_podcast(script, voice1, voice2, num_hosts):
     lines = [line for line in script.split('\n') if line.strip()]
     audio_segments = []
     
@@ -155,7 +156,6 @@ app.layout = dbc.Container([
     
     dbc.Card([
         dbc.CardBody([
-            dbc.Input(id="api-key-input", type="password", placeholder="Enter your Gemini API Key"),
             dbc.Textarea(id="content-input", placeholder="Paste your content or upload a document", rows=5, className="my-3"),
             dcc.Upload(
                 id='document-upload',
@@ -192,10 +192,28 @@ app.layout = dbc.Container([
                 inline=True,
                 className="my-3"
             ),
-            dbc.Select(id="lang1-select", options=[{"label": lang, "value": lang} for lang in language_names.values()], className="my-2"),
-            dbc.Select(id="voice1-select", className="my-2"),
-            dbc.Select(id="lang2-select", options=[{"label": lang, "value": lang} for lang in language_names.values()], className="my-2"),
-            dbc.Select(id="voice2-select", className="my-2"),
+            dbc.Select(
+                id="lang1-select", 
+                options=[{"label": lang, "value": lang} for lang in language_names.values()], 
+                value="English (United States)",
+                className="my-2"
+            ),
+            dbc.Select(
+                id="voice1-select", 
+                value="en-US-AriaNeural",
+                className="my-2"
+            ),
+            dbc.Select(
+                id="lang2-select", 
+                options=[{"label": lang, "value": lang} for lang in language_names.values()], 
+                value="English (United States)",
+                className="my-2"
+            ),
+            dbc.Select(
+                id="voice2-select", 
+                value="en-US-BrianNeural",
+                className="my-2"
+            ),
             dbc.Button("Generate Script", id="generate-btn", color="primary", className="mt-3"),
             dcc.Loading(
                 id="loading-script",
@@ -219,7 +237,7 @@ app.layout = dbc.Container([
     ])
 ], fluid=True)
 
-# Callbacks
+# Callbacks (continued)
 @app.callback(
     Output("voice1-select", "options"),
     Input("lang1-select", "value")
@@ -246,13 +264,12 @@ def update_voice2_options(lang):
     [Output("script-output", "value"),
      Output("script-progress", "value")],
     Input("generate-btn", "n_clicks"),
-    [State("api-key-input", "value"),
-     State("content-input", "value"),
+    [State("content-input", "value"),
      State("duration", "value"),
      State("num-hosts", "value")],
     prevent_initial_call=True
 )
-def generate_script(n_clicks, api_key, content, duration, num_hosts):
+def generate_script(n_clicks, content, duration, num_hosts):
     if n_clicks is None:
         raise PreventUpdate
     try:
@@ -260,7 +277,7 @@ def generate_script(n_clicks, api_key, content, duration, num_hosts):
             time.sleep(0.5)  # Simulate progress
             # Instead of yielding, we'll just pass and update at the end
             pass
-        script = generate_podcast_script(api_key, content, duration, num_hosts)
+        script = generate_podcast_script(content, duration, num_hosts)
         return script, 100
     except Exception as e:
         logger.error(f"Error generating script: {str(e)}")
@@ -271,19 +288,18 @@ def generate_script(n_clicks, api_key, content, duration, num_hosts):
      Output("download-audio", "data"),
      Output("podcast-progress", "value")],
     Input("generate-podcast-btn", "n_clicks"),
-    [State("api-key-input", "value"),
-     State("script-output", "value"),
+    [State("script-output", "value"),
      State("voice1-select", "value"),
      State("voice2-select", "value"),
      State("num-hosts", "value")],
     prevent_initial_call=True
 )
-def render_and_download_podcast(n_clicks, api_key, script, voice1, voice2, num_hosts):
+def render_and_download_podcast(n_clicks, script, voice1, voice2, num_hosts):
     if n_clicks is None:
         raise PreventUpdate
     try:
         # Run the async function in a synchronous context
-        sample_rate, audio_data = asyncio.run(render_podcast(api_key, script, voice1, voice2, num_hosts))
+        sample_rate, audio_data = asyncio.run(render_podcast(script, voice1, voice2, num_hosts))
         
         # Convert numpy array to WAV
         wav_audio = AudioSegment(
@@ -351,4 +367,4 @@ def update_content(contents, filename):
 if __name__ == '__main__':
     print("Starting the Dash application...")
     app.run(debug=True, host='0.0.0.0', port=7860)
-    print("Dash application has finished running.")
+    print("Dash application has finished running.")