Update app.py

app.py

@@ -55,22 +55,22 @@ def vector_embedding():
     else:
         st.write("Vectors already initialized.")
 
-prompt1 = st.text_input("Enter Your Question From Documents")
-
-# Inside the button click for document embedding
-if st.button("Documents Embedding"):
-    try:
-        vector_embedding()
-    except Exception as e:
-        st.error(f"Error initializing vector store: {e}")
-
-# Inside the button click for processing user question
 if prompt1:
     if "vectors" not in st.session_state:
         st.error("Vectors are not initialized. Please click 'Documents Embedding' first.")
     else:
+        document_chain = create_stuff_documents_chain(llm, prompt)
+        retriever = st.session_state.vectors.as_retriever()
+        retrieval_chain = create_retrieval_chain(retriever, document_chain)
         try:
-            # Your try block code goes here
-            pass
+            start = time.process_time()
+            response = retrieval_chain.invoke({'input': prompt1})
+            st.write("Response time: ", time.process_time() - start)
+            st.write(response['answer'])
+
+            with st.expander("Document Similarity Search"):
+                for i, doc in enumerate(response["context"]):
+                    st.write(doc.page_content)
+                    st.write("--------------------------------")
         except Exception as e:
-            st.error(f"Error processing your question: {e}")
+            st.error(f"Failed to retrieve the answer: {e}")