Update app.py

app.py

@@ -11,8 +11,8 @@ from google_auth_oauthlib.flow import InstalledAppFlow
 from googleapiclient.discovery import build
 from googleapiclient.http import MediaIoBaseDownload,MediaFileUpload
 from google.oauth2 import service_account
-
-SCOPES = ['https://www.googleapis.com/auth/drive']
+from dotenv import load_dotenv
+load_dotenv()
 
 # Load credentials from environment variables
 SERVICE_ACCOUNT_INFO = {
@@ -29,6 +29,9 @@ SERVICE_ACCOUNT_INFO = {
   "universe_domain": "googleapis.com"
 }
 
+SCOPES = ['https://www.googleapis.com/auth/drive']
+
+
 def authenticate():
     # Authenticate using service account credentials
     creds = service_account.Credentials.from_service_account_info(SERVICE_ACCOUNT_INFO, scopes=['https://www.googleapis.com/auth/drive'])
@@ -117,7 +120,8 @@ def convert_pdf_to_excel(pdf_file):
 
     extracted_data_for_mapping = whole_data.drop('Date Qty',axis=1)
     extracted_data_for_mapping = extracted_data_for_mapping.drop_duplicates(subset=["Part No.","Part Color Code","Part Name"])
-    extracted_data_for_mapping.columns = ['Customer Part no as per pdf','Customer Part name as per pdf','Customer Part color  as per pdf']
+    extracted_data_for_mapping.columns = ['Customer Part no as per pdf','Customer Part color  as per pdf','Customer Part name as per pdf']
+    extracted_data_for_mapping['Item Code'] = ""
 
     whole_data.to_excel(extracted_file, index=False)
     extracted_data_for_mapping.to_excel(data_for_mapping, index=False)
@@ -204,7 +208,6 @@ def pull_mapping_file_from_drive():
         fh.close()
         return 1
     
-    print('No Excel files found.')
     return 0
 
 def main():
@@ -220,13 +223,11 @@ def main():
         extracted_file,data_for_mapping = convert_pdf_to_excel(uploaded_file)
         
         file_present = pull_mapping_file_from_drive()
-        print(file_present)
 
         if file_present:
             mapping_data_from_drive = pd.read_excel(MAPPING_FILENAME)
-            print(mapping_data_from_drive)
             extracted_data_for_mapping = pd.read_excel(data_for_mapping)
-            extracted_data_for_mapping = extracted_data_for_mapping.merge(mapping_data_from_drive, on = ['Customer Part no as per pdf','Customer Part name as per pdf','Customer Part color  as per pdf'], how='outer')
+            extracted_data_for_mapping = extracted_data_for_mapping[['Customer Part no as per pdf','Customer Part name as per pdf','Customer Part color  as per pdf']].merge(mapping_data_from_drive, on = ['Customer Part no as per pdf','Customer Part name as per pdf','Customer Part color  as per pdf'], how='outer')
             extracted_data_for_mapping.to_excel(data_for_mapping,index=False)
 
         # Download link for the Excel file
@@ -252,14 +253,14 @@ def main():
             # mapping_data = mapping_data.rename(columns = {'Customer Part no as per pdf':'Part No.'})
             data_for_mapping = "Data Mapping.xlsx"
             extracted_data_for_mapping = pd.read_excel(data_for_mapping)
-            print(extracted_data_for_mapping)
-            print(mapping_data)
-
-            extracted_data_for_mapping = extracted_data_for_mapping[~extracted_data_for_mapping['Customer Part no as per pdf'].isin(mapping_data['Customer Part no as per pdf'])]
+            if 'Item Code' not in extracted_data_for_mapping.columns:
+                extracted_data_for_mapping['Item Code'] = ""
+            
+            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"])
+                mapping_uploaded_file = st.file_uploader("Upload the Data Master file with Item Code mapping", type=["xlsx","ods"])
             # else:
             #     st.markdown("#### Using the Mapping file available in Google Drive")
             #     mapping_uploaded_file =  MAPPING_FILENAME