Update app.py

app.py

@@ -115,57 +115,105 @@ def handle_userinput(user_question):
                 "{{MSG}}", message.content), unsafe_allow_html=True)
 
 
+def initialize_conversation_chain(text_chunks):
+    vectorstore = get_vectorstore(text_chunks)
+    return get_conversation_chain(vectorstore)
 
 def main():
-    
-    st.set_page_config(page_title="Chat with multiple PDFs",
-                       page_icon="logo1.png" )
+    st.set_page_config(page_title="Chat with multiple PDFs", page_icon="logo1.png")
     st.write(css, unsafe_allow_html=True)
 
-    if "conversation" not in st.session_state:
+    if "conversation" not in st.session_state or st.session_state.conversation is None:
         st.session_state.conversation = None
+
     if "chat_history" not in st.session_state:
         st.session_state.chat_history = None
 
     st.header("Chat with multiple PDFs :books:")
     user_question = st.text_input("Ask a question about your documents:")
+
     if user_question:
         handle_userinput(user_question)
 
     with st.sidebar:
         st.subheader("Your documents")
-        pdf_docs = st.file_uploader(
-            "Upload your PDFs here and click on 'Process'", accept_multiple_files=True)
-        
+        pdf_docs = st.file_uploader("Upload your PDFs here and click on 'Process'", accept_multiple_files=True)
 
         if st.button("Process"):
             with st.spinner("Processing"):
-                # get pdf text
+                # Get PDF text
                 raw_text = get_pdf_text(pdf_docs)
 
-                # get the text chunks
+                # Get the text chunks
                 text_chunks = get_text_chunks(raw_text)
 
-                # create vector store
-                vectorstore = get_vectorstore(text_chunks)
-
-                # create conversation chain
-                st.session_state.conversation = get_conversation_chain(
-                    vectorstore)
-                
-
-
                 # Clear chat history
                 st.session_state.chat_history = None
-                
+
+                # Initialize conversation chain
+                st.session_state.conversation = initialize_conversation_chain(text_chunks)
+
     if st.session_state.conversation is not None:
         if st.session_state.chat_history is None:
             # Greet the user
             greeting = "Hello! How can I assist you with your documents?"
             st.write(bot_template.replace("{{MSG}}", greeting), unsafe_allow_html=True)
 
+if __name__ == '__main__':
+    main()
+
+            
+
+# def main():
+    
+#     st.set_page_config(page_title="Chat with multiple PDFs",
+#                        page_icon="logo1.png" )
+#     st.write(css, unsafe_allow_html=True)
+
+#     if "conversation" not in st.session_state:
+#         st.session_state.conversation = None
+#     if "chat_history" not in st.session_state:
+#         st.session_state.chat_history = None
+
+#     st.header("Chat with multiple PDFs :books:")
+#     user_question = st.text_input("Ask a question about your documents:")
+#     if user_question:
+#         handle_userinput(user_question)
+
+#     with st.sidebar:
+#         st.subheader("Your documents")
+#         pdf_docs = st.file_uploader(
+#             "Upload your PDFs here and click on 'Process'", accept_multiple_files=True)
+        
+
+#         if st.button("Process"):
+#             with st.spinner("Processing"):
+#                 # get pdf text
+#                 raw_text = get_pdf_text(pdf_docs)
+
+#                 # get the text chunks
+#                 text_chunks = get_text_chunks(raw_text)
+
+#                 # create vector store
+#                 vectorstore = get_vectorstore(text_chunks)
+
+#                 # create conversation chain
+#                 st.session_state.conversation = get_conversation_chain(
+#                     vectorstore)
                 
 
 
-if __name__ == '__main__':
-    main()
+#                 # Clear chat history
+#                 st.session_state.chat_history = None
+                
+#     if st.session_state.conversation is not None:
+#         if st.session_state.chat_history is None:
+#             # Greet the user
+#             greeting = "Hello! How can I assist you with your documents?"
+#             st.write(bot_template.replace("{{MSG}}", greeting), unsafe_allow_html=True)
+
+                
+
+
+# if __name__ == '__main__':
+#     main()