Update app.py

app.py

@@ -65,26 +65,20 @@ class Chatbot:
         st.header("Hi!! How Can I Help You ?")
 
         query = st.text_input('> ')
+        if st.button('Search'):
+            self.run_google_search(query)
         result = self.qa({'query': query})
         st.write(result['result'])
-        st.button('Not Satisfied! Talk to our Expert Here..')
 
     def run_google_search(self, query):
         search = GoogleSerperAPIWrapper()
-        search.run
+        search.run(query)
+
 
 if __name__ == "__main__":
     chatbot = Chatbot()
     data = chatbot.load_data()
     docs = chatbot.split_documents(data)
     retriever = chatbot.create_embeddings(docs)
-    qa = chatbot.create_qa_model(retriever)
-
-    st.title('Chatbot Trained on Indian Exam Articles')
-    st.header("Hi!! How Can I Help You ?")
-
-    query = st.text_input('> ')
-    result = qa({'query': query})
-    st.write(result['result'])
-    st.button('Not Satisfied! Talk to our Expert Here..')
-
+    chatbot.qa = chatbot.create_qa_model(retriever)
+    chatbot.run_chatbot()