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, question_data: dict) -> Tuple[List[str], List[str], List[str]]:
+    def _detect_and_process_direct_attachments(self, attachment_name) -> 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)
@@ -484,10 +484,7 @@ class IntelligentAgent:
         # Check for attachments in various fields
         attachments = []
         
-        # Common fields where attachments might be found
-        attachment_fields = ['file_name']
-        
-        for field in attachment_fields:
+        for field in [attachment_name]:
             if field in question_data:
                 field_data = question_data[field]
                 if isinstance(field_data, list):
@@ -810,6 +807,7 @@ Answer:"""
         try:
             # Detect and process URLs
             print(f"2. Detecting and processing URLs...")
+            
             url_context = self._extract_and_process_urls(question_text)
             
             if self.debug:
@@ -820,7 +818,8 @@ Answer:"""
         try:    
             # Detect and download attachments
             print(f"3. Searching for images, audio or code attachments...")
-            image_files, audio_files, code_files = self._detect_and_process_direct_attachments(question_data)
+            attachment_name = question_data.get('file_name', '')
+            image_files, audio_files, code_files = self._detect_and_process_direct_attachments(attachment_name)
             
             # Process attachments to get context
             attachment_context = self._process_attachments(image_files, audio_files, code_files)