Update Gradio_UI.py

Gradio_UI.py

@@ -167,15 +167,14 @@ def stream_to_gradio(
 
 
 class GradioUI:
-    def __init__(self, agent: MultiStepAgent, file_upload_folder: str | None = None): # Parameter kept for consistency
+    def __init__(self, agent: MultiStepAgent, file_upload_folder: str | None = None):
         if not _is_package_available("gradio"):
             raise ModuleNotFoundError("Install 'gradio': `pip install 'smolagents[gradio]'`")
         self.agent = agent
-        self.file_upload_folder = None # Effectively disabling file uploads from UI side
+        self.file_upload_folder = None 
         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."""
+    def _get_created_document_path(self):
         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 document paths.")
             for log_entry in reversed(self.agent.interaction_logs): 
@@ -186,7 +185,8 @@ 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):
-                        print(f"DEBUG Gradio UI (_get_created_document_path): Python Interpreter Observations: '''{observations[:500]}...'''") 
+                        # CRITICAL DEBUG LINE: Print the exact observations string
+                        print(f"DEBUG Gradio UI (_get_created_document_path): Python Interpreter Observations: '''{observations}'''") 
                         
                         match = re.search(
                             r"(?:Document created \((?:docx|pdf|txt)\):|Document converted to PDF:)\s*(/tmp/[a-zA-Z0-9_]+/generated_document\.(?:docx|pdf|txt))",
@@ -200,49 +200,46 @@ class GradioUI:
                             normalized_path = os.path.normpath(extracted_path)
                             if os.path.exists(normalized_path):
                                 print(f"DEBUG Gradio UI: Validated path for download: {normalized_path}")
-                                return normalized_path # Return the path
+                                return normalized_path 
                             else:
                                 print(f"DEBUG Gradio UI: Path from create_document output ('{normalized_path}') does not exist.")
         print("DEBUG Gradio UI: No valid generated document path found in agent logs.")
-        return None # Return None if no valid path found
+        return None 
 
     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}]
         
-        # Initial yield: show user message, keep file_download_display_component hidden
-        yield updated_chat_history, gr.update(value=None, visible=False)
+        yield updated_chat_history, gr.update(value=None, visible=False) # For file_download_display_component
 
         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 progressively to update chat, keep file_download_display_component hidden
-            yield updated_chat_history + agent_responses_for_history, gr.update(value=None, visible=False) 
+            yield updated_chat_history + agent_responses_for_history, gr.update(value=None, visible=False) # For file_download_display_component
 
         final_chat_display_content = updated_chat_history + agent_responses_for_history
         
-        # 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, 
+            # CORRECTED: Use gr.update() for the File component
+            yield final_chat_display_content, gr.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)
+            # CORRECTED: Use gr.update() for the File component
+            yield final_chat_display_content, gr.update(value=None, visible=False)
 
 
-    def log_user_message(self, text_input_value: str): # File uploads removed
+    def log_user_message(self, text_input_value: str):
         full_prompt = text_input_value
         print(f"DEBUG Gradio: Prepared prompt for agent: {full_prompt[:300]}...") 
         return full_prompt, "" 
 
-    # prepare_and_show_download_file and download_action_button are removed as we try direct display
+    # prepare_and_show_download_file is not needed if we directly update the gr.File component
 
     def launch(self, **kwargs):
         with gr.Blocks(fill_height=True, theme=gr.themes.Soft(primary_hue=gr.themes.colors.blue)) as demo:
@@ -267,13 +264,12 @@ class GradioUI:
                     )
                 
                 with gr.Column(scale=1): 
-                    # --- "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
+                    # "Generated File" section directly shows the gr.File component
+                    gr.Markdown("### Generated Document") 
                     file_download_display_component = gr.File(
                         label="Downloadable Document", 
-                        visible=False, # Initially hidden
-                        interactive=False # User cannot upload to this
+                        visible=False, 
+                        interactive=False 
                     )
 
             text_message_input.submit(
@@ -283,11 +279,8 @@ class GradioUI:
             ).then(
                 self.interact_with_agent, 
                 [prepared_prompt_for_agent, chatbot_display], 
-                # Outputs: chatbot, and the file_download_display_component
-                [chatbot_display, file_download_display_component] 
+                [chatbot_display, file_download_display_component] # Outputs update chatbot and file component
             )
-
-            # No explicit download_action_button.click event needed anymore
             
         demo.launch(debug=True, share=kwargs.get("share", False), **kwargs)