fdfdff

app_gradio_spaces.py

@@ -132,7 +132,7 @@ def create_chat_ui(setup_status="Not started", services_status="Not started"):
         # Output area
         with gr.Row():
             with gr.Column():
-                chatbot = gr.Chatbot(label="Conversation", height=400)
+                chatbot = gr.Chatbot(label="Conversation", height=400, type="messages")
                 audio_output = gr.Audio(label="Generated Speech", interactive=False)
         
         # Function to handle setup button
@@ -179,15 +179,26 @@ def create_chat_ui(setup_status="Not started", services_status="Not started"):
         
         # Auto-setup on HF Spaces
         if HF_SPACES:
-            # Run setup automatically in a separate thread
-            def auto_setup():
-                time.sleep(2)  # Wait for UI to load
-                setup_output.update(value="Auto-starting setup process...")
-                setup_status = setup_environment()
-                setup_output.update(value=setup_status)
-                services_btn.update(interactive=True)
-            
-            threading.Thread(target=auto_setup, daemon=True).start()
+            def perform_auto_setup_on_load():
+                # Update UI to show setup is starting
+                yield {
+                    setup_output: gr.Textbox.update(value="Auto-starting setup process...")
+                }
+                
+                # Actual setup call
+                final_setup_status_message = setup_environment() 
+
+                # Update UI with final status and enable next button
+                yield {
+                    setup_output: gr.Textbox.update(value=final_setup_status_message),
+                    services_btn: gr.Button.update(interactive=True)
+                }
+
+            demo.load(
+                perform_auto_setup_on_load, 
+                None, 
+                [setup_output, services_btn] 
+            )
     
     return demo