sanjeevbora commited on
Commit
801253f
·
verified ·
1 Parent(s): 38d70b5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -85,16 +85,19 @@ def test_rag(qa, query):
85
  return qa.run(query)
86
 
87
  user_prompt = st.chat_input("Ask me anything about the content of the PDF(s):")
 
88
  if user_prompt:
89
  st.session_state.messages.append({'role': 'user', "content": user_prompt})
90
  books_retriever = test_rag(books_db_client_retriever, user_prompt)
 
91
  # Extracting the relevant answer using regex
92
  corrected_text_match = re.search(r"Helpful Answer:(.*)", books_retriever, re.DOTALL)
93
-
94
  if corrected_text_match:
95
  corrected_text_books = corrected_text_match.group(1).strip()
96
  else:
97
  corrected_text_books = "No helpful answer found."
 
98
  st.session_state.messages.append({'role': 'assistant', "content": corrected_text_books})
99
 
100
  for message in st.session_state.messages:
 
85
  return qa.run(query)
86
 
87
  user_prompt = st.chat_input("Ask me anything about the content of the PDF(s):")
88
+ print("user input:", user_prompt)
89
  if user_prompt:
90
  st.session_state.messages.append({'role': 'user', "content": user_prompt})
91
  books_retriever = test_rag(books_db_client_retriever, user_prompt)
92
+ print("books retriver:",books_retriever)
93
  # Extracting the relevant answer using regex
94
  corrected_text_match = re.search(r"Helpful Answer:(.*)", books_retriever, re.DOTALL)
95
+ print("corrected text match:", corrected_text_match)
96
  if corrected_text_match:
97
  corrected_text_books = corrected_text_match.group(1).strip()
98
  else:
99
  corrected_text_books = "No helpful answer found."
100
+ print("corrected text books: ",corrected_text_books)
101
  st.session_state.messages.append({'role': 'assistant', "content": corrected_text_books})
102
 
103
  for message in st.session_state.messages: