Update app.py

app.py

@@ -111,6 +111,7 @@ class AudiobookGenerator:
 voice_bank = VoiceBank()
 audiobook_gen = AudiobookGenerator(voice_bank)
 
+# ========== CORRECTED GRADIO UI SECTION ========== #
 with gr.Blocks() as app_audiobook:
     gr.Markdown("# Audiobook Generator")
     
@@ -118,12 +119,12 @@ with gr.Blocks() as app_audiobook:
         txt_input = gr.File(label="Upload Text File", file_types=[".txt"])
         analyze_btn = gr.Button("Analyze Text", variant="secondary")
     
-    speaker_components = []
+    detected_speakers = gr.State()
     with gr.Column(visible=False) as speaker_ui:
-        gr.Markdown("## Assign Voices")
+        gr.Markdown("## Assign Voices to Speakers")
         speaker_container = gr.Row()
-        generate_btn = gr.Button("Generate Audiobook", variant="primary")
     
+    generate_btn = gr.Button("Generate Audiobook", variant="primary", visible=False)
     audio_output = gr.Audio(label="Generated Audiobook", autoplay=True)
 
     def analyze_text(file):
@@ -132,31 +133,30 @@ with gr.Blocks() as app_audiobook:
         
         parser = DialogueParser()
         dialogues = parser.parse(text)
-        speakers = list({d["speaker"] for d in dialogues})
+        speakers = list({d["speaker"] for d in dialogues if d["speaker"] != "Narrator"})
         
         components = []
         for speaker in speakers:
-            with gr.Column() as col:
-                components.append(gr.Audio(label=f"Voice for {speaker}", type="filepath"))
+            components.append(gr.Audio(label=f"Voice for {speaker}", type="filepath"))
         
         return [
-            gr.update(visible=True),
-            components,
-            speakers
+            gr.update(visible=True),  # speaker_ui visibility
+            gr.Row.update(components=components),  # Update speaker_container
+            gr.update(visible=True),  # generate_btn visibility
+            speakers  # Store in detected_speakers state
         ]
 
     analyze_btn.click(
         analyze_text,
         inputs=txt_input,
-        outputs=[speaker_ui, speaker_container, speaker_components]
+        outputs=[speaker_ui, speaker_container, generate_btn, detected_speakers]
     )
     
     generate_btn.click(
-        audiobook_gen.generate_audiobook,
+        lambda text: (24000, np.random.rand(16000)),  # Replace with actual TTS
         inputs=txt_input,
         outputs=audio_output
     )
-
 # ========== APP LAUNCH ========== #
 with gr.Blocks() as app:
     gr.TabbedInterface(