Update app.py

app.py

@@ -150,10 +150,20 @@ except ValueError as e:
 # File upload
 st.subheader("Step 1: Upload Documents")
 uploaded_files = st.file_uploader("Upload files (PDF, TXT, CSV)", accept_multiple_files=True)
+
 if st.button("Process Documents"):
     if uploaded_files:
-        result = rag_system.process_documents(uploaded_files)  # Pass uploaded files directly
-        st.success(result) if "successfully" in result else st.error(result)
+        # Process the uploaded files
+        result = rag_system.process_documents(uploaded_files)
+        
+        # Ensure that result is a string and display appropriately
+        if isinstance(result, str):
+            if "successfully" in result:
+                st.success(result)
+            else:
+                st.error(result)
+        else:
+            st.error("An unexpected error occurred during document processing.")
     else:
         st.warning("No files uploaded.")
 
@@ -169,4 +179,3 @@ if rag_system.qa_chain:
             st.chat_message("assistant").write(answer)
 else:
     st.info("Please process documents before asking questions.")
-