Update app.py

app.py

@@ -198,7 +198,6 @@ def generate_pdf(input_data, prediction, heatmap_fig, gantt_fig):
         f"Task Actual Duration: {input_data['task_actual_duration']} days",
         f"Workforce Gap: {input_data['workforce_gap']}%",
         f"Workforce Skill Level: {input_data['workforce_skill_level']}",
-        f"Workforce Shift Hours: {input_data['workforce_shift_hours']}",
         f"Weather Impact Score: {input_data['weather_impact_score']}",
         f"Weather Condition: {input_data['weather_condition']}",
         f"Weather Forecast Date: {input_data['weather_forecast_date']}",
@@ -254,7 +253,7 @@ def generate_pdf(input_data, prediction, heatmap_fig, gantt_fig):
     buffer.seek(0)
     return buffer
 
-# Function to save data to Salesforce, including PDF and Status__c
+# Function to save data to Salesforce, including PDF
 def save_to_salesforce(input_data, prediction, pdf_buffer):
     if sf is None:
         return "Salesforce connection not established."
@@ -334,42 +333,67 @@ 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)}"
 
-# Phase and Task selection (outside the form for immediate updates)
-col1, col2 = st.columns(2)
-with col1:
-    phase = st.selectbox("Phase", [""] + ["Planning", "Design", "Construction"], index=0, key="phase_select")
-    
-    # Update task options and reset task when phase changes
-    if phase != st.session_state.get("phase", ""):
-        st.session_state.phase = phase
-        st.session_state.task = ""
-        if "task_select" in st.session_state:
-            del st.session_state.task_select
-        logger.info(f"Phase changed to {phase}, resetting task and task_select")
-    
-    # Define task options based on phase
-    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, key="task_select")
-
-# Input form for other fields
+# Input form with improved UI and dynamic Task field
 with st.form("project_form"):
-    col1, col2 = st.columns(2)
+    # Project Details Section
+    st.markdown("### Project Details")
+    col1, col2 = st.columns([1, 1])
     
     with col1:
-        project_name = st.text_input("Project Name")
+        project_name = st.text_input("Project Name", placeholder="Enter project name")
+        phase = st.selectbox("Phase", [""] + ["Planning", "Design", "Construction"], index=0, key="phase_select")
+        
+        # Update session state when phase changes
+        if phase != st.session_state.get("phase", ""):
+            st.session_state.phase = phase
+            st.session_state.task = ""
+            if "task_select" in st.session_state:
+                del st.session_state.task_select
+            logger.info(f"Phase changed to {phase}, resetting task and task_select")
+        
+        # Define task options based on phase
+        task_options_list = [""] + task_options.get(phase, []) if phase else [""]
+        logger.info(f"Task options for phase '{phase}': {task_options_list}")
+        
+        # Use a placeholder to dynamically update the Task selectbox
+        task_placeholder = st.empty()
+        task = task_placeholder.selectbox("Task", task_options_list, index=0, key="task_select")
+    
+    with col2:
         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:
+    # Divider
+    st.markdown("---")
+    
+    # Workforce Details Section
+    st.markdown("### Workforce Details")
+    col3, col4 = st.columns([1, 1])
+    
+    with col3:
         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)
+    
+    with col4:
         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}")
+    
+    # Divider
+    st.markdown("---")
+    
+    # Location and Weather Section
+    st.markdown("### Location and Weather")
+    col5, col6 = st.columns([1, 1])
+    
+    with col5:
         project_location = st.text_input("Project Location (City)", placeholder="e.g., New York")
+    
+    with col6:
         weather_forecast_date = st.date_input("Weather Forecast Date", min_value=datetime(2025, 1, 1), value=None)
     
+    # Submit Button
+    st.write("")  # Add some spacing
     submit_button = st.form_submit_button("Fetch Weather and Predict Delay")
 
 # Process form submission