Update app.py

app.py

@@ -92,10 +92,11 @@ def main():
         # Get filenames from uploaded files
         filename1 = st.session_state.file1.name
         filename2 = st.session_state.file2.name
-        
-        if st.session_state.display_data["file1"] != st.session_state.file1 or st.session_state.display_data["file2"] != st.session_state.file2:
-            st.session_state.display_data = {"file1": st.session_state.file1, "file2": st.session_state.file2, "i": 0}
-            st.session_state.extracted_texts = {"file1": st.session_state.file1, "file2": st.session_state.file2, "extracted_text_1": extract_text_with_pypdf(st.session_state.file1), "extracted_text_2": extract_text_with_pypdf(st.session_state.file2)}
+
+        with st.spinner("Extracting text from PDFs"):
+            if st.session_state.display_data["file1"] != st.session_state.file1 or st.session_state.display_data["file2"] != st.session_state.file2:
+                st.session_state.display_data = {"file1": st.session_state.file1, "file2": st.session_state.file2, "i": 0}
+                st.session_state.extracted_texts = {"file1": st.session_state.display_data["file1"], "file2": st.session_state.display_data["file2"], "extracted_text_1": extract_text_with_pypdf(st.session_state.file1), "extracted_text_2": extract_text_with_pypdf(st.session_state.file2)}
         
         try:
             extracted_text1 = st.session_state.extracted_texts["extracted_text_1"]
@@ -106,7 +107,7 @@ def main():
             with col2.expander(filename2):
                 st.write("\n\n".join("\n\n------------------------\n\n".join(extracted_text2).splitlines()))
             
-            st.success(f"Content of files **{filename1}** and **{filename2}** have been extracted successfully.")
+            st.success(f"PDF text extraction complete")
         except Exception as e:
             st.error(f"Error saving files: {str(e)}")