Update app.py

app.py

@@ -10,6 +10,8 @@ import re
 import base64
 import logging
 from dash.exceptions import PreventUpdate
+import pandas as pd
+import time
 
 # Set up logging
 logging.basicConfig(level=logging.INFO)
@@ -195,9 +197,23 @@ app.layout = dbc.Container([
             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.Button("Generate Script", id="generate-btn", color="primary", className="mt-3"),
-            dbc.Textarea(id="script-output", rows=10, className="my-3"),
-            dbc.Button("Render Podcast", id="render-btn", color="success", className="mt-3"),
-            html.Div(id="audio-output", className="my-3"),
+            dcc.Loading(
+                id="loading-script",
+                type="default",
+                children=[
+                    dbc.Progress(id="script-progress", value=0, className="my-3"),
+                    dbc.Textarea(id="script-output", rows=10, className="my-3"),
+                ]
+            ),
+            dbc.Button("Generate Podcast", id="generate-podcast-btn", color="success", className="mt-3"),
+            dcc.Loading(
+                id="loading-podcast",
+                type="default",
+                children=[
+                    dbc.Progress(id="podcast-progress", value=0, className="my-3"),
+                    html.Div(id="audio-output", className="my-3"),
+                ]
+            ),
             dcc.Download(id="download-audio")
         ])
     ])
@@ -227,50 +243,59 @@ def update_voice2_options(lang):
     return [{"label": v, "value": v} for v in voices]
 
 @app.callback(
-    Output("script-output", "value"),
+    [Output("script-output", "value"),
+     Output("script-progress", "value")],
     Input("generate-btn", "n_clicks"),
-    State("api-key-input", "value"),
-    State("content-input", "value"),
-    State("duration", "value"),
-    State("num-hosts", "value"),
+    [State("api-key-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):
     if n_clicks is None:
         raise PreventUpdate
     try:
-        return generate_podcast_script(api_key, content, duration, num_hosts)
+        for i in range(10):
+            time.sleep(0.5)  # Simulate progress
+            yield dash.no_update, (i + 1) * 10
+        script = generate_podcast_script(api_key, content, duration, num_hosts)
+        return script, 100
     except Exception as e:
         logger.error(f"Error generating script: {str(e)}")
-        return f"Error: {str(e)}"
+        return f"Error: {str(e)}", 0
 
 @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"),
+    [Output("audio-output", "children"),
+     Output("download-audio", "data"),
+     Output("podcast-progress", "value")],
+    Input("generate-podcast-btn", "n_clicks"),
+    [State("api-key-input", "value"),
+     State("script-output", "value"),
+     State("voice1-select", "value"),
+     State("voice2-select", "value"),
+     State("num-hosts", "value")],
     prevent_initial_call=True
 )
 async def render_and_download_podcast(n_clicks, api_key, script, voice1, voice2, num_hosts):
     if n_clicks is None:
         raise PreventUpdate
     try:
+        for i in range(10):
+            time.sleep(0.5)  # Simulate progress
+            yield dash.no_update, dash.no_update, (i + 1) * 10
         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")
+        return html.Audio(src=audio_src, controls=True), dcc.send_bytes(audio_bytes, "podcast.wav"), 100
     except Exception as e:
         logger.error(f"Error rendering podcast: {str(e)}")
-        return html.Div(f"Error: {str(e)}"), None
+        return html.Div(f"Error: {str(e)}"), None, 0
 
 @app.callback(
-    Output("lang2-select", "style"),
-    Output("voice2-select", "style"),
+    [Output("lang2-select", "style"),
+     Output("voice2-select", "style")],
     Input("num-hosts", "value")
 )
 def update_second_voice_visibility(num_hosts):