Update app.py

app.py

@@ -112,6 +112,10 @@ def process_equipment_utilization(equip, proj, use_h, idle_h, move_f, cost_h, la
             if value is None or np.isnan(value):
                 raise ValueError(f"Invalid value for {field}: {value}. Must be a valid number.")
 
+        # Ensure last_maint is a string or None
+        if last_maint is None or pd.isna(last_maint):
+            last_maint = "N/A"
+
         summary = {
             "Equipment Name": equip,
             "Project": proj,
@@ -294,6 +298,10 @@ def batch_upload(csv_file):
             if df[col].isna().any():
                 raise ValueError(f"Column '{col}' contains missing values. Please ensure all rows have valid equipment and project names.")
 
+        # Handle NaN in last_maintenance column
+        if 'last_maintenance' in df.columns:
+            df['last_maintenance'] = df['last_maintenance'].apply(lambda x: "N/A" if pd.isna(x) else str(x))
+
         # Validate ai_suggestion if provided
         if 'ai_suggestion' in df.columns:
             df['ai_suggestion'] = df['ai_suggestion'].apply(
@@ -304,11 +312,14 @@ def batch_upload(csv_file):
         csv_files = []
         for idx, row in df.iterrows():
             try:
+                last_maint = row.get('last_maintenance', 'N/A')
+                if pd.isna(last_maint):
+                    last_maint = "N/A"
                 rec = process_equipment_utilization(
                     row['equipment_name'], row['project_name'],
                     row['usage_hours'], row['idle_hours'],
                     row['movement_frequency'], row['cost_per_hour'],
-                    row.get('last_maintenance', 'N/A'), row.get('ai_suggestion', '')
+                    last_maint, row.get('ai_suggestion', '')
                 )
                 ids.append(rec['Salesforce_Record_Id'])
                 csv_files.append(rec['CSV_Report_Link'])