Update Gradio_UI.py

Gradio_UI.py

@@ -280,6 +280,7 @@ class GradioUI:
         with gr.Blocks(fill_height=True) as demo:
             stored_messages = gr.State([])
             file_uploads_log = gr.State([])
+
             chatbot = gr.Chatbot(
                 label="Agent",
                 avatar_images=(
@@ -288,7 +289,8 @@ class GradioUI:
                 ),
                 scale=1,
             )
-            # If an upload folder is provided, enable the upload feature
+
+            # Optional: File uploads
             if self.file_upload_folder is not None:
                 upload_file = gr.File(label="Upload a file")
                 upload_status = gr.Textbox(label="Upload Status", interactive=False, visible=False)
@@ -297,15 +299,22 @@ class GradioUI:
                     [upload_file, file_uploads_log],
                     [upload_status, file_uploads_log],
                 )
+
             text_input = gr.Textbox(lines=1, label="Chat Message")
+
             text_input.submit(
                 self.log_user_message,
                 [text_input, file_uploads_log],
                 [stored_messages, text_input],
-            ).then(self.interact_with_agent, [stored_messages, chatbot], [chatbot])
+            ).then(
+                self.interact_with_agent,
+                [stored_messages, chatbot],
+                [chatbot]
+            )
 
         demo.launch(debug=True, share=True, **kwargs)
 
+
     def _chat(self, message: str, history: list):
         """
         Chat handler used by Gradio's ChatInterface.