updated the chatbot.py ("Can you elaborate the question")

chatbot.py

@@ -35,10 +35,6 @@ def process_query(query):
         context = [m['metadata']['Output'] for m in xc['matches']]
         context_str = " ".join(context)
         formatted_query = f"answer the question: {query} context: {context_str}"
-    else:
-        # Handle the case where 'matches' isn't found or isn't in the expected format
-        context_str = ""
-        formatted_query = f"answer the question: {query} context: {context_str}"
 
     # Generate answer using T5 model
     output_text = context_str
@@ -52,4 +48,10 @@ def process_query(query):
     ids = generator.generate(inputs, num_beams=2, min_length=10, max_length=60, repetition_penalty=1.2)
     answer = tokenizer.decode(ids[0], skip_special_tokens=True, clean_up_tokenization_spaces=False)
 
+    nli_keywords = ['not_equivalent', 'not_entailment', 'entailment', 'neutral']
+    
+    # If any of the keywords are found in the answer, return the fallback message
+    if any(keyword in answer.lower() for keyword in nli_keywords):
+        return "Klasmeyt, can you elaborate your question?"
+
     return answer