Update app.py

app.py

@@ -86,20 +86,22 @@ def test_rag(qa, query):
 
 user_prompt = st.chat_input("Ask me anything about the content of the PDF(s):")
 print("user input:", user_prompt)
-if user_prompt:
-    print("user input after if condition: ", user_prompt)
-    st.session_state.messages.append({'role': 'user', "content": user_prompt})
-    books_retriever = test_rag(books_db_client_retriever, user_prompt)
-    print("books retriver:",books_retriever)
-    # Extracting the relevant answer using regex
-    corrected_text_match = re.search(r"Helpful Answer:(.*)", books_retriever, re.DOTALL)
-    print("corrected text match:", corrected_text_match)
-    if corrected_text_match:
-        corrected_text_books = corrected_text_match.group(1).strip()
-    else:
-        corrected_text_books = "No helpful answer found."
-    print("corrected text books: ",corrected_text_books)
-    st.session_state.messages.append({'role': 'assistant', "content": corrected_text_books})
+if st.button("Submit"):
+    print("user input after submit button: ", user_prompt)
+    if user_prompt:
+        print("user input after if user prompt condition: ", user_prompt)
+        st.session_state.messages.append({'role': 'user', "content": user_prompt})
+        books_retriever = test_rag(books_db_client_retriever, user_prompt)
+        print("books retriver:",books_retriever)
+        # Extracting the relevant answer using regex
+        corrected_text_match = re.search(r"Helpful Answer:(.*)", books_retriever, re.DOTALL)
+        print("corrected text match:", corrected_text_match)
+        if corrected_text_match:
+            corrected_text_books = corrected_text_match.group(1).strip()
+        else:
+            corrected_text_books = "No helpful answer found."
+        print("corrected text books: ",corrected_text_books)
+        st.session_state.messages.append({'role': 'assistant', "content": corrected_text_books})
 
 for message in st.session_state.messages:
     with st.chat_message(message['role']):