Update Chat_with_pdf_LLM.py

Chat_with_pdf_LLM.py

@@ -39,7 +39,7 @@ def LLM_pdf(model_name = 'google/flan-t5-large'):
           st.session_state['generated'] = []
       if 'past' not in st.session_state:
           st.session_state['past'] = []
-      print(st.session_state['generated'],st.session_state['past'])
+      # print(st.session_state['generated'],st.session_state['past'])
       
       # show user input
       user_question = st.text_input("Ask a question about your PDF:")
@@ -52,10 +52,15 @@ def LLM_pdf(model_name = 'google/flan-t5-large'):
         response = chain.run(input_documents=docs,question=user_question)
            
         #st.write(response)
+          
+        # # append user_input and output to state
         st.session_state.past.append(user_question)
         st.session_state.generated.append(response)
           
+      # If responses have been generated by the model
       if st.session_state['generated']:
+        # Reverse iteration through the list
         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')
+            # message from streamlit_chat
+            message(st.session_state['past'][::-1][i], is_user=True, key=str(i) + '_user')
+            message(st.session_state['generated'][::-1][i], key=str(i))