Update app.py

app.py

@@ -469,7 +469,7 @@ class IntelligentAgent:
         
         return "\n\n" + "="*50 + "\n".join(formatted_content) + "\n" + "="*50
 
-    def _detect_and_process_direct_attachments(self, attachment_name) -> Tuple[List[str], List[str], List[str]]:
+    def _detect_and_process_direct_attachments(self, attachment_name: str) -> Tuple[List[str], List[str], List[str]]:
         """
         Detect and process attachments that are directly attached to questions (not as URLs).
         Returns (image_files, audio_files, code_files)
@@ -480,43 +480,22 @@ class IntelligentAgent:
         
         # Create temporary directory for attachments
         temp_dir = tempfile.mkdtemp(prefix="agent_attachments_")
-        
-        # Check for attachments in various fields
-        attachments = []
-        
-        for field in [attachment_name]:
-            if field in question_data:
-                field_data = question_data[field]
-                if isinstance(field_data, list):
-                    attachments.extend(field_data)
-                elif field_data:  # Non-empty data
-                    attachments.append(field_data)
-        
-        # Process each attachment
-        for i, attachment in enumerate(attachments):
-            try:
-                # Determine file name
-                file_name = None
-                file_type = ''
-                
-                if isinstance(attachment, dict):
-                    # Extract metadata if available
-                    file_name = attachment.get('name') or attachment.get('filename') or f"attachment_{i}"
-                    file_type = attachment.get('type', '').lower() or attachment.get('mime_type', '').lower()
-                else:
-                    file_name = f"attachment_{i}"
+   
+        # Process attachment
+        try:
+            file_type = ''
                 
-                # Save attachment to file
-                file_path = save_attachment_to_file(attachment, temp_dir, file_name)
+            # Save attachment to file
+            file_path = save_attachment_to_file(file_name, temp_dir, file_name)
                 
-                if not file_path:
-                    continue
+            if not file_path:
+                continue
                 
-                # Categorize based on extension or type
-                file_ext = Path(file_path).suffix.lower()
+            # Categorize based on extension or type
+            file_ext = Path(file_path).suffix.lower()
                 
-                # Determine category
-                is_image = (
+            # Determine category
+            is_image = (
                     'image' in file_type or 
                     file_ext in ['.jpg', '.jpeg', '.png', '.gif', '.bmp', '.webp', '.tiff']
                 )