Update app.py

app.py

@@ -1,4 +1,5 @@
 import streamlit as st
+import streamlit.components.v1 as components
 import pandas as pd
 import matplotlib.pyplot as plt
 import os
@@ -179,7 +180,7 @@ with st.form("project_form"):
     col1, col2 = st.columns(2)
     
     with col1:
-        project_name = st.text_input("Project Name", value="Highway Expansion")
+        project_name = st.text_input("Project Name")
         phase = st.selectbox("Phase", [""] + ["Planning", "Design", "Construction"], index=0, key="phase_select")
         
         if phase != st.session_state.phase:
@@ -187,21 +188,20 @@ with st.form("project_form"):
             st.session_state.task = ""
         task_options_list = [""] + task_options.get(phase, []) if phase else [""]
         task = st.selectbox("Task", task_options_list, index=0, key="task_select")
-        current_progress = st.number_input("Current Progress (%)", min_value=0.0, max_value=100.0, step=1.0, value=70.0)
-        task_expected_duration = st.number_input("Task Expected Duration (days)", min_value=0, step=1, value=10)
-        task_actual_duration = st.number_input("Task Actual Duration (days)", min_value=0, step=1, value=12)
+        current_progress = st.number_input("Current Progress (%)", min_value=0.0, max_value=100.0, step=1.0, value=0.0)
+        task_expected_duration = st.number_input("Task Expected Duration (days)", min_value=0, step=1, value=0)
+        task_actual_duration = st.number_input("Task Actual Duration (days)", min_value=0, step=1, value=0)
     
     with col2:
-        workforce_gap = st.number_input("Workforce Gap (%)", min_value=0.0, max_value=100.0, step=1.0, value=20.0)
-        workforce_skill_level = st.selectbox("Workforce Skill Level", ["Low", "Medium", "High"], index=1)
-        workforce_shift_hours = st.number_input("Workforce Shift Hours", min_value=0, step=1, value=6)
+        workforce_gap = st.number_input("Workforce Gap (%)", min_value=0.0, max_value=100.0, step=1.0, value=0.0)
+        workforce_skill_level = st.selectbox("Workforce Skill Level", ["", "Low", "Medium", "High"], index=0)
+        workforce_shift_hours = st.number_input("Workforce Shift Hours", min_value=0, step=1, value=0)
         st.write(f"**Selected Shift Hours**: {workforce_shift_hours}")
-        weather_impact_score = st.number_input("Weather Impact Score (0-100)", min_value=0, max_value=100, step=1, value=60)
+        weather_impact_score = st.number_input("Weather Impact Score (0-100)", min_value=0, max_value=100, step=1, value=0)
         weather_condition = get_weather_condition(weather_impact_score)
         st.write(f"**Weather Condition**: {weather_condition}")
-        weather_forecast_date = st.date_input("Weather Forecast Date", min_value=datetime(2025, 1, 1), value=datetime(2025, 5, 23))
+        weather_forecast_date = st.date_input("Weather Forecast Date", min_value=datetime(2025, 1, 1), value=None)
     
-    st.info("Note: AI insights are generated using DistilBART, which may take ~5-10 seconds on CPU.")
     submit_button = st.form_submit_button("Predict Delay")
 
 # Process form submission
@@ -219,7 +219,7 @@ if submit_button:
         "workforce_shift_hours": workforce_shift_hours,
         "weather_impact_score": weather_impact_score,
         "weather_condition": weather_condition,
-        "weather_forecast_date": weather_forecast_date.strftime("%Y-%m-%d")
+        "weather_forecast_date": weather_forecast_date.strftime("%Y-%m-%d") if weather_forecast_date else ""
     }
     
     error = validate_inputs(input_data)
@@ -257,7 +257,7 @@ if submit_button:
                 new Chart(ctx, {json.dumps(chart_config)});
             </script>
             """
-            st.markdown(chart_html, unsafe_allow_html=True)
+            components.html(chart_html, height=250)
             logger.info("Chart.js heatmap rendered")
             
             # Generate matplotlib figure for PDF