Update app.py

app.py

@@ -257,30 +257,32 @@ document_chain = create_stuff_documents_chain(llm, prompt)
 retriever = st.session_state.vector.as_retriever()
 retrieval_chain = create_retrieval_chain(retriever, document_chain)
 
-prompt = st.text_input("Input your prompt here")
-
-
-# If the user hits enter
-if prompt:
-    # Then pass the prompt to the LLM
-    start = time.process_time()
-    response = retrieval_chain.invoke({"input": prompt})
-    # print(f"Response time: {time.process_time() - start}")
-    st.write(f"Response time: {time.process_time() - start} seconds")
-
-    st.write(response["answer"])
-
-    # With a streamlit expander
-    with st.expander("Document Similarity Search"):
-        # Find the relevant chunks
-        for i, doc in enumerate(response["context"]):
-            # print(doc)
-            # st.write(f"Source Document # {i+1} : {doc.metadata['source'].split('/')[-1]}")
-            st.write(doc)
-            st.write(f"Source Document # {i+1} : {doc.metadata['source'].split('/')[-1]}")
+while True:
+    
+    prompt = st.text_input("Input your prompt here")
+
+
+    # If the user hits enter
+    if prompt:
+        # Then pass the prompt to the LLM
+        start = time.process_time()
+        response = retrieval_chain.invoke({"input": prompt})
+        # print(f"Response time: {time.process_time() - start}")
+        st.write(f"Response time: {time.process_time() - start} seconds")
+
+        st.write(response["answer"])
+
+        # With a streamlit expander
+        with st.expander("Document Similarity Search"):
+            # Find the relevant chunks
+            for i, doc in enumerate(response["context"]):
+                # print(doc)
+                # st.write(f"Source Document # {i+1} : {doc.metadata['source'].split('/')[-1]}")
+                st.write(doc)
+                st.write(f"Source Document # {i+1} : {doc.metadata['source'].split('/')[-1]}")
             
             
-            st.write(doc.page_content)
-            st.write("--------------------------------")
+                st.write(doc.page_content)
+                st.write("--------------------------------")
 
-st.write("---------------------------------")
+    st.write("---------------------------------")