Update app.py

app.py

@@ -25,23 +25,6 @@ class ChatBot:
             json.dump(self.knowledge_base, f, indent=4)
         st.write(f"Knowledge base saved: {self.knowledge_base}")
 
-    def learn_and_response(self, user_input):
-        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")
-            if st.button("Submit Response"):
-                if response:
-                    self.teach_response(user_input, response)
-                    st.success(f"Response saved for '{user_input}': {response}")
-                    return response
-                else:
-                    st.warning("Please provide a response before submitting.")
-        else:
-            st.write(f"Found response for '{user_input}': {response}")
-        return response
-
     def find_response(self, user_input):
         """Find a response in the knowledge base."""
         for question in self.knowledge_base.get("questions", []):
@@ -49,11 +32,24 @@ class ChatBot:
                 return question['response']
         return None
 
+    def learn_and_response(self, user_input):
+        response = self.find_response(user_input)
+        if response:
+            st.write(f"Bot: {response}")
+        else:
+            st.write(f"I don't have a response for '{user_input}'. Please teach me:")
+            response_input = st.text_input(f"Teach me a response for '{user_input}':", key="teach_input")
+            if st.button("Submit Response"):
+                if response_input:
+                    self.teach_response(user_input, response_input)
+                    st.success(f"Response saved for '{user_input}': {response_input}")
+                else:
+                    st.warning("Please provide a response before submitting.")
+
     def teach_response(self, user_input, response):
         """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__":
@@ -64,8 +60,6 @@ if __name__ == "__main__":
     user_input = st.text_input("Enter your query below:", key="user_input")
     if st.button("Submit Query"):
         if user_input:
-            response = chatbot.learn_and_response(user_input)
-            if response:
-                st.write(f"Bot: {response}")
+            chatbot.learn_and_response(user_input)
         else:
             st.warning("Please enter a query before submitting.")