Update Gradio_UI.py

Gradio_UI.py

@@ -13,10 +13,8 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-# import mimetypes # No longer needed if file upload is removed
 import os
 import re
-# import shutil # No longer needed if file upload is removed
 from typing import Optional
 import tempfile 
 from PIL import Image as PILImage 
@@ -169,17 +167,17 @@ def stream_to_gradio(
 
 
 class GradioUI:
-    def __init__(self, agent: MultiStepAgent, file_upload_folder: str | None = None):
+    def __init__(self, agent: MultiStepAgent, file_upload_folder: str | None = None): # Parameter kept for consistency
         if not _is_package_available("gradio"):
             raise ModuleNotFoundError("Install 'gradio': `pip install 'smolagents[gradio]'`")
         self.agent = agent
-        self.file_upload_folder = None # File upload UI is disabled
+        self.file_upload_folder = None # Effectively disabling file uploads from UI side
         self._latest_file_path_for_download = None
 
-    def _check_for_created_file(self):
-        self._latest_file_path_for_download = None 
+    def _get_created_document_path(self): # Renamed for clarity
+        """Checks agent logs for a path from create_document and returns it if valid, else None."""
         if hasattr(self.agent, 'interaction_logs') and self.agent.interaction_logs:
-            print(f"DEBUG Gradio UI: Checking {len(self.agent.interaction_logs)} interaction log entries for created files.")
+            print(f"DEBUG Gradio UI: Checking {len(self.agent.interaction_logs)} interaction log entries for created document paths.")
             for log_entry in reversed(self.agent.interaction_logs): 
                 if isinstance(log_entry, ActionStep):
                     observations = getattr(log_entry, 'observations', None)
@@ -188,8 +186,7 @@ class GradioUI:
                     is_python_interpreter_step = any(tc.name == "python_interpreter" for tc in tool_calls)
 
                     if is_python_interpreter_step and observations and isinstance(observations, str):
-                        # CRITICAL DEBUG LINE: Print the exact observations string
-                        print(f"DEBUG Gradio UI (_check_for_file): Python Interpreter Observations: '''{observations}'''") 
+                        print(f"DEBUG Gradio UI (_get_created_document_path): Python Interpreter Observations: '''{observations[:500]}...'''") 
                         
                         match = re.search(
                             r"(?:Document created \((?:docx|pdf|txt)\):|Document converted to PDF:)\s*(/tmp/[a-zA-Z0-9_]+/generated_document\.(?:docx|pdf|txt))",
@@ -198,52 +195,54 @@ class GradioUI:
                         )
                         
                         if match:
-                            extracted_path = match.group(1) # Path is group 1 due to simplified regex
+                            extracted_path = match.group(1) 
                             print(f"DEBUG Gradio UI: Regex matched. Extracted path: '{extracted_path}'")
                             normalized_path = os.path.normpath(extracted_path)
                             if os.path.exists(normalized_path):
-                                self._latest_file_path_for_download = normalized_path
-                                print(f"DEBUG Gradio UI: File path for download SET: {self._latest_file_path_for_download}")
-                                return True 
+                                print(f"DEBUG Gradio UI: Validated path for download: {normalized_path}")
+                                return normalized_path # Return the path
                             else:
                                 print(f"DEBUG Gradio UI: Path from create_document output ('{normalized_path}') does not exist.")
-                        # else:
-                            # print(f"DEBUG Gradio UI: 'create_document' output pattern not found in this observation block.")
-        print("DEBUG Gradio UI: No valid generated file path (from create_document) found for download button.")
-        return False
+        print("DEBUG Gradio UI: No valid generated document path found in agent logs.")
+        return None # Return None if no valid path found
 
     def interact_with_agent(self, prompt_text: str, current_chat_history: list):
         print(f"DEBUG Gradio: interact_with_agent called with prompt: '{prompt_text}'")
 
         updated_chat_history = current_chat_history + [{"role": "user", "content": prompt_text}]
         
-        yield updated_chat_history, gr.update(visible=False), gr.update(value=None, visible=False)
+        # Initial yield: show user message, keep file_download_display_component hidden
+        yield updated_chat_history, gr.update(value=None, visible=False)
 
         agent_responses_for_history = []
         for msg_dict in stream_to_gradio(self.agent, task=prompt_text, reset_agent_memory=False):
             agent_responses_for_history.append(msg_dict)
-            yield updated_chat_history + agent_responses_for_history, gr.update(visible=False), gr.update(value=None, visible=False) 
+            # Yield progressively to update chat, keep file_download_display_component hidden
+            yield updated_chat_history + agent_responses_for_history, gr.update(value=None, visible=False) 
 
-        file_found_for_download = self._check_for_created_file()
-        
         final_chat_display_content = updated_chat_history + agent_responses_for_history
-        print(f"DEBUG Gradio: Final chat history for display: {len(final_chat_display_content)} messages. File found for download button: {file_found_for_download}")
-        yield final_chat_display_content, gr.update(visible=file_found_for_download), gr.update(value=None, visible=False)
+        
+        # After all agent messages are processed, check for a created document path
+        document_path_to_display = self._get_created_document_path() 
+        
+        if document_path_to_display:
+            print(f"DEBUG Gradio: Document found for display: {document_path_to_display}")
+            # Update chat and make the file component visible with the path
+            yield final_chat_display_content, gr.File.update(value=document_path_to_display, 
+                                                             label=os.path.basename(document_path_to_display), 
+                                                             visible=True)
+        else:
+            # Update chat, keep file component hidden
+            print(f"DEBUG Gradio: No document found for display.")
+            yield final_chat_display_content, gr.File.update(value=None, visible=False)
+
 
-    def log_user_message(self, text_input_value: str):
+    def log_user_message(self, text_input_value: str): # File uploads removed
         full_prompt = text_input_value
         print(f"DEBUG Gradio: Prepared prompt for agent: {full_prompt[:300]}...") 
         return full_prompt, "" 
 
-    def prepare_and_show_download_file(self):
-        if self._latest_file_path_for_download and os.path.exists(self._latest_file_path_for_download):
-            print(f"DEBUG Gradio UI: Preparing download for UI component: {self._latest_file_path_for_download}")
-            return gr.File.update(value=self._latest_file_path_for_download, 
-                                  label=os.path.basename(self._latest_file_path_for_download), 
-                                  visible=True)
-        else:
-            print("DEBUG Gradio UI: No valid file path to prepare for download component.")
-            return gr.File.update(visible=False, value=None) 
+    # prepare_and_show_download_file and download_action_button are removed as we try direct display
 
     def launch(self, **kwargs):
         with gr.Blocks(fill_height=True, theme=gr.themes.Soft(primary_hue=gr.themes.colors.blue)) as demo:
@@ -268,30 +267,28 @@ class GradioUI:
                     )
                 
                 with gr.Column(scale=1): 
