Update app.py

app.py

@@ -39,6 +39,7 @@ class CompanyKnowledgeBase:
     
     def find_answer(self, query, threshold=0.8):
         if not self.dataset:
+            print('Context not found in the dataset.')
             return None
         
         try:
@@ -52,6 +53,7 @@ class CompanyKnowledgeBase:
             if D[0][0] < threshold:
                 # Find the matched question
                 matched_question = list(self.question_to_answer.keys())[I[0][0]]
+                print('Found matched:' + matched_question)
                 return self.question_to_answer[matched_question]
         
         except Exception as e: