Update Gradio_UI.py

Gradio_UI.py

@@ -183,27 +183,39 @@ class GradioUI:
         self._latest_file_path_for_download = None
 
     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): 
-                if isinstance(log_entry, ActionStep):
-                    observations = getattr(log_entry, 'observations', None)
-                    if observations and isinstance(observations, str):
-                        print(f"DEBUG Gradio UI: Checking observations: {observations[:200]}") 
-                        path_match = re.search(r"(/tmp/[a-zA-Z0-9_]+/generated_document\.(?:docx|pdf|txt))", observations)
-                        if path_match:
-                            extracted_path = 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 observations): {self._latest_file_path_for_download}")
-                                return True 
-                            else:
-                                print(f"DEBUG Gradio UI: Path from observations ('{normalized_path}') does not exist.")
-        print("DEBUG Gradio UI: No valid generated file path found in agent logs for download.")
-        return False
-
+    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): 
+            if isinstance(log_entry, ActionStep):
+                observations = getattr(log_entry, 'observations', None)
+                
+                if observations and isinstance(observations, str):
+                    print(f"DEBUG Gradio UI: Log Entry Observations (for file check): {observations[:300]}") # Log more
+                    
+                    # Regex to find paths specifically printed by our create_document tool
+                    # It prints "Document created (docx): /path/to/file"
+                    # or "Document created (txt): /path/to/file"
+                    # or "Document converted to PDF: /path/to/file"
+                    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) # Group 1 is the 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 (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 no match for create_document output specifically
+                        # print(f"DEBUG Gradio UI: No 'create_document' output pattern found in these observations.")
+    print("DEBUG Gradio UI: No valid generated file path (from create_document) found in agent logs for download.")
+    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}")