Update app.py

app.py

@@ -1,15 +1,14 @@
 import dash
 from dash import dcc, html, Input, Output, State, callback
 import dash_bootstrap_components as dbc
-import base64
-import io
 import google.generativeai as genai
 import numpy as np
 import edge_tts
 import asyncio
+import io
 import re
+import base64
 import logging
-import json
 from dash.exceptions import PreventUpdate
 
 # Set up logging
@@ -62,37 +61,36 @@ app.layout = dbc.Container([
                     'margin': '10px 0'
                 }
             ),
-            dcc.RadioItems(
+            dbc.RadioItems(
                 id="duration",
                 options=[
-                    {'label': '1-5 min', 'value': '1-5 min'},
-                    {'label': '5-10 min', 'value': '5-10 min'},
-                    {'label': '10-15 min', 'value': '10-15 min'}
+                    {"label": "1-5 min", "value": "1-5 min"},
+                    {"label": "5-10 min", "value": "5-10 min"},
+                    {"label": "10-15 min", "value": "10-15 min"}
                 ],
-                value='1-5 min',
+                value="1-5 min",
                 inline=True
             ),
-            dcc.RadioItems(
+            dbc.RadioItems(
                 id="num-hosts",
                 options=[
-                    {'label': '1 host', 'value': 1},
-                    {'label': '2 hosts', 'value': 2}
+                    {"label": "1 host", "value": 1},
+                    {"label": "2 hosts", "value": 2}
                 ],
                 value=2,
                 inline=True
             ),
-            dcc.Dropdown(id="lang1-select", options=[{'label': lang, 'value': lang} for lang in language_names.values()], value="English (United States)"),
-            dcc.Dropdown(id="voice1-select"),
-            html.Div([
-                dcc.Dropdown(id="lang2-select", options=[{'label': lang, 'value': lang} for lang in language_names.values()], value="English (United States)"),
-                dcc.Dropdown(id="voice2-select")
-            ], id="second-voice-container"),
+            dbc.Select(id="lang1-select", options=[{"label": lang, "value": lang} for lang in language_names.values()]),
+            dbc.Select(id="voice1-select"),
+            dbc.Select(id="lang2-select", options=[{"label": lang, "value": lang} for lang in language_names.values()]),
+            dbc.Select(id="voice2-select"),
             dbc.Button("Generate Script", id="generate-btn", color="primary", className="mt-3"),
             dbc.Textarea(id="script-output", rows=10, className="mt-3"),
             dbc.Button("Render Podcast", id="render-btn", color="success", className="mt-3"),
-            html.Div(id="audio-output")
+            html.Div(id="audio-output"),
+            dcc.Download(id="download-audio")
         ])
-    ], className="mt-4")
+    ])
 ], fluid=True)
 
 # Callbacks
@@ -101,49 +99,22 @@ app.layout = dbc.Container([
     Input("lang1-select", "value")
 )
 def update_voice1_options(lang):
+    if lang == "None":
+        return []
     selected_lang = next((key for key, value in language_names.items() if value == lang), None)
-    voices = asyncio.run(get_voice_list()).get(selected_lang, [])
-    return [{'label': voice, 'value': voice} for voice in voices]
+    voices = voice_dict.get(selected_lang, [])
+    return [{"label": v, "value": v} for v in voices]
 
 @app.callback(
     Output("voice2-select", "options"),
     Input("lang2-select", "value")
 )
 def update_voice2_options(lang):
+    if lang == "None":
+        return []
     selected_lang = next((key for key, value in language_names.items() if value == lang), None)
-    voices = asyncio.run(get_voice_list()).get(selected_lang, [])
-    return [{'label': voice, 'value': voice} for voice in voices]
-
-@app.callback(
-    Output("second-voice-container", "style"),
-    Input("num-hosts", "value")
-)
-def update_second_voice_visibility(num_hosts):
-    return {'display': 'block' if num_hosts == 2 else 'none'}
-
-@app.callback(
-    Output("content-input", "value"),
-    Input("document-upload", "contents"),
-    State("document-upload", "filename")
-)
-def update_content(contents, filename):
-    if contents is not None:
-        content_type, content_string = contents.split(',')
-        decoded = base64.b64decode(content_string)
-        try:
-            if 'csv' in filename:
-                df = pd.read_csv(io.StringIO(decoded.decode('utf-8')))
-                return df.to_string()
-            elif 'xls' in filename:
-                df = pd.read_excel(io.BytesIO(decoded))
-                return df.to_string()
-            elif 'txt' in filename:
-                return decoded.decode('utf-8')
-            else:
-                return 'Unsupported file type'
-        except Exception as e:
-            return f'Error processing file: {str(e)}'
-    return ''
+    voices = voice_dict.get(selected_lang, [])
+    return [{"label": v, "value": v} for v in voices]
 
 @app.callback(
     Output("script-output", "value"),
@@ -151,36 +122,54 @@ def update_content(contents, filename):
     State("api-key-input", "value"),
     State("content-input", "value"),
     State("duration", "value"),
-    State("num-hosts", "value")
+    State("num-hosts", "value"),
+    prevent_initial_call=True
 )
 def generate_script(n_clicks, api_key, content, duration, num_hosts):
     if n_clicks is None:
         raise PreventUpdate
     try:
-        script = generate_podcast_script(api_key, content, duration, num_hosts)
-        return script
+        return generate_podcast_script(api_key, content, duration, num_hosts)
     except Exception as e:
-        return f"Error generating script: {str(e)}"
+        logger.error(f"Error generating script: {str(e)}")
+        return f"Error: {str(e)}"
 
 @app.callback(
     Output("audio-output", "children"),
+    Output("download-audio", "data"),
     Input("render-btn", "n_clicks"),
     State("api-key-input", "value"),
     State("script-output", "value"),
     State("voice1-select", "value"),
     State("voice2-select", "value"),
-    State("num-hosts", "value")
+    State("num-hosts", "value"),
+    prevent_initial_call=True
 )
-def render_podcast_audio(n_clicks, api_key, script, voice1, voice2, num_hosts):
+async def render_and_download_podcast(n_clicks, api_key, script, voice1, voice2, num_hosts):
     if n_clicks is None:
         raise PreventUpdate
     try:
-        sample_rate, audio_data = asyncio.run(render_podcast(api_key, script, voice1, voice2, num_hosts))
-        audio_base64 = base64.b64encode(audio_data.tobytes()).decode('utf-8')
-        return html.Audio(src=f"data:audio/wav;base64,{audio_base64}", controls=True)
+        sample_rate, audio_data = await render_podcast(api_key, script, voice1, voice2, num_hosts)
+        audio_bytes = audio_data.tobytes()
+        audio_base64 = base64.b64encode(audio_bytes).decode('utf-8')
+        audio_src = f"data:audio/wav;base64,{audio_base64}"
+        return html.Audio(src=audio_src, controls=True), dcc.send_bytes(audio_bytes, "podcast.wav")
     except Exception as e:
-        return html.Div(f"Error rendering podcast: {str(e)}", style={'color': 'red'})
+        logger.error(f"Error rendering podcast: {str(e)}")
+        return html.Div(f"Error: {str(e)}"), None
+
+@app.callback(
+    Output("lang2-select", "style"),
+    Output("voice2-select", "style"),
+    Input("num-hosts", "value")
+)
+def update_second_voice_visibility(num_hosts):
+    if num_hosts == 2:
+        return {"display": "block"}, {"display": "block"}
+    else:
+        return {"display": "none"}, {"display": "none"}
 
+# Run the app
 if __name__ == '__main__':
     print("Starting the Dash application...")
     app.run(debug=True, host='0.0.0.0', port=7860)