Update Gradio_UI.py

Gradio_UI.py

@@ -179,8 +179,8 @@ class GradioUI:
                 os.makedirs(self.file_upload_folder, exist_ok=True)
         self._latest_file_path_for_download = None
 
-    def _check_for_created_file(self): # Line 185 (ensure correct indentation below this line)
-        self._latest_file_path_for_download = None # Line 186 - START OF INDENTED BLOCK
+    def _check_for_created_file(self):
+        self._latest_file_path_for_download = 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.")
             for log_entry in reversed(self.agent.interaction_logs): 
@@ -188,32 +188,43 @@ class GradioUI:
                     observations = getattr(log_entry, 'observations', None)
                     
                     if observations and isinstance(observations, str):
-                        # This print statement is crucial for debugging this method
-                        print(f"DEBUG Gradio UI (check_for_file): Log Entry Observations: '{observations[:300]}'") 
-                        
-                        # Refined regex to look for the specific output of your create_document tool
-                        match = re.search(
-                            r"(?:Document created \((?:docx|pdf|txt)\):|Document converted to PDF:)\s*(/tmp/[a-zA-Z0-9_]+/generated_document\.(?:docx|pdf|txt))",
-                            observations
-                        )
-                        
-                        if match:
-                            extracted_path = match.group(1) 
-                            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 (from create_document output): {self._latest_file_path_for_download}")
-                                return True 
-                            else:
-                                print(f"DEBUG Gradio UI: Path from create_document output ('{normalized_path}') does not exist.")
-                        # else: # Optional: log if the specific pattern isn't found in this observation
-                            # print(f"DEBUG Gradio UI: 'create_document' output pattern not found in this observation.")
+                        # Log the full observations for the relevant step to help debug the regex
+                        # We are looking for the output of python_interpreter which contains create_document's print
+                        tool_calls = getattr(log_entry, 'tool_calls', [])
+                        is_python_interpreter_step = any(tc.name == "python_interpreter" for tc in tool_calls)
+
+                        if is_python_interpreter_step: # Only parse observations from python_interpreter
+                            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 should match:
+                            # "Document created (docx): /tmp/random/generated_document.docx"
+                            # "Document created (txt): /tmp/random/generated_document.txt"
+                            # "Document converted to PDF: /tmp/random/generated_document.pdf"
+                            # It allows for optional "Execution logs:" prefix and surrounding newlines/whitespace.
+                            match = re.search(
+                                r"(?:Document created \((?:docx|pdf|txt)\):|Document converted to PDF:)\s*(/tmp/[a-zA-Z0-9_]+/generated_document\.(?:docx|pdf|txt))",
+                                observations,
+                                re.MULTILINE # Important if observations string has multiple lines
+                            )
+                            
+                            if match:
+                                extracted_path = match.group(1) 
+                                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 # File found and path set
+                                else:
+                                    print(f"DEBUG Gradio UI: Path from create_document output ('{normalized_path}') does not exist.")
+                            # else: # Optional: log if the specific pattern isn't found in this observation
+                                # 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 in agent logs for download.")
-        return False # END OF INDENTED BLOCK for _check_for_created_file
+        return False
 
     def interact_with_agent(self, prompt_text: str, current_chat_history: list):
         print(f"DEBUG Gradio: interact_with_agent called with prompt: '{prompt_text}'")
-        print(f"DEBUG Gradio: Current chat history (input type {type(current_chat_history)}): {current_chat_history}")
+        # print(f"DEBUG Gradio: Current chat history (input type {type(current_chat_history)}): {current_chat_history}")
 
         updated_chat_history = current_chat_history + [{"role": "user", "content": prompt_text}]
         
@@ -224,11 +235,11 @@ class GradioUI:
             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) 
 
-        file_found = self._check_for_created_file()
+        file_found = self._check_for_created_file() # This call is crucial
         
         final_chat_display = updated_chat_history + agent_responses_for_history
-        print(f"DEBUG Gradio: Final chat history for display: {len(final_chat_display)} messages.")
-        yield final_chat_display, gr.update(visible=file_found), gr.update(value=None, visible=False)
+        print(f"DEBUG Gradio: Final chat history for display: {len(final_chat_display)} messages. File found for download: {file_found}")
+        yield final_chat_display, gr.update(visible=file_found), gr.update(value=None, visible=False) # Update download button visibility
 
     def upload_file(self, file, file_uploads_log_state):
         if file is None: