Update app.py

app.py

@@ -10,6 +10,7 @@ from datetime import datetime
 from reportlab.lib.pagesizes import letter
 from reportlab.pdfgen import canvas
 from reportlab.lib.utils import ImageReader
+from reportlab.lib.colors import red, black
 
 # Set up logging
 logging.basicConfig(level=logging.DEBUG)
@@ -50,7 +51,7 @@ def process_uploaded_file(file):
     return planned_cost, actual_spend, line_items
 
 # Function to generate a bar chart for the PDF report and Gradio UI with amount labels
-def generate_plot(planned_cost_inr, actual_spend_inr, forecast_cost_inr):
+def generate_bar_plot(planned_cost_inr, actual_spend_inr, forecast_cost_inr):
     fig, ax = plt.subplots(figsize=(8, 6))
     categories = ['Planned Cost', 'Actual Spend', 'Forecasted Cost']
     values = [planned_cost_inr, actual_spend_inr, forecast_cost_inr]
@@ -82,8 +83,47 @@ def generate_plot(planned_cost_inr, actual_spend_inr, forecast_cost_inr):
     plt.close()
     return gradio_image, buf_pdf
 
-# Function to generate a PDF report locally
-def generate_pdf(planned_cost_inr, actual_spend_inr, forecast_cost_inr, total_risk, risk_percentage, insights, status, top_causes, category, project_phase, material_cost_index, labor_index, scope_change_impact, alert_message, chart_image):
+# Function to generate a pie chart for risk distribution
+def generate_pie_chart_data(cost_deviation_factor, material_cost_factor, labor_cost_factor, scope_change_factor):
+    labels = ['Cost Deviation', 'Material Cost', 'Labor Cost', 'Scope Change']
+    values = [
+        max(cost_deviation_factor * 100, 0),
+        max(material_cost_factor * 100, 0),
+        max(labor_cost_factor * 100, 0),
+        max(scope_change_factor * 100, 0)
+    ]
+    # Ensure at least a small value to display the chart properly
+    total = sum(values)
+    if total == 0:
+        values = [25, 25, 25, 25]  # Equal distribution if no risk factors
+    return {
+        "type": "pie",
+        "data": {
+            "labels": labels,
+            "datasets": [{
+                "label": "Risk Distribution",
+                "data": values,
+                "backgroundColor": ["#FF6384", "#36A2EB", "#FFCE56", "#4BC0C0"],
+                "borderColor": ["#FF6384", "#36A2EB", "#FFCE56", "#4BC0C0"],
+                "borderWidth": 1
+            }]
+        },
+        "options": {
+            "responsive": true,
+            "plugins": {
+                "legend": {
+                    "position": "top"
+                },
+                "title": {
+                    "display": true,
+                    "text": "Risk Factor Distribution"
+                }
+            }
+        }
+    }
+
+# Function to generate a PDF report with red zone styling for critical status
+def generate_pdf(planned_cost_inr, actual_spend_inr, forecast_cost_inr, total_risk, risk_percentage, insights, status, top_causes, category, project_phase, material_cost_index, labor_index, scope_change_impact, alert_message, bar_chart_image):
     pdf_path = f"budget_report_{datetime.now().strftime('%Y%m%d_%H%M%S')}.pdf"
     c = canvas.Canvas(pdf_path, pagesize=letter)
     width, height = letter
