Samarth991 commited on
Commit
0d5d2ec
·
1 Parent(s): f18883e

added summary

Browse files
Files changed (1) hide show
  1. QA_Bot.py +3 -1
QA_Bot.py CHANGED
@@ -26,13 +26,15 @@ def QA_Bot(vectorstore,API_KEY,documents):
26
  st.markdown(message["content"])
27
  summary_response = summarize_data(documents,API_KEY)
28
  print(summary_response)
29
-
 
30
  # React to user input
31
  if prompt := st.chat_input("What is up?"):
32
  # Display user message in chat message container
33
  st.chat_message("user").markdown(prompt)
34
  # Add user message to chat history
35
  st.session_state.messages.append({"role": "user", "content": prompt})
 
36
 
37
  ai_response = Q_A(vectorstore,prompt,API_KEY)
38
  response = f"Echo: {ai_response}"
 
26
  st.markdown(message["content"])
27
  summary_response = summarize_data(documents,API_KEY)
28
  print(summary_response)
29
+
30
+
31
  # React to user input
32
  if prompt := st.chat_input("What is up?"):
33
  # Display user message in chat message container
34
  st.chat_message("user").markdown(prompt)
35
  # Add user message to chat history
36
  st.session_state.messages.append({"role": "user", "content": prompt})
37
+ st.session_state.messages.append({"role": "assistant", "content": summary_response})
38
 
39
  ai_response = Q_A(vectorstore,prompt,API_KEY)
40
  response = f"Echo: {ai_response}"