Update app.py

app.py

@@ -303,7 +303,13 @@ def main():
         else:
             st.error("Error: Converted Excel file not found")
         
+        st.markdown("##### Click the button below if you want to upload a new mapping file")
+        if st.button("Delete Mapping file in Cloud", key="delete"):
+            delete_master_file()
+            file_present = pull_mapping_file_from_drive()
+
         st.markdown("### STEP 2")
+        mapping_uploaded_file = None
         if not file_present:
             st.markdown("#### Upload the Data Master file with Item Code mapping")
             mapping_uploaded_file = st.file_uploader("Upload the Data Master file with Item Code mapping", type=["xlsx","ods"])
@@ -318,10 +324,37 @@ def main():
             extracted_data_for_mapping = extracted_data_for_mapping[extracted_data_for_mapping['Item Code'].isna()]
             unmapped_part_no = extracted_data_for_mapping['Customer Part no as per pdf'].nunique()
             if unmapped_part_no>0:
-                st.markdown("#### There are {} Part No. with No ItemCode present. Upload a new file after mapping them".format(unmapped_part_no))
-                mapping_uploaded_file = st.file_uploader("Upload the Data Master file with Item Code mapping", type=["xlsx","ods"])
+                st.markdown("There are {} Part No. with No ItemCode present. Upload a new file after mapping them".format(unmapped_part_no))
+                st.markdown("Do you want to skip this or Upload a new Mapping File")
+
+                if 'button_pressed' not in st.session_state:
+                    st.session_state.button_pressed = None
+                
+                # placeholder = st.empty()  # Create a placeholder
+
+                if st.session_state.button_pressed is None:
+                    if st.button("Skip"):
+                        st.session_state.button_pressed = "Skip"
+                        # placeholder.empty()  # Clear the placeholder content
+
+                    if st.button("Upload a new Master Mapping"):
+                        st.session_state.button_pressed = "Upload a new Master Mapping"
+                        # placeholder.empty()  # Clear the placeholder content
+                        
+                        
+                if st.session_state.button_pressed is not None:
+                    # Common block of code that uses the variable
+
+                    if st.session_state.button_pressed == "Skip":
+                        mapping_uploaded_file =  MAPPING_FILENAME
+                        # Add your code that runs when Yes is pressed
+
+                    elif st.session_state.button_pressed == "Upload a new Master Mapping":
+                        mapping_uploaded_file = st.file_uploader("Upload the Data Master file with Item Code mapping", type=["xlsx","ods"])
+                        # Add your code that runs when No is pressed
+
             else:
-                st.markdown("#### All Part No. are mapped with ItemCode so using the Mapping file available in Google Drive")
+                st.markdown("All Part No. are mapped with ItemCode so using the Mapping file available in Google Drive")
                 mapping_uploaded_file =  MAPPING_FILENAME
 
         if mapping_uploaded_file is not None:
@@ -332,7 +365,7 @@ def main():
 
             # Provide a link to download the final Excel file after mapping
             st.markdown("### FINAL DOWNLOAD")
-            st.markdown("### Final Excel File After Mapping")
+            st.markdown("Final Excel File After Mapping")
 
             final_excel_file = 'Final Data.xlsx'
             mapped_data.to_excel(final_excel_file, index=False)
@@ -348,6 +381,9 @@ def main():
                 )
             else:
                 st.error("Error: Converted Excel file not found")
+
+            st.session_state.button_pressed = None
+            
  
 if __name__ == "__main__":
     main()