Update app.py

app.py

@@ -26,7 +26,6 @@ class ChatBot:
         st.write(f"Knowledge base saved: {self.knowledge_base}")
 
     def learn_and_response(self, user_input):
-        # Try to find a response
         response = self.find_response(user_input)
         if response is None:
             st.write(f"I don't have a response for '{user_input}'. Please teach me:")
@@ -35,9 +34,8 @@ class ChatBot:
             if st.button("Submit Response"):
                 if response:
                     self.teach_response(user_input, response)
-                    self.save_knowledge_base()  # Save after teaching
-                    st.success("Response saved!")
-                    return response  # Return the learned response immediately
+                    st.success(f"Response saved for '{user_input}': {response}")
+                    return response
                 else:
                     st.warning("Please provide a response before submitting.")
         else:
@@ -45,17 +43,18 @@ class ChatBot:
         return response
 
     def find_response(self, user_input):
-        # Search for the user input in the knowledge base
+        """Find a response in the knowledge base."""
         for question in self.knowledge_base.get("questions", []):
             if question['question'].lower() == user_input.lower():
                 return question['response']
         return None
 
     def teach_response(self, user_input, response):
-        # Add the new question-response pair to the knowledge base
+        """Teach the bot a new response."""
         new_question = {'question': user_input.lower(), 'response': response}
         self.knowledge_base['questions'].append(new_question)
         st.write(f"New question added: {new_question}")
+        self.save_knowledge_base()  # Save the updated knowledge base
 
 if __name__ == "__main__":
     # Initialize the chatbot with the JSON file