Update Gradio_UI.py

Gradio_UI.py

@@ -134,7 +134,6 @@ def stream_to_gradio(
                 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}")
-            # MODIFIED: Yield tuple (filepath, alt_text)
             yield {"role": "assistant", "content": (image_path_for_gradio, "Generated Image")} 
             return
         except Exception as e:
@@ -151,7 +150,6 @@ def stream_to_gradio(
         image_path = final_answer_processed.to_string()
         print(f"DEBUG Gradio (stream_to_gradio): final_answer_processed is AgentImage. Path: {image_path}")
         if image_path and os.path.exists(image_path):
-             # MODIFIED: Yield tuple (filepath, alt_text)
              yield {"role": "assistant", "content": (image_path, "Generated Image (from AgentImage)")} 
         else:
             err_msg = f"Error: Image path from AgentImage ('{image_path}') not found or invalid."
@@ -161,7 +159,6 @@ def stream_to_gradio(
         audio_path = final_answer_processed.to_string()
         print(f"DEBUG Gradio (stream_to_gradio): AgentAudio path: {audio_path}")
         if audio_path and os.path.exists(audio_path):
-            # MODIFIED: Yield tuple (filepath, alt_text) for consistency, though Gradio might just use path for audio
             yield {"role": "assistant", "content": (audio_path, "Generated Audio")}
         else:
             err_msg = f"Error: Audio path from AgentAudio ('{audio_path}') not found"
@@ -182,40 +179,38 @@ class GradioUI:
                 os.makedirs(self.file_upload_folder, exist_ok=True)
         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: 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
-                    )
+    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
+        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 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
-    
+                    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.")
+        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
+
     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}")