Update app.py

app.py

@@ -71,19 +71,22 @@ class Chatbot:
 
     def run_google_search(self, query):
         search = GoogleSerperAPIWrapper()
-        search.run(query)
+        return 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)
+    retrievers = chatbot.load_embeddings()
+    qa = chatbot.create_qa_model(retrievers)
 
     st.title('Chatbot Trained on Indian Exam Articles')
     st.header("Hi!! How Can I Help You ?")
 
-    query = st.text_input('> ')
+    query = st.text_input('ENTER TEXT HERE ')
     result = qa({'query': query})
     st.write(result['result'])
-    st.button('Not Satisfied! Talk to our Expert Here..')
+    if st.button('Not Satisfied! Talk to our Expert Here..'):
+        st.write(run_google_search(query))
+