Update app.py

app.py

@@ -128,29 +128,25 @@ def generate(
         # Return None if the TTS is not enabled or the password is incorrect
         return None
 
-# Create a gr.Blocks instance
-with gr.Blocks(outputs=gr.Audio(label="Audio")) as demo: # Add the gr.Audio component to the outputs parameter
-    # Create a gr.Column block with scale=1
-    with gr.Column(scale=1):
-        # Add the chatbot and the additional inputs
-        mychatbot = gr.Chatbot(
-            avatar_images=["./user.png", "./stella.jpg"], bubble_full_width=False, show_label=False, show_copy_button=True, likeable=True
-        )
-        checkbox = gr.Checkbox(label="Enable TTS")
-        passwordInput = gr.Textbox(label="TTS Password", type="password")
-        demo = gr.ChatInterface(
-            fn=generate,  # Replace with your actual generate function
-            chatbot=mychatbot,
-            title="Stella 🌸",
-            retry_btn=None,
-            undo_btn=None,
-            additional_inputs=[checkbox, passwordInput],
-            # Remove the outputs parameter from the ChatInterface
-        )
-    # Create another gr.Column block with scale=1
-    with gr.Column(scale=1):
-        # Add some empty space
-        pass
-
-# Launch the demo
-demo.queue().launch(show_api=False)
+iface = gr.Blocks()
+
+with iface.container():
+    checkbox = gr.Checkbox(label="Enable TTS")
+    passwordInput = gr.Textbox(label="TTS Password", type="password")
+
+    chat = gr.ChatInterface(
+        fn=generate,
+        chatbot=mychatbot,
+        title="Stella ",
+        retry_btn=None,
+        undo_btn=None,
+        additional_inputs=[checkbox, passwordInput],
+    )
+
+    hidden_audio_output = gr.Textbox(label="Audio Output", visible=False)
+    audio_player = gr.Audio(source="Audio Output")
+
+    # Connect the hidden Textbox to the Audio component
+    hidden_audio_output.link(audio_player, source="file")
+
+iface.launch(show_api=False)