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 huggingface_hub import HfApi
 
 # Set up logging
 logging.basicConfig(level=logging.INFO)
@@ -20,8 +22,12 @@ 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 the API key from HF secret
+gemini_api_key = os.environ.get("GEMINI_API_KEY")
+if not gemini_api_key:
+    raise ValueError("GEMINI_API_KEY not found in environment variables. Please set it as a secret in your Hugging Face Space.")
+
+genai.configure(api_key=gemini_api_key)
 
 def generate_podcast_script(api_key, content, duration, num_hosts):
     genai.configure(api_key=api_key)
@@ -40,6 +46,8 @@ def generate_podcast_script(api_key, content, duration, num_hosts):
         Do not use any special characters or markdown. Only include the monologue with proper punctuation.
         Ensure the content flows naturally and stays relevant to the topic.
         Limit the script length to match the requested duration of {duration}.
+        Do not put an intro our outro music as I only need the dialog
+        The dialog must have proper punctuation like apostrophes
         """
     else:
         prompt = f"""
@@ -54,6 +62,8 @@ def generate_podcast_script(api_key, content, duration, num_hosts):
         Do not use any special characters or markdown. Only include the alternating dialogue lines with proper punctuation.
         Ensure the conversation flows naturally and stays relevant to the topic.
         Limit the script length to match the requested duration of {duration}.
+        Do not put an intro our outro music as I only need the dialog
+        The dialog must have proper punctuation like apostrophes
         """
     
     response = model.generate_content(prompt)
@@ -155,7 +165,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',
@@ -246,26 +255,37 @@ 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:
         for i in range(10):
             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)  # Remove api_key parameter
+        model = genai.GenerativeModel('gemini-2.5-pro-preview-03-25')
         return script, 100
     except Exception as e:
         logger.error(f"Error generating script: {str(e)}")
         return f"Error: {str(e)}", 0
 
+@app.callback(
+    [Output("audio-output", "children"),
+     Output("download-audio", "data"),
+     Output("podcast-progress", "value")],
+    Input("generate-podcast-btn", "n_clicks"),
+    [State("script-output", "value"),
+     State("voice1-select", "value"),
+     State("voice2-select", "value"),
+     State("num-hosts", "value")],
+    prevent_initial_call=True
+)
+
 @app.callback(
     [Output("audio-output", "children"),
      Output("download-audio", "data"),
@@ -283,8 +303,7 @@ def render_and_download_podcast(n_clicks, api_key, script, voice1, voice2, num_h
         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))  # Remove api_key parameter
         # Convert numpy array to WAV
         wav_audio = AudioSegment(
             audio_data.tobytes(),