Update query.py

query.py

@@ -85,15 +85,17 @@ class VectaraQuery():
                         text = result['text']
                         print(f"Processing text: {text}")  # Debugging line
 
-                        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)
+                        # Adjusting regex patterns to be more flexible
+                        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"
                         notes = notes_match.group(1).strip() if notes_match else "Not available"
                         
                         response = f"Reason: {reason}\nAlternative: {alternative}\nNotes: {notes}"
+                        print(f"Generated response: {response}")  # Debugging line
                         return response
 
         return "No relevant information found."