@@ -95,6 +135,11 @@ def generate_pdf(planned_cost_inr, actual_spend_inr, forecast_cost_inr, total_ri
     # Prediction Metrics with Indian formatting
     c.setFont("Helvetica", 12)
     y_position = height - 100
+
+    # Apply red zone styling for critical status
+    text_color = red if status == "Critical" else black
+    c.setFillColor(text_color)
+
     c.drawString(50, y_position, f"Category: {category}")
     y_position -= 20
     c.drawString(50, y_position, f"Project Phase: {project_phase}")
@@ -124,8 +169,8 @@ def generate_pdf(planned_cost_inr, actual_spend_inr, forecast_cost_inr, total_ri
     c.drawString(50, y_position, f"Alert: {alert_message}")
     y_position -= 40
 
-    # Add the chart
-    chart_reader = ImageReader(chart_image)
+    # Add the bar chart
+    chart_reader = ImageReader(bar_chart_image)
     c.drawImage(chart_reader, 50, y_position - 300, width=500, height=300)
 
     c.showPage()
@@ -175,7 +220,7 @@ def predict_risk(file, category, material_cost_index, labor_index, scope_change_
     try:
         planned_cost_inr, actual_spend_inr, line_items = process_uploaded_file(file)
     except Exception as e:
-        return f"Error processing file: {str(e)}", None, None, None
+        return f"Error processing file: {str(e)}", None, None, None, None, None, None
 
     # Validate numeric inputs
     try:
@@ -184,7 +229,7 @@ def predict_risk(file, category, material_cost_index, labor_index, scope_change_
         scope_change_impact = float(scope_change_impact) if scope_change_impact else 0
     except ValueError:
         logger.error("Invalid input: Inputs must be numeric.")
-        return "Error: All numeric inputs (material cost index, labor index, scope change impact) must be valid numbers.", None, None, None
+        return "Error: All numeric inputs (material cost index, labor index, scope change impact) must be valid numbers.", None, None, None, None, None, None
 
     logger.debug(f"Starting prediction with inputs: planned_cost_inr={planned_cost_inr}, actual_spend_inr={actual_spend_inr}, "
                  f"category={category}, material_cost_index={material_cost_index}, labor_index={labor_index}, "
@@ -229,15 +274,19 @@ def predict_risk(file, category, material_cost_index, labor_index, scope_change_
     # Check for >10% deviation and generate alert
     deviation = ((forecast_cost_inr - planned_cost_inr) / planned_cost_inr * 100) if planned_cost_inr > 0 else 0
     alert_message = "Alert: Forecasted cost exceeds planned cost by more than 10%. Notify finance and engineering teams." if deviation > 10 else "No alert triggered."
+    alert_style = "background-color: #ffcccc; padding: 10px; border: 1px solid red; border-radius: 5px;" if deviation > 10 else "background-color: #ccffcc; padding: 10px; border: 1px solid green; border-radius: 5px;"
 
     logger.debug(f"Calculation results: total_risk={total_risk}, risk_percentage={risk_percentage}, forecast_cost_inr={forecast_cost_inr}, "
                  f"insights={insights}, status={status}, top_causes={top_causes}, alert_message={alert_message}")
 
-    # Generate the plot for Gradio and PDF
-    chart_image, chart_image_pdf = generate_plot(planned_cost_inr, actual_spend_inr, forecast_cost_inr)
+    # Generate the bar chart for Gradio and PDF
+    bar_chart_image, bar_chart_image_pdf = generate_bar_plot(planned_cost_inr, actual_spend_inr, forecast_cost_inr)
+
+    # Generate pie chart for risk distribution
+    pie_chart_data = generate_pie_chart_data(cost_deviation_factor, material_cost_factor, labor_cost_factor, scope_change_factor)
 
     # Generate local PDF report
-    pdf_file = generate_pdf(planned_cost_inr, actual_spend_inr, forecast_cost_inr, total_risk, risk_percentage, insights, status, top_causes, category, project_phase, material_cost_index, labor_index, scope_change_impact, alert_message, chart_image_pdf)
+    pdf_file = generate_pdf(planned_cost_inr, actual_spend_inr, forecast_cost_inr, total_risk, risk_percentage, insights, status, top_causes, category, project_phase, material_cost_index, labor_index, scope_change_impact, alert_message, bar_chart_image_pdf)
 
     # Generate Excel file
     excel_file = generate_excel(planned_cost_inr, actual_spend_inr, forecast_cost_inr, total_risk, risk_percentage, insights, status, top_causes, category, project_phase, material_cost_index, labor_index, scope_change_impact, alert_message)
@@ -250,8 +299,7 @@ def predict_risk(file, category, material_cost_index, labor_index, scope_change_
         f"Risk Level: {risk_level}\n"
         f"Risk Percentage: {risk_percentage}%\n"
         f"Status: {status}\n"
-        f"Insights: {insights} due to {top_causes.lower()}.\n"
-        f"Alert: {alert_message}\n\n"
+        f"Insights: {insights} due to {top_causes.lower()}.\n\n"
         f"Project Details\n"
         f"----------------------------------------\n"
         f"Category: {category}\n"
@@ -261,7 +309,7 @@ def predict_risk(file, category, material_cost_index, labor_index, scope_change_
         f"Scope Change Impact: {scope_change_impact}%\n\n"
         f"Forecast Chart\n"
         f"----------------------------------------\n"
-        f"[Chart displayed below]\n\n"
+        f"[Bar chart displayed below]\n\n"
         f"Detailed Metrics\n"
         f"----------------------------------------\n"
         f"Total Risk: {total_risk}\n"
@@ -272,7 +320,11 @@ def predict_risk(file, category, material_cost_index, labor_index, scope_change_
         f"Local PDF Report: [Download link below]\n"
         f"Excel Report: [Download link below]"
     )
-    return output_text, chart_image, pdf_file, excel_file
+
+    # Dashboard content (simplified risk level display)
+    dashboard_content = f"**Risk Level for {category}: {risk_level} ({risk_percentage}%)**"
+
+    return output_text, bar_chart_image, pie_chart_data, pdf_file, excel_file, f"<div style='{alert_style}'>{alert_message}</div>", dashboard_content
 
 # Function to update Material Cost Index explanation based on category
 def update_material_cost_explanation(category):
@@ -309,7 +361,7 @@ def update_labor_explanation(category):
         f"A value above 150 flags a potential risk. Example: If the index is 160, labor costs are 60% higher than the baseline."
     )
 
-# Custom CSS for the golden button with adjusted size
+# Custom CSS for the golden button and alert card
 custom_css = """
 #submit-button {
     background-color: #FFD700 !important;  /* Golden background */
@@ -331,7 +383,7 @@ custom_css = """
 # Custom Gradio Blocks interface
 with gr.Blocks(title="Budget Overrun Risk Estimator", css=custom_css) as demo:
     gr.Markdown("# Budget Overrun Risk Estimator")
-    gr.Markdown("Upload a CSV file with budget line items (columns: 'Planned_Cost', 'Actual_Spend') or enter totals manually. All numeric fields are required.")
+    gr.Markdown("Upload a CSV file with budget line items (columns: 'Planned_Cost', 'Actual_Spend') to estimate budget overrun risk. All numeric fields are required.")
 
     with gr.Row():
         with gr.Column():
@@ -353,8 +405,21 @@ with gr.Blocks(title="Budget Overrun Risk Estimator", css=custom_css) as demo:
                 submit_button = gr.Button("Submit", elem_id="submit-button")
 
         with gr.Column():
+            gr.Markdown("## Dashboard")
+            dashboard_output = gr.Markdown(label="Risk Level Dashboard")
+            
+            gr.Markdown("## Prediction Results")
             output_text = gr.Textbox(label="Prediction Results", lines=20, max_lines=30)
-            output_chart = gr.Image(label="Forecast Chart")
+            
+            gr.Markdown("## Forecast Chart")
+            bar_chart_output = gr.Image(label="Budget Overview (Bar Chart)")
+            
+            gr.Markdown("## Risk Distribution")
+            pie_chart_output = gr.Plot(label="Risk Factor Distribution (Pie Chart)")
+            
+            gr.Markdown("## Alerts")
+            alert_output = gr.HTML(label="Alert Notification")
+            
             output_pdf = gr.File(label="Download Local PDF Report")
             output_excel = gr.File(label="Download Excel Report")
 
@@ -372,10 +437,11 @@ with gr.Blocks(title="Budget Overrun Risk Estimator", css=custom_css) as demo:
 
     # Clear button functionality
     clear_button.click(
-        fn=lambda: (None, "Plumbing", "", "", "", "Planning", "", None, None, None),
+        fn=lambda: (None, "Plumbing", "", "", "", "Planning", "", None, None, None, "", ""),
         outputs=[
             file_input, category_input, material_cost_input, labor_index_input,
-            scope_change_input, project_phase_input, output_text, output_chart, output_pdf, output_excel
+            scope_change_input, project_phase_input, output_text, bar_chart_output,
+            pie_chart_output, output_pdf, output_excel, alert_output, dashboard_output
         ]
     )
 
@@ -383,7 +449,7 @@ with gr.Blocks(title="Budget Overrun Risk Estimator", css=custom_css) as demo:
     submit_button.click(
         fn=predict_risk,
         inputs=[file_input, category_input, material_cost_input, labor_index_input, scope_change_input, project_phase_input],
-        outputs=[output_text, output_chart, output_pdf, output_excel]
+        outputs=[output_text, bar_chart_output, pie_chart_output, output_pdf, output_excel, alert_output, dashboard_output]
     )
 
 # Launch the app