Update app.py

app.py

@@ -6,6 +6,12 @@ from langchain.vectorstores import Chroma
 from langchain.chains import RetrievalQA
 from langchain.embeddings.sentence_transformer import SentenceTransformerEmbeddings
 
+def get_text():
+    input_text = st.text_input("You: ", key="input")
+    return input_text
+
+submit = st.button('Get Answer')
+user_input = get_text()
 
 
 loader = TextLoader('India.txt')
@@ -22,10 +28,13 @@ embeddings = SentenceTransformerEmbeddings(model_name="all-MiniLM-L6-v2")
 db = Chroma.from_documents(texts, embeddings)
 db._collection.get(include=['embeddings'])
 retriever = db.as_retriever(search_kwargs={"k": 1})
-docs = retriever.get_relevant_documents("What is the capital of india?")
-st.write("Answer")
-document = docs[0]
-page_content = document.page_content
 
-st.write(page_content)
+if user_input and submit:
+    
+    docs = retriever.get_relevant_documents(user_input)
+    st.write("Answer")
+    document = docs[0]
+    page_content = document.page_content
+
+    st.write(page_content)
 # st.text(file_content)