Update app.py

app.py

@@ -93,9 +93,9 @@ def get_conversation_chain(vectorstore):
     )
     return conversation_chain
 
-def main():
-
 
+def main():
+    
     if "conversation" not in st.session_state:
         st.session_state.conversation = None
     if "chat_history" not in st.session_state:
@@ -108,11 +108,10 @@ def main():
     user_question = st.text_input("Ask a question about your documents or data:")
 
     if user_question:
-        handle_userinput(user_question, st.session_state.vectorstore, st.session_state.dfs)  # Pass dfs
+        handle_userinput(user_question, st.session_state.vectorstore, st.session_state.dfs)
 
     with st.sidebar:
         st.subheader("Your files")
-        st.sidebar.image("logoqb.jpeg", use_container_width=True)
         uploaded_files = st.file_uploader(
             "Upload PDFs, CSVs, or Excel files (up to 3)", accept_multiple_files=True
         )
@@ -120,35 +119,53 @@ def main():
         if st.button("Process"):
             with st.spinner("Processing"):
                 pdf_docs = []
-                dfs = []  # List to store DataFrames
+                dfs = []
+                pdf_uploaded = False
+                data_uploaded = False
+
                 for uploaded_file in uploaded_files:
                     file_extension = uploaded_file.name.split(".")[-1].lower()
+
                     if file_extension == "pdf":
+                        if data_uploaded:  # Check if data was already uploaded
+                            if st.session_state.dfs:
+                                st.session_state.dfs = None
+                            data_uploaded = False
+                            st.warning("Switching to PDF mode. Data files removed.")  # Inform user
                         pdf_docs.append(uploaded_file)
-                    elif file_extension in ["csv", "xlsx", "xls"]:  # Handle Excel too
+                        pdf_uploaded = True
+                    elif file_extension in ["csv", "xlsx", "xls"]:
+                        if pdf_uploaded:  # Check if PDF was already uploaded
+                            if st.session_state.vectorstore:
+                                st.session_state.vectorstore = None
+                                st.session_state.conversation = None
+                            pdf_uploaded = False
+                            st.warning("Switching to Data mode. PDF files removed.")  # Inform user
                         try:
                             if file_extension == 'csv':
                                 df = pd.read_csv(uploaded_file)
                             else:
                                 df = pd.read_excel(uploaded_file)
                             dfs.append(df)
+                            data_uploaded = True
                         except Exception as e:
                             st.error(f"Error reading {uploaded_file.name}: {e}")
-                            st.stop() #Stop execution if error in file
+                            st.stop()
 
-                if pdf_docs:  # Process PDFs if any are uploaded
+                if pdf_docs:
                     raw_text = get_pdf_text(pdf_docs)
                     text_chunks = get_text_chunks(raw_text)
                     st.session_state.vectorstore = get_vectorstore(text_chunks)
                     st.session_state.conversation = get_conversation_chain(st.session_state.vectorstore)
                 else:
-                    st.session_state.vectorstore = None  # Clear if no PDFs
+                    st.session_state.vectorstore = None
+                    st.session_state.conversation = None
 
-                if dfs:  # Store DataFrames in session state
+                if dfs:
                     st.session_state.dfs = dfs
                 else:
-                    st.session_state.dfs = None  # Clear if no data files
-
+                    st.session_state.dfs = None
 
 if __name__ == "__main__":
     main()
+