Update Chat_with_pdf_OpenAI.py

Chat_with_pdf_OpenAI.py

@@ -38,10 +38,10 @@ def openai_pdf():
       embeddings = OpenAIEmbeddings()
       knowledge_base = FAISS.from_texts(chunks, embeddings)
         
-      if 'generated' not in st.session_state:
-          st.session_state['generated'] = []
-      if 'past' not in st.session_state:
-          st.session_state['past'] = []
+      # if 'generated' not in st.session_state:
+          # st.session_state['generated'] = []
+      # if 'past' not in st.session_state:
+          # st.session_state['past'] = []
       
       # show user input
       user_question = st.text_input("Ask a question about your PDF:")
@@ -54,11 +54,11 @@ def openai_pdf():
           response = chain.run(input_documents=docs, question=user_question)
           print(cb)
            
-        #st.write(response)
-        st.session_state.past.append(user_question)
-        st.session_state.generated.append(response)
+        st.write(response)
+        # st.session_state.past.append(user_question)
+        # st.session_state.generated.append(response)
           
-      if st.session_state['generated']:
-        for i in range(len(st.session_state['generated'])-1, -1, -1):
-            message(st.session_state["generated"][i], key=str(i))
-            message(st.session_state['past'][i], is_user=True, key=str(i) + '_user')
+      # if st.session_state['generated']:
+        # for i in range(len(st.session_state['generated'])-1, -1, -1):
+            # message(st.session_state["generated"][i], key=str(i))
+            # message(st.session_state['past'][i], is_user=True, key=str(i) + '_user')