Update app.py

app.py

@@ -59,6 +59,19 @@ project_choices = [
 ]
 ai_suggestion_choices = ["Move", "Repair", "Replace"]  # Removed "Pause Rent"
 
+def validate_date(last_maint):
+    try:
+        # Check if the provided date is in the future
+        if last_maint and last_maint != "N/A":
+            entered_date = datetime.datetime.strptime(last_maint, "%Y-%m-%d")
+            current_date = datetime.datetime.today()
+
+            if entered_date > current_date:
+                raise ValueError(f"Last Maintenance Date cannot be in the future. You entered: {last_maint}")
+        return last_maint
+    except ValueError as ve:
+        raise ValueError(f"Invalid date format for Last Maintenance. Please use YYYY-MM-DD. Error: {ve}")
+
 def call_ai_model(usage, idle, freq, cost, last):
     try:
         usage = float(usage) if usage is not None and not np.isnan(usage) else 0.0
@@ -99,6 +112,9 @@ def process_equipment_utilization(equip, proj, use_h, idle_h, move_f, cost_h, la
         if not sf:
             raise ValueError("Salesforce connection is not initialized. Please check credentials and try again.")
 
+        # Validate the Last Maintenance date
+        last_maint = validate_date(last_maint)
+
         # Always run AI model to get suggestion, confidence, score
         ai_sug_generated, conf, score = call_ai_model(use_h, idle_h, move_f, cost_h, last_maint)