Update Gradio_UI.py

Gradio_UI.py

@@ -123,16 +123,14 @@ def stream_to_gradio(
     final_answer_content = all_step_logs[-1] 
 
     actual_content_for_handling = final_answer_content
-    # Check if final_answer_content is a wrapper like FinalAnswerStep and extract the core content
-    if hasattr(final_answer_content, 'final_answer') and not isinstance(final_answer_content, (str, PILImage.Image, tuple)): # Added tuple to avoid unwrapping already formatted image content
+    if hasattr(final_answer_content, 'final_answer') and not isinstance(final_answer_content, (str, PILImage.Image, tuple)):
          actual_content_for_handling = final_answer_content.final_answer
          print(f"DEBUG Gradio: Extracted actual_content_for_handling from FinalAnswerStep: {type(actual_content_for_handling)}")
 
-    # Priority 1: Handle raw PIL Image object for direct display
     if isinstance(actual_content_for_handling, PILImage.Image):
         print("DEBUG Gradio (stream_to_gradio): Actual content IS a raw PIL Image.")
         try:
-            with tempfile.NamedTemporaryFile(delete=False, suffix=".png") as tmp_file: # delete=False is important
+            with tempfile.NamedTemporaryFile(delete=False, suffix=".png") as tmp_file: 
                 actual_content_for_handling.save(tmp_file, format="PNG")
                 image_path_for_gradio = tmp_file.name
             print(f"DEBUG Gradio: Saved PIL image to temp path: {image_path_for_gradio}")
@@ -143,7 +141,6 @@ def stream_to_gradio(
             yield {"role": "assistant", "content": f"**Final Answer (Error displaying image):** {e}"}
             return
 
-    # Priority 2: Use smolagents' type handling if not a raw PIL image
     final_answer_processed = handle_agent_output_types(actual_content_for_handling)
     print(f"DEBUG Gradio: final_answer_processed type after handle_agent_output_types: {type(final_answer_processed)}")
 
@@ -168,20 +165,15 @@ def stream_to_gradio(
             print(f"DEBUG Gradio: {err_msg}")
             yield {"role": "assistant", "content": f"**Final Answer ({err_msg})**"}
     else: 
-        # This will display the string representation of the object if not specifically handled
         yield {"role": "assistant", "content": f"**Final Answer:**\n{str(final_answer_processed)}"}
 
 
 class GradioUI:
-    def __init__(self, agent: MultiStepAgent, file_upload_folder: str | None = None): # file_upload_folder kept for potential future use
+    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 = file_upload_folder # Commented out as per request to simplify
-        # if self.file_upload_folder is not None:
-        #     if not os.path.exists(self.file_upload_folder):
-        #         os.makedirs(self.file_upload_folder, exist_ok=True)
-        self.file_upload_folder = None # Explicitly disable for now
+        self.file_upload_folder = None # File upload UI is disabled
         self._latest_file_path_for_download = None
 
     def _check_for_created_file(self):
@@ -191,26 +183,22 @@ class GradioUI:
             for log_entry in reversed(self.agent.interaction_logs): 
                 if isinstance(log_entry, ActionStep):
                     observations = getattr(log_entry, 'observations', None)
-                    tool_calls = getattr(log_entry, 'tool_calls', []) # Get tool calls if any
+                    tool_calls = getattr(log_entry, 'tool_calls', [])
                     
-                    # Check if this step involved python_interpreter
                     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 (_check_for_file): Python Interpreter Observations: '''{observations[:500]}...'''") # Log snippet
+                        # CRITICAL DEBUG LINE: Print the exact observations string
+                        print(f"DEBUG Gradio UI (_check_for_file): Python Interpreter Observations: '''{observations}'''") 
                         
-                        # Regex to find paths specifically printed by our create_document tool
-                        # This pattern expects: "Document created (docx): /tmp/random/generated_document.docx"
                         match = re.search(
-                            # Capture group 1 is the prefix, group 2 is the path
-                            r"(Document created \((?:docx|pdf|txt)\):|Document converted to PDF:)\s*(/tmp/[a-zA-Z0-9_]+/generated_document\.(?:docx|pdf|txt))",
+                            r"(?:Document created \((?:docx|pdf|txt)\):|Document converted to PDF:)\s*(/tmp/[a-zA-Z0-9_]+/generated_document\.(?:docx|pdf|txt))",
                             observations,
                             re.MULTILINE 
                         )
                         
                         if match:
-                            # extracted_prefix = match.group(1) # e.g., "Document created (docx):"
-                            extracted_path = match.group(2) # The actual path
+                            extracted_path = match.group(1) # Path is group 1 due to simplified regex
                             print(f"DEBUG Gradio UI: Regex matched. Extracted path: '{extracted_path}'")
                             normalized_path = os.path.normpath(extracted_path)
                             if os.path.exists(normalized_path):
@@ -229,31 +217,21 @@ class GradioUI:
 
         updated_chat_history = current_chat_history + [{"role": "user", "content": prompt_text}]
         
-        # Initial yield: show user message, hide download components until agent run is complete
         yield updated_chat_history, gr.update(visible=False), 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 progressively to update chat, keep download components hidden during streaming
             yield updated_chat_history + agent_responses_for_history, gr.update(visible=False), gr.update(value=None, visible=False) 
 
-        # After all agent messages are processed and added to history
-        final_chat_display_content = updated_chat_history + agent_responses_for_history
-        
-        # Now check for created files to decide visibility of download button
         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}")
-        # Final yield: update chat, set visibility of download button, keep file display component hidden (it's shown on button click)
         yield final_chat_display_content, gr.update(visible=file_found_for_download), gr.update(value=None, visible=False)
 
-
-    def log_user_message(self, text_input_value: str): # Removed current_file_uploads as upload is disabled
+    def log_user_message(self, text_input_value: str):
         full_prompt = text_input_value
-        # if current_file_uploads: # This part is now disabled
-        #     files_str = ", ".join([os.path.basename(f) for f in current_file_uploads])
-        #     full_prompt += f"\n\n[Uploaded files for context: {files_str}]"
         print(f"DEBUG Gradio: Prepared prompt for agent: {full_prompt[:300]}...") 
         return full_prompt, "" 
 
@@ -269,13 +247,12 @@ class GradioUI:
 
     def launch(self, **kwargs):
         with gr.Blocks(fill_height=True, theme=gr.themes.Soft(primary_hue=gr.themes.colors.blue)) as demo:
-            # file_uploads_log_state = gr.State([]) # No longer needed as upload is disabled
             prepared_prompt_for_agent = gr.State("")
 
             gr.Markdown("## Smol Talk with your Agent") 
 
             with gr.Row(equal_height=False): 
-                with gr.Column(scale=3): # Main chat column
+                with gr.Column(scale=3): 
                     chatbot_display = gr.Chatbot(
                         type="messages", 
                         avatar_images=(None, "https://huggingface.co/datasets/huggingface/brand-assets/resolve/main/hf-logo-round.png"),
@@ -290,26 +267,19 @@ class GradioUI:
                         show_label=False 
                     )
                 
-                with gr.Column(scale=1): # Sidebar column
-                    # --- File Upload Section Removed/Commented Out ---
+                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): 
-                    #         file_uploader = gr.File(label="Upload a supporting file (PDF, DOCX, TXT, JPG, PNG)")
-                    #         upload_status_text = gr.Textbox(label="Upload Status", interactive=False, lines=1) 
-                    #         file_uploader.upload( 
-                    #             self.upload_file,
-                    #             [file_uploader, file_uploads_log_state],
-                    #             [upload_status_text, file_uploads_log_state],
-                    #         )
+                    #         # ... (upload components would be here)
                     
-                    with gr.Accordion("Generated File", open=True): # Keep this section
+                    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) 
 
-            # Event Handling Chain for Text Submission
             text_message_input.submit(
                 self.log_user_message, 
-                [text_message_input], # Removed file_uploads_log_state from inputs
+                [text_message_input], # Only text_input as file uploads are disabled
                 [prepared_prompt_for_agent, text_message_input] 
             ).then(
                 self.interact_with_agent,