Update app.py

app.py

@@ -162,6 +162,14 @@ def generate_pdf(input_data, prediction, heatmap_fig):
     # Prediction Results
     story.append(Paragraph("Prediction Results", styles['Heading2']))
     high_risk_text = "<br/>".join(format_high_risk_phases(prediction['high_risk_phases']))
+    
+    # Check for 2-week risk alert in AI insights
+    two_week_alert = next((insight for insight in prediction['ai_insights'].split("; ") if "2-Week Risk Alert" in insight), None)
+    if two_week_alert:
+        story.append(Paragraph("2-Week Risk Alert", styles['Heading3']))
+        story.append(Paragraph(two_week_alert, styles['Normal']))
+        story.append(Spacer(1, 12))
+    
     prediction_fields = [
         f"Delay Probability: {prediction['delay_probability']:.2f}%",
         f"High Risk Phases:<br/>{high_risk_text}",
@@ -189,7 +197,7 @@ def save_to_salesforce(input_data, prediction, pdf_buffer):
     if sf is None:
         return "Salesforce connection not established."
     try:
-        # Prepare data for Delay_Predictor__c object
+        # Prepare data for Project_Delay_Predictor__c object
         sf_data = {
             "Project_Name__c": input_data["project_name"],
             "Phase__c": input_data["phase"],
@@ -208,8 +216,8 @@ def save_to_salesforce(input_data, prediction, pdf_buffer):
             "AI_Insights__c": prediction["ai_insights"],
             "High_Risk_Phases__c": "; ".join(format_high_risk_phases(prediction["high_risk_phases"]))
         }
-        # Create a new record in Delay_Predictor__c
-        result = sf.Delay_Predictor__c.create(sf_data)
+        # Create a new record in Project_Delay_Predictor__c
+        result = sf.Project_Delay_Predictor__c.create(sf_data)
         if not result["success"]:
             return f"Salesforce save failed: {result['errors']}"
 
@@ -244,8 +252,8 @@ def save_to_salesforce(input_data, prediction, pdf_buffer):
         pdf_url = f"{sf_instance_url}/sfc/servlet.shepherd/document/download/{content_document_id}"
         logger.info(f"Generated PDF URL: {pdf_url}")
 
-        # Update the Delay_Predictor__c record with the PDF URL
-        update_result = sf.Delay_Predictor__c.update(record_id, {"PDF_Report__c": pdf_url})
+        # Update the Project_Delay_Predictor__c record with the PDF URL
+        update_result = sf.Project_Delay_Predictor__c.update(record_id, {"PDF_Report__c": pdf_url})
         if update_result != 204:
             logger.error(f"Failed to update PDF_Report__c with URL: {pdf_url}")
             return f"Failed to update PDF_Report__c field: {update_result}"
@@ -318,7 +326,7 @@ if submit_button:
             else:
                 input_data["weather_impact_score"] = weather_data["weather_impact_score"]
                 input_data["weather_condition"] = weather_data["weather_condition"]
-                st.write(f"**Weather Data for {project_location} on {input_data['weather_forecast_date']}**:")
+                st.write(f"**Weather Data for {project_location} on {input_data['weather_forecast_date']}:**")
                 st.write(f"- Condition: {weather_data['weather_condition']}")
                 st.write(f"- Impact Score: {weather_data['weather_impact_score']}")
                 st.write(f"- Temperature: {weather_data['temperature']}°C")