Update app.py

app.py

@@ -97,10 +97,7 @@ def refine_answer(question, retrieved_answer):
     system_content = "You are an assistant that refines answers to make them more relevant and natural."
     return llama_query(prompt, system_content)
 
-def get_answer(user_question, threshold=0.01):
-    if not check_blood_donation_relevance(user_question):
-        return "I'm sorry, but your question doesn't seem to be related to blood donation. Could you please ask a question about blood donation?", 0
-
+def get_answer(user_question, threshold=0.3):
     language = detect_language(user_question)
 
     if language == 'swahili':
@@ -119,10 +116,20 @@ def get_answer(user_question, threshold=0.01):
 
         return refined_answer, final_score
     else:
-        default_message = "The system couldn't find a sufficient answer to your question. Do you want to learn anything else about blood donation?"
-        if language == 'swahili':
-            default_message = translate_to_swahili(default_message)
-        return default_message, final_score
+        # Use LLaMa to determine if the question is related to blood donation
+        if check_blood_donation_relevance(english_question):
+            llama_response = llama_query(
+                f"Please provide a brief, general answer to this blood donation related question: {english_question}",
+                "You are an assistant knowledgeable about blood donation. Provide brief, accurate answers."
+            )
+            if language == 'swahili':
+                llama_response = translate_to_swahili(llama_response)
+            return llama_response, 0
+        else:
+            off_topic_message = "I'm sorry, but your question doesn't seem to be related to blood donation. Could you please ask a question about blood donation?"
+            if language == 'swahili':
+                off_topic_message = translate_to_swahili(off_topic_message)
+            return off_topic_message, 0
 
 # Gradio app
 def gradio_app(user_question):