Update Gradio_UI.py

Gradio_UI.py

@@ -215,15 +215,15 @@ class GradioUI:
         import gradio as gr
 
         if file is None:
-            return gr.Textbox("No file uploaded", visible=True), file_uploads_log, None
+            return gr.Textbox("No file uploaded", visible=True), file_uploads_log
 
         try:
             mime_type, _ = mimetypes.guess_type(file.name)
         except Exception as e:
-            return gr.Textbox(f"Error: {e}", visible=True), file_uploads_log, None
+            return gr.Textbox(f"Error: {e}", visible=True), file_uploads_log
 
         if mime_type not in allowed_file_types:
-            return gr.Textbox("File type disallowed", visible=True), file_uploads_log, None
+            return gr.Textbox("File type disallowed", visible=True), file_uploads_log
 
         # Sanitize file name
         original_name = os.path.basename(file.name)
@@ -242,29 +242,28 @@ class GradioUI:
         sanitized_name = "".join(sanitized_name)
 
         # Save the uploaded file to the specified folder
-        output_path = os.path.join(self.file_upload_folder, sanitized_name)
-        shutil.copyfile(file.name, output_path)
-        file_uploads_log.append(output_path)
-    
-        return gr.Textbox(f"Uploaded: {sanitized_name}", visible=True), file_uploads_log, output_path
-    
-    def log_user_message(self, message, file_uploads_log, messages):
-        import gradio as gr
-    
-        if not messages:
-            messages = []
-    
-        messages.append(gr.ChatMessage(role="user", content=message))
-        return messages, ""
-    
+        file_path = os.path.join(self.file_upload_folder, os.path.basename(sanitized_name))
+        shutil.copy(file.name, file_path)
+
+        return gr.Textbox(f"File uploaded: {file_path}", visible=True), file_uploads_log + [file_path]
+
+    def log_user_message(self, text_input, file_uploads_log):
+        return (
+            text_input
+            + (
+                f"\nYou have been provided with these files, which might be helpful or not: {file_uploads_log}"
+                if len(file_uploads_log) > 0
+                else ""
+            ),
+            "",
+        )
+
     def launch(self, **kwargs):
         import gradio as gr
-    
+
         with gr.Blocks(fill_height=True) as demo:
             stored_messages = gr.State([])
             file_uploads_log = gr.State([])
-            current_file_path = gr.State(None)
-    
             chatbot = gr.Chatbot(
                 label="Agent",
                 type="messages",
@@ -275,44 +274,23 @@ class GradioUI:
                 resizeable=True,
                 scale=1,
             )
-    
+            # If an upload folder is provided, enable the upload feature
             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)
-    
                 upload_file.change(
                     self.upload_file,
                     [upload_file, file_uploads_log],
-                    [upload_status, file_uploads_log, current_file_path],
+                    [upload_status, file_uploads_log],
                 )
-    
-                # New button to send the uploaded document to the agent
-                send_doc_btn = gr.Button("Send Document")
-    
-                # Handler function for document sending
-                def send_document(file_path, messages):
-                    if not file_path:
-                        messages.append(gr.ChatMessage(role="assistant", content="No document uploaded."))
-                        return messages
-                    # Construct prompt with file path
-                    prompt = f"Please process this document: {file_path}"
-                    messages.append(gr.ChatMessage(role="user", content=prompt))
-                    for msg in stream_to_gradio(self.agent, task=prompt, reset_agent_memory=False):
-                        messages.append(msg)
-                        yield messages
-                    yield messages
-    
-                send_doc_btn.click(
-                    send_document,
-                    inputs=[current_file_path, stored_messages],
-                    outputs=chatbot,
-                )
-    
             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, file_uploads_log],
                 [stored_messages, text_input],
             ).then(self.interact_with_agent, [stored_messages, chatbot], [chatbot])
-    
-        demo.launch(debug=True, share=True, **kwargs)
+
+        demo.launch(debug=True, share=True, **kwargs)
+
+
+__all__ = ["stream_to_gradio", "GradioUI"]