post processing

gemini_agent.py

@@ -272,12 +272,17 @@ Provide only the direct answer. If it's a quote, give just the quoted text. If i
             if move_match:
                 answer = move_match.group(1)
 
-        # Post-processing for strict list extraction
-        if any(kw in question.lower() for kw in ["alphabetize", "comma separated", "list", "ingredients", "page numbers", "vegetables"]):
-            # Extract only a comma-separated list of words (allowing spaces)
-            list_match = re.findall(r'[A-Za-z][A-Za-z ]*', answer)
-            if list_match:
-                answer = ', '.join([item.strip() for item in list_match if item.strip()])
+        # Post-processing for sorted, deduplicated lists
+        if 'page numbers' in question.lower() or 'comma-delimited list' in question.lower():
+            # Extract numbers, deduplicate, sort, and join
+            nums = re.findall(r'\d+', answer)
+            nums = sorted(set(int(n) for n in nums))
+            answer = ', '.join(str(n) for n in nums)
+        elif 'alphabetize' in question.lower() or 'alphabetized' in question.lower() or 'ingredients' in question.lower() or 'vegetables' in question.lower():
+            # Extract words/phrases, deduplicate, sort, and join
+            items = [item.strip() for item in answer.split(',') if item.strip()]
+            items = sorted(set(items), key=lambda x: x.lower())
+            answer = ', '.join(items)
 
         return answer