Update app.py

app.py

@@ -121,28 +121,47 @@ def main():
         st.warning("Please enter a valid HuggingFace API token.")
         return
 
+    # Initialize session state variables
+    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 = []
+
+    # User input for questions
     user_question = st.text_input("Ask a question about your documents:")
     if user_question:
-        handle_userinput(user_question)
+        if st.session_state.conversation:
+            handle_userinput(user_question)
+        else:
+            st.warning("Please upload and process files first!")
 
+    # File uploader and processing
     docs = st.file_uploader("Upload your PDFs here and click on 'Process'", accept_multiple_files=True)
     if st.button("Process"):
         with st.spinner("Processing"):
-            doc_list = []
-            for file in docs:
-                if file.type == 'text/plain':
-                    doc_list.extend(get_text_file(file))
-                elif file.type in ['application/octet-stream', 'application/pdf']:
-                    doc_list.extend(get_pdf_text(file))
-                elif file.type == 'text/csv':
-                    doc_list.extend(get_csv_file(file))
-                elif file.type == 'application/json':
-                    doc_list.extend(get_json_file(file))
-
-            text_chunks = get_text_chunks(doc_list)
-            vectorstore = get_vectorstore(text_chunks)
-
-            st.session_state.conversation = get_conversation_chain(vectorstore, tokenH)
+            if docs:
+                doc_list = []
+                for file in docs:
+                    if file.type == 'text/plain':
+                        doc_list.extend(get_text_file(file))
+                    elif file.type in ['application/octet-stream', 'application/pdf']:
+                        doc_list.extend(get_pdf_text(file))
+                    elif file.type == 'text/csv':
+                        doc_list.extend(get_csv_file(file))
+                    elif file.type == 'application/json':
+                        doc_list.extend(get_json_file(file))
+
+                # Generate text chunks
+                text_chunks = get_text_chunks(doc_list)
+
+                # Create vector store
+                vectorstore = get_vectorstore(text_chunks)
+
+                # Initialize conversation chain
+                st.session_state.conversation = get_conversation_chain(vectorstore, tokenH)
+                st.success("Documents processed successfully!")
+            else:
+                st.warning("Please upload at least one document to process.")
 
 
 if __name__ == '__main__':