Update app.py

app.py

@@ -151,6 +151,7 @@ def save_to_salesforce(df, reminders_df):
         reminder_device_ids = set(reminders_df['device_id']) if not reminders_df.empty else set()
         logging.info(f"Processing {len(df)} records for Salesforce")
 
+        # Process all records from filtered_df
         for idx, row in df.iterrows():
             status = str(row['status']).lower()
             log_type = str(row['log_type']).lower()
@@ -183,6 +184,29 @@ def save_to_salesforce(df, reminders_df):
             }
             records.append(record)
 
+        # Process reminders_df explicitly to ensure AMC reminders are saved
+        if not reminders_df.empty:
+            logging.info(f"Processing {len(reminders_df)} AMC reminder records for Salesforce")
+            for idx, row in reminders_df.iterrows():
+                status = str(row['status']).lower()
+                log_type = str(row['log_type']).lower()
+                status_mapped = picklist_mapping['Status__c'].get(status, status_values[0] if status_values else 'Active')
+                log_type_mapped = picklist_mapping['Log_Type__c'].get(log_type, log_type_values[0] if log_type_values else 'Smart Log')
+                
+                amc_date_str = row['amc_date'].strftime('%Y-%m-%d') if pd.notna(row['amc_date']) else None
+                if amc_date_str:
+                    record = {
+                        'Device_Id__c': str(row['device_id'])[:50],
+                        'Log_Type__c': log_type_mapped,
+                        'Status__c': status_mapped,
+                        'Timestamp__c': row['timestamp'].isoformat() if pd.notna(row['timestamp']) else None,
+                        'Usage_Hours__c': float(row['usage_hours']) if pd.notna(row['usage_hours']) else 0.0,
+                        'Downtime__c': float(row['downtime']) if pd.notna(row['downtime']) else 0.0,
+                        'AMC_Date__c': amc_date_str
+                    }
+                    records.append(record)
+                    logging.info(f"Added AMC reminder for Device ID {row['device_id']} with AMC Date {amc_date_str}")
+
         if records:
             batch_size = 100
             for i in range(0, len(records), batch_size):