Update app.py

app.py

@@ -379,10 +379,10 @@ with gr.Blocks(theme="CultriX/gradio-theme", css=css, delete_cache=(60, 60)) as
     gr.Markdown("## Multi-Agent Task Solver with Human-in-the-Loop")
 
     with gr.Row():
-        # Wrap the Chatbot in a div with the class "chat-container"
-        with gr.Column(): # Add a column for better layout
-            with gr.Box(elem_classes="chat-container"):  # Use gr.Box for styling
-                chat_output = gr.Chatbot(label="Conversation", type="messages")
+        with gr.Column():  # Add a column for better layout
+            chat_output = gr.Chatbot(label="Conversation", type="messages")
+            chat_output.wrap = gr.HTML("<div class='chat-container'></div>") # Wrap after creation
+
 
     # Hidden state to store the plain-text log.
     log_state = gr.State(value="")
@@ -403,5 +403,4 @@ with gr.Blocks(theme="CultriX/gradio-theme", css=css, delete_cache=(60, 60)) as
         download_button.click(fn=download_log, inputs=log_state, outputs=download_file)
 
 if __name__ == "__main__":
-    demo.launch(share=True)
-
+    demo.launch(share=True)