Update app.py

app.py

@@ -93,7 +93,8 @@ picklist_mapping = {
         'ops': 'UV Verification',
         'maintenance': 'Smart Log',
         'cell': 'Cell Analysis',
-        'uv': 'UV Verification'
+        'uv': 'UV Verification',
+        'weight log': 'Smart Log'  # Added mapping for 'weight log'
     }
 }
 
@@ -200,11 +201,18 @@ def save_to_salesforce(df, summary, anomalies, amc_reminders, insights):
             amc_date_str = None
             if pd.notna(row['amc_date']):
                 try:
-                    amc_date = pd.to_datetime(row['amc_date']).strftime('%Y-%m-%d')
-                    amc_date_str = amc_date
-                    amc_date_dt = datetime.strptime(amc_date, '%Y-%m-%d')
-                    if status == "Active" and current_date.date() <= amc_date_dt.date() <= next_30_days.date():
-                        logging.info(f"Record qualifies for AMC Reminders: Device ID {row['device_id']}, AMC Date {amc_date_str}")
+                    # Try parsing as DD-MM-YYYY
+                    amc_date = pd.to_datetime(row['amc_date'], format='%d-%m-%Y', errors='coerce')
+                    if pd.isna(amc_date):
+                        # If that fails, try YYYY-MM-DD
+                        amc_date = pd.to_datetime(row['amc_date'], format='%Y-%m-%d', errors='coerce')
+                    if pd.isna(amc_date):
+                        logging.error(f"Failed to parse AMC Date for Device ID {row['device_id']}: {row['amc_date']}")
+                    else:
+                        amc_date_str = amc_date.strftime('%Y-%m-%d')
+                        amc_date_dt = datetime.strptime(amc_date_str, '%Y-%m-%d')
+                        if status == "Active" and current_date.date() <= amc_date_dt.date() <= next_30_days.date():
+                            logging.info(f"Record qualifies for AMC Reminders: Device ID {row['device_id']}, AMC Date {amc_date_str}")
                 except Exception as e:
                     logging.error(f"Failed to parse AMC Date for Device ID {row['device_id']}: {str(e)}")
                     amc_date_str = None