Update app.py

app.py

@@ -469,64 +469,65 @@ class IntelligentAgent:
         
         return "\n\n" + "="*50 + "\n".join(formatted_content) + "\n" + "="*50
 
-    def _detect_and_process_direct_attachments(self, attachment_name: str) -> Tuple[List[str], List[str], List[str]]:
+    def _detect_and_process_direct_attachment(self, attachment_name: str) -> Tuple[List[str], List[str], List[str]]:
         """
-        Detect and process attachments that are directly attached to questions (not as URLs).
+        Detect and process a single attachment directly attached to a question (not as a URL).
         Returns (image_files, audio_files, code_files)
         """
         image_files = []
         audio_files = []
         code_files = []
-        
+
         # Create temporary directory for attachments
         temp_dir = tempfile.mkdtemp(prefix="agent_attachments_")
-   
-        # Process attachment
+
         try:
+        # Here, file_type should ideally come from metadata or inferred from content — 
+        # since only attachment_name is passed, we'll rely on the file extension.
             file_type = ''
-                
+
             # Save attachment to file
-            file_path = save_attachment_to_file(file_name, temp_dir, file_name)
-                
+            file_path = save_attachment_to_file(attachment_name, temp_dir, attachment_name)
             if not file_path:
-                continue
-                
-            # Categorize based on extension or type
+                if getattr(self, 'debug', False):
+                   print(f"Failed to save attachment: {attachment_name}")
+                return image_files, audio_files, code_files
+
+            # Get file extension
             file_ext = Path(file_path).suffix.lower()
-                
+
             # Determine category
             is_image = (
-                    'image' in file_type or 
-                    file_ext in ['.jpg', '.jpeg', '.png', '.gif', '.bmp', '.webp', '.tiff']
-                )
+            'image' in file_type or 
+            file_ext in ['.jpg', '.jpeg', '.png', '.gif', '.bmp', '.webp', '.tiff']
+            )
             is_audio = (
-                    'audio' in file_type or 
-                    file_ext in ['.mp3', '.wav', '.m4a', '.ogg', '.flac', '.aac']
-                )
+            'audio' in file_type or 
+            file_ext in ['.mp3', '.wav', '.m4a', '.ogg', '.flac', '.aac']
+            )
             is_code = (
-                    'python' in file_type or 'code' in file_type or 'text' in file_type or
-                    file_ext in ['.py', '.txt', '.js', '.html', '.css', '.json', '.xml']
-                )
-                
+            'python' in file_type or 'code' in file_type or 'text' in file_type or
+            file_ext in ['.py', '.txt', '.js', '.html', '.css', '.json', '.xml']
+            )
+
             # Categorize the file
             if is_image:
-                    image_files.append(file_path)
+                image_files.append(file_path)
             elif is_audio:
-                    audio_files.append(file_path)
+                audio_files.append(file_path)
             elif is_code:
-                    code_files.append(file_path)
+                code_files.append(file_path)
             else:
-                    # Default to code/text for unknown types
-                    code_files.append(file_path)
-                    
+                # Default to code/text for unknown types
+                code_files.append(file_path)
+
         except Exception as e:
-            if self.debug:
-                print(f"Error processing attachment {i}: {e}")
-            continue
-        
-        if self.debug:
-            print(f"...Found direct attachments: {len(image_files)} images, {len(audio_files)} audio, {len(code_files)} code files")
-        
+            if getattr(self, 'debug', False):
+                print(f"Error processing attachment {attachment_name}: {e}")
+
+        if getattr(self, 'debug', False):
+            print(f"...Processed attachment: {len(image_files)} images, {len(audio_files)} audio, {len(code_files)} code files")
+                
         return image_files, audio_files, code_files
 
     def _process_attachments(self, image_files: List[str] = None, audio_files: List[str] = None, code_files: List[str] = None) -> str: