Update app.py

app.py

@@ -125,12 +125,52 @@ def handle_user_input(question):
         else:
             st.write(bot_template.replace("{{MSG}}", message.content), unsafe_allow_html=True)
 
+# def main():
+#     st.set_page_config(page_title='Chat with Your own PDFs', page_icon=':books:')
+
+#     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 Your own PDFs :books:')
+#     question = st.text_input("Ask anything to your PDF: ")
+
+#     if question:
+#         handle_user_input(question)
+
+#     with st.sidebar:
+#         st.subheader("Upload your Documents Here: ")
+#         pdf_files = st.file_uploader("Choose your PDF Files and Press OK", type=['pdf'], accept_multiple_files=True)
+
+#         if st.button("OK"):
+#             with st.spinner("Processing your PDFs..."):
+
+#                 # Get PDF Text
+#                 raw_text = get_pdf_text(pdf_files)
+
+#                 # Get Text Chunks
+#                 text_chunks = get_chunk_text(raw_text)
+
+#                 # Create Vector Store
+
+#                 vector_store = get_vector_store(text_chunks)
+#                 st.write("DONE")
+
+#                 # Create conversation chain
+
+#                 st.session_state.conversation =  get_conversation_chain(vector_store)
+
 def main():
     st.set_page_config(page_title='Chat with Your own PDFs', page_icon=':books:')
 
     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:
+        # Initialize conversation only if it's not already present in session state
         st.session_state.conversation = None
 
     if "chat_history" not in st.session_state:
@@ -140,6 +180,9 @@ def main():
     question = st.text_input("Ask anything to your PDF: ")
 
     if question:
+        # Check if conversation is still None (not properly initialized)
+        if st.session_state.conversation is None:
+            st.session_state.conversation = initialize_conversation_chain()
         handle_user_input(question)
 
     with st.sidebar:
@@ -148,7 +191,6 @@ def main():
 
         if st.button("OK"):
             with st.spinner("Processing your PDFs..."):
-
                 # Get PDF Text
                 raw_text = get_pdf_text(pdf_files)
 
@@ -156,12 +198,18 @@ def main():
                 text_chunks = get_chunk_text(raw_text)
 
                 # Create Vector Store
-
                 vector_store = get_vector_store(text_chunks)
                 st.write("DONE")
 
                 # Create conversation chain
+                st.session_state.conversation = get_conversation_chain(vector_store)
+
+
+def initialize_conversation_chain():
+    return None  # Replace with the actual initialization
+
+
+main()
 
-                st.session_state.conversation =  get_conversation_chain(vector_store)
 
 main()