Update main.py

main.py

@@ -92,23 +92,21 @@ async def process_analysis_request(request: Request):
         questions_file_name, attached_file_names = None, []
         first_txt_file_name = None
         q_pattern = re.compile(r'question', re.IGNORECASE)
-
+        
         for filename in file_contents.keys():
             is_txt = filename.lower().endswith(".txt")
-            if is_txt and first_txt_file_name is None:
-                first_txt_file_name = filename
             if is_txt and q_pattern.search(filename):
-                if questions_file_name is None: questions_file_name = filename
-                else: attached_file_names.append(filename)
+                questions_file_name = filename
+            elif is_txt and first_txt_file_name is None:
+                first_txt_file_name = filename
             else:
                 attached_file_names.append(filename)
         
-        if questions_file_name is None:
+        if not questions_file_name:
             questions_file_name = first_txt_file_name
-            if questions_file_name in attached_file_names:
-                attached_file_names.remove(questions_file_name)
         
-        if not questions_file_name: raise HTTPException(status.HTTP_400_BAD_REQUEST, "No .txt question file found.")
+        if not questions_file_name:
+            raise HTTPException(status.HTTP_400_BAD_REQUEST, "No .txt question file found.")
         
         task_content = file_contents[questions_file_name].decode("utf-8")