Update app.py

app.py

@@ -96,7 +96,10 @@ def convert_pdf_to_excel(pdf_file):
 
                 if 'Part Name' in data[index].strip():
                     each_table_data.append(data[index+1])
-                    
+                
+                if 'Inventory Category' in data[index].strip():
+                    each_table_data.append(data[index+1].replace('Receive Type',""))
+
                 if data[index].strip() == 'ADJ':
                     row_start_index = index + 1
                     
@@ -113,13 +116,13 @@ def convert_pdf_to_excel(pdf_file):
             whole_data.append(each_table_data)
 
     whole_data = pd.DataFrame(whole_data)
-    whole_data.columns = ["Part No.","Part Color Code","Part Name",'Date Qty']
+    whole_data.columns = ["Part No.","Part Color Code","Part Name",'Inventory Category','Date Qty']
     extracted_file = "Data Extracted.xlsx"
     data_for_mapping = "Data Mapping.xlsx"
 
     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 color  as per pdf','Customer Part name as per pdf']
+    extracted_data_for_mapping = extracted_data_for_mapping.drop_duplicates(subset=["Part No.","Part Color Code","Part Name","Inventory Category"])
+    extracted_data_for_mapping.columns = ['Customer Part no as per pdf','Customer Part color  as per pdf','Customer Part name as per pdf',"Inventory Category"]
     extracted_data_for_mapping['Item Code'] = ""
 
     whole_data.to_excel(extracted_file, index=False)
@@ -140,7 +143,7 @@ def map_data_to_template(excel_file, mapping_file):
     extracted_data[['SchDate','Qty']]= pd.DataFrame(extracted_data['Date Qty'].to_list(), index= extracted_data.index)
     extracted_data = extracted_data.drop('Date Qty',axis=1)
     extracted_data = extracted_data[~extracted_data['SchDate'].isna()]
-    mapped_data = extracted_data.merge(mapping_data, on =['Part No.'],how='outer')[['Item Code','SchDate','Qty']]
+    mapped_data = extracted_data.merge(mapping_data, on =['Part No.'],how='outer')[['Item Code','SchDate','Qty','Inventory Category']]
     mapped_data = mapped_data[~mapped_data["SchDate"].isna()]
     mapped_data = mapped_data[~mapped_data["SchDate"].str.strip().isin(["",None])]
     mapped_data['SOType'] = "R"
@@ -227,7 +230,7 @@ def main():
         if file_present:
             mapping_data_from_drive = pd.read_excel(MAPPING_FILENAME)
             extracted_data_for_mapping = pd.read_excel(data_for_mapping)
-            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 = extracted_data_for_mapping[['Customer Part no as per pdf','Customer Part name as per pdf','Customer Part color  as per pdf',"Inventory Category"]].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