-                    # File Upload UI is disabled
-                    # if self.file_upload_folder is not None: 
-                    #     with gr.Accordion("File Upload", open=False): 
-                    #         # ... (upload components would be here)
-                    
-                    with gr.Accordion("Generated File", open=True): 
-                        download_action_button = gr.Button("Download Generated File", visible=False) 
-                        file_download_display_component = gr.File(label="Downloadable Document", visible=False, interactive=False) 
+                    # --- "Generated File" section now directly shows the gr.File component ---
+                    # No separate "Download" button, file appears if created.
+                    gr.Markdown("### Generated Document") # Renamed title for clarity
+                    file_download_display_component = gr.File(
+                        label="Downloadable Document", 
+                        visible=False, # Initially hidden
+                        interactive=False # User cannot upload to this
+                    )
 
             text_message_input.submit(
                 self.log_user_message, 
-                [text_message_input], # Only text_input as file uploads are disabled
+                [text_message_input],
                 [prepared_prompt_for_agent, text_message_input] 
             ).then(
                 self.interact_with_agent, 
                 [prepared_prompt_for_agent, chatbot_display], 
-                [chatbot_display, download_action_button, file_download_display_component] 
+                # Outputs: chatbot, and the file_download_display_component
+                [chatbot_display, file_download_display_component] 
             )
 
-            download_action_button.click(
-                self.prepare_and_show_download_file, 
-                [], 
-                [file_download_display_component] 
-            )
+            # No explicit download_action_button.click event needed anymore
+            
         demo.launch(debug=True, share=kwargs.get("share", False), **kwargs)
 
 __all__ = ["stream_to_gradio", "GradioUI"]