Update chatbot.py

chatbot.py

@@ -1,9 +1,17 @@
-from knowledge_base import get_answer_from_knowledge_base
+from knowledge_base import get_answer_from_knowledge_base, load_qa_model
 
 def generate_answer(question):
-    # Here we are getting the answer from a knowledge base.
+    # Check if the question is in the custom answers dictionary
+    custom_answers = {
+        "Who is the chief minister of AP?": "Y. S. Jagan Mohan Reddy",
+        # Add other custom answers if necessary
+    }
+
+    if question in custom_answers:
+        return custom_answers[question]
+
+    # If not, get the answer from the knowledge base using Hugging Face model
     answer = get_answer_from_knowledge_base(question)
-    
     if answer:
         return answer
     else: