Update app.py

app.py

@@ -23,26 +23,32 @@ class ChatBot:
             json.dump(self.knowledge_base, f, indent=4)
 
     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:")
+            # Display a text input to teach the bot
             response = st.text_input(f"Teach me a response for '{user_input}':", key="teach_input")
+            # When the user provides a response and submits it, teach the bot
             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
                 else:
                     st.warning("Please provide a response before submitting.")
         return response
 
     def find_response(self, user_input):
+        # Search for the user input 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
         new_question = {'question': user_input.lower(), 'response': response}
         if 'questions' not in self.knowledge_base:
             self.knowledge_base['questions'] = []