Update app.py

app.py

@@ -334,36 +334,54 @@ def save_to_salesforce(input_data, prediction, pdf_buffer):
         logger.error(f"Error saving to Salesforce: {str(e)}")
         return f"Error saving to Salesforce: {str(e)}"
 
-# Input form
-with st.form("project_form"):
-    col1, col2 = st.columns(2)
-    
-    with col1:
-        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:
-            st.session_state.phase = phase
-            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=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)
+# Input section
+st.markdown("### Project Details")
+col1, col2 = st.columns([1, 1])  # Equal width columns for better alignment
+
+with col1:
+    project_name = st.text_input("Project Name", help="Enter the name of the project")
+    phase = st.selectbox(
+        "Phase",
+        [""] + ["Planning", "Design", "Construction"],
+        index=0 if st.session_state.phase == "" else ["", "Planning", "Design", "Construction"].index(st.session_state.phase),
+        key="phase_select",
+        help="Select the project phase"
+    )
     
-    with col2:
-        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}")
-        project_location = st.text_input("Project Location (City)", placeholder="e.g., New York")
-        weather_forecast_date = st.date_input("Weather Forecast Date", min_value=datetime(2025, 1, 1), value=None)
+    # Update task options when phase changes
+    if phase != st.session_state.phase:
+        st.session_state.phase = phase
+        st.session_state.task = ""  # Reset task when phase changes
+        logger.info(f"Phase changed to {phase}, resetting task")
     
-    submit_button = st.form_submit_button("Fetch Weather and Predict Delay")
-
-# Process form submission
-if submit_button:
-    logger.info("Processing form submission")
+    task_options_list = [""] + task_options.get(phase, []) if phase else [""]
+    logger.info(f"Task options for phase '{phase}': {task_options_list}")
+    task = st.selectbox(
+        "Task",
+        task_options_list,
+        index=0 if st.session_state.task == "" else task_options_list.index(st.session_state.task) if st.session_state.task in task_options_list else 0,
+        key="task_select",
+        help="Select the task corresponding to the phase"
+    )
+    st.session_state.task = task
+    current_progress = st.number_input("Current Progress (%)", min_value=0.0, max_value=100.0, step=1.0, value=0.0, help="Enter the current progress percentage")
+    task_expected_duration = st.number_input("Task Expected Duration (days)", min_value=0, step=1, value=0, help="Enter the expected duration in days")
+    task_actual_duration = st.number_input("Task Actual Duration (days)", min_value=0, step=1, value=0, help="Enter the actual duration in days")
+
+with col2:
+    workforce_gap = st.number_input("Workforce Gap (%)", min_value=0.0, max_value=100.0, step=1.0, value=0.0, help="Enter the workforce gap percentage")
+    workforce_skill_level = st.selectbox("Workforce Skill Level", ["", "Low", "Medium", "High"], index=0, help="Select the workforce skill level")
+    workforce_shift_hours = st.number_input("Workforce Shift Hours", min_value=0, step=1, value=0, help="Enter the shift hours")
+    st.write(f"**Selected Shift Hours**: {workforce_shift_hours}")
+    project_location = st.text_input("Project Location (City)", placeholder="e.g., New York", help="Enter the city for weather data")
+    weather_forecast_date = st.date_input("Weather Forecast Date", min_value=datetime(2025, 1, 1), value=None, help="Select the forecast date")
+
+# Predict button
+predict_button = st.button("Fetch Weather and Predict Delay")
+
+# Process inputs when button is clicked
+if predict_button:
+    logger.info("Processing prediction request")
     input_data = {
         "project_name": project_name,
         "phase": phase,