Update Gradio_UI.py

Gradio_UI.py

@@ -264,6 +264,33 @@ class GradioUI:
         with gr.Blocks(fill_height=True) as demo:
             stored_messages = gr.State([])
             file_uploads_log = gr.State([])
+            
+            # Upload UI row (only if file_upload_folder is set)
+            if self.file_upload_folder is not None:
+                with gr.Row():
+                    upload_file = gr.File(
+                        label="Upload a file (PDF, DOCX, TXT)",
+                        file_types=['.pdf', '.docx', '.txt'],
+                        interactive=True,
+                    )
+                    upload_status = gr.Textbox(
+                        label="Upload Status",
+                        interactive=False,
+                        visible=True,
+                        max_lines=1,
+                        show_label=True,
+                    )
+                # Hook upload event
+                upload_file.change(
+                    self.upload_file,
+                    inputs=[upload_file, file_uploads_log],
+                    outputs=[upload_status, file_uploads_log],
+                )
+            else:
+                # If no upload folder provided, these widgets don’t exist
+                upload_file = None
+                upload_status = None
+    
             chatbot = gr.Chatbot(
                 label="Agent",
                 type="messages",
@@ -275,24 +302,14 @@ class GradioUI:
                 scale=1,
             )
     
-            if self.file_upload_folder is not None:
-                upload_file = gr.File(label="Upload PDF or Doc", file_types=['.pdf', '.docx', '.txt'])
-                upload_status = gr.Textbox(label="Upload Status", interactive=False, visible=True)
-                upload_file.change(
-                    self.upload_file,
-                    [upload_file, file_uploads_log],
-                    [upload_status, file_uploads_log],
-                )
-    
-            text_input = gr.Textbox(lines=1, label="Chat Message", placeholder="Ask your question here...")
+            text_input = gr.Textbox(lines=1, label="Chat Message")
     
+            # When user submits the text input
             def on_submit(text, files_log):
-                if not text.strip():
-                    return gr.update(value="Please enter a question before submitting."), ""
                 combined_prompt = (
                     text
                     + (
-                        f"\n\nI have the following files you can use: {files_log}"
+                        f"\nYou have been provided with these files, which might be helpful or not: {files_log}"
                         if len(files_log) > 0
                         else ""
                     )
@@ -301,17 +318,14 @@ class GradioUI:
     
             text_input.submit(
                 on_submit,
-                [text_input, file_uploads_log],
-                [stored_messages, text_input],
+                inputs=[text_input, file_uploads_log],
+                outputs=[stored_messages, text_input],
             ).then(
                 self.interact_with_agent,
-                [stored_messages, chatbot],
-                [chatbot],
+                inputs=[stored_messages, chatbot],
+                outputs=chatbot,
             )
     
         demo.launch(debug=True, share=True, **kwargs)
 
-
-
-
 __all__ = ["stream_to_gradio", "GradioUI"]