Update app.py

app.py

@@ -69,21 +69,18 @@ except Exception as e:
 def generate_heatmap(phase, weather, model):
     print("Generating heatmap...")
     try:
-        # Prepare data for heatmap
         absentee_range = np.linspace(0, 100, 20)
         delay_log_range = np.linspace(0, 20, 20)
         framing = 1 if phase == "Framing" else 0
         foundation = 1 if phase == "Foundation" else 0
         weather_encoded = weather_map.get(weather, 0)
         
-        # Generate predictions for heatmap
         Z = np.zeros((len(delay_log_range), len(absentee_range)))
         for i, delay_log in enumerate(delay_log_range):
             for j, absentee in enumerate(absentee_range):
                 input_data = [[framing, foundation, weather_encoded, absentee, delay_log]]
                 Z[i, j] = model.predict(input_data)[0]
         
-        # Create heatmap
         plt.figure(figsize=(8, 6))
         sns.heatmap(Z, xticklabels=np.round(absentee_range, 1), yticklabels=np.round(delay_log_range, 1),
                     cmap="YlOrRd", annot=True, fmt=".1f", cbar_kws={'label': 'Predicted Delay %'})
@@ -91,7 +88,6 @@ def generate_heatmap(phase, weather, model):
         plt.ylabel("Previous Delay Log")
         plt.title(f"Delay Prediction Heatmap (Phase: {phase}, Weather: {weather})")
         
-        # Save heatmap
         output_dir = "pdf_reports"
         os.makedirs(output_dir, exist_ok=True)
         timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
@@ -123,7 +119,7 @@ def generate_pdf_report(phase, weather, absentee_pct, delay_log, prediction, ris
             f"Weather: {weather}",
             f"Absentee Percentage: {absentee_pct}%",
             f"Previous Delay Log: {delay_log}",
-            f"Predicted Delay: {prediction}%",
+            f"Predicted Delay: {prediction}%{' 🚩' if prediction >= 75 else ''}",
             f"Risk Level: {risk}",
             "AI Insight:"
         ]
@@ -299,7 +295,8 @@ try:
             prediction, risk, insight, pdf_base64, pdf_path, heatmap_path = predict_delay(phase, weather, absentee, delay_log)
             if prediction is None:
                 return f"Error: {insight}"
-            return (f"Predicted Delay: {prediction}%\n"
+            flag = " 🚩" if prediction >= 75 else ""
+            return (f"Predicted Delay: {prediction}%{flag}\n"
                     f"Risk Level: {risk}\n"
                     f"Insight: {insight}\n"
                     f"PDF Report: {'Saved locally at ' + pdf_path if pdf_path else 'Failed to generate'}\n"