Update query.py

query.py

@@ -79,15 +79,15 @@ class VectaraQuery():
                 response_set = res['responseSet']
                 
                 if response_set:
-                    for result in response_set:
+                    for result in response_set['response']:
                         if 'text' not in result:
                             continue
                         text = result['text']
                         print(f"Processing text: {text}")  # Debugging line
 
-                        reason_match = re.search(r"Reason Why it Can't be Used: (.*?)\n", text, re.DOTALL)
-                        alternative_match = re.search(r"Alternative: (.*?)\n", text, re.DOTALL)
-                        notes_match = re.search(r"Notes: (.*?)\n", text, re.DOTALL)
+                        reason_match = re.search(r"Reason Why it Can't be Used:\s*(.*?)(?:\n|$)", text, re.DOTALL)
+                        alternative_match = re.search(r"Alternative:\s*(.*?)(?:\n|$)", text, re.DOTALL)
+                        notes_match = re.search(r"Notes:\s*(.*?)(?:\n|$)", text, re.DOTALL)
                         
                         reason = reason_match.group(1).strip() if reason_match else "Not available"
                         alternative = alternative_match.group(1).strip() if alternative_match else "Not available"