Update app.py

app.py

@@ -516,72 +516,72 @@ def main():
 
     # Edit Tasks
     elif menu == "✏️ Edit Tasks":
-    tasks = get_tasks()
-    if tasks:
-        task_options = [f"{t['task']} ({t['project']})" for t in tasks]
-        selected = st.selectbox("Select Task", task_options)
-        task = next(t for t in tasks if f"{t['task']} ({t['project']})" == selected)
-        
-        with st.form(key="edit_form"):
-            col1, col2 = st.columns(2)
-            
-            # Left Column
-            with col1:
-                new_task = st.text_area("Description", value=task['task'], height=100)
-                new_type = st.selectbox(
-                    "Type", 
-                    ["Design", "Procurement", "Construction", "Testing", "Other"],
-                    index=["Design", "Procurement", "Construction", "Testing", "Other"].index(task['type'])
-                )
+        tasks = get_tasks()
+        if tasks:
+            task_options = [f"{t['task']} ({t['project']})" for t in tasks]
+            selected = st.selectbox("Select Task", task_options)
+            task = next(t for t in tasks if f"{t['task']} ({t['project']})" == selected)
             
-            # Right Column
-            with col2:
-                projects = [p.id for p in db.collection("projects").stream()] + ["Add New Project"]
-                new_project = st.selectbox(
-                    "Project", 
-                    projects, 
-                    index=projects.index(task['project']) if task['project'] in projects else 0
-                )
-                if new_project == "Add New Project":
-                    new_project = st.text_input("New Project Name")
+            with st.form(key="edit_form"):
+                col1, col2 = st.columns(2)
                 
-                new_status = st.selectbox(
-                    "Status", 
-                    ["Pending", "In Progress", "Completed"], 
-                    index=["Pending", "In Progress", "Completed"].index(task['status'])
-                )
-                new_date = st.date_input(
-                    "Due Date", 
-                    value=datetime.strptime(task['date'], "%Y-%m-%d").date()
-                )
-            
-            # Attachment Management
-            st.subheader("📎 Attachments")
-            attachments = task.get('attachments', [])  # Safely get attachments
-            if attachments:
-                for att in attachments:
-                    col1, col2 = st.columns([4,1])
-                    with col1:
-                        st.markdown(f"""
-                        <div style="margin:0.5rem 0; padding:0.5rem; border-radius:5px; background:#f0f2f6;">
-                            <div style="display:flex; align-items:center; gap:1rem;">
-                                <span style="flex:1;">📄 {att['name']}</span>
-                                <a href="{att['url']}" target="_blank" style="text-decoration:none;">
-                                    🔗 View
-                                </a>
+                # Left Column
+                with col1:
+                    new_task = st.text_area("Description", value=task['task'], height=100)
+                    new_type = st.selectbox(
+                        "Type", 
+                        ["Design", "Procurement", "Construction", "Testing", "Other"],
+                        index=["Design", "Procurement", "Construction", "Testing", "Other"].index(task['type'])
+                    )
+                
+                # Right Column
+                with col2:
+                    projects = [p.id for p in db.collection("projects").stream()] + ["Add New Project"]
+                    new_project = st.selectbox(
+                        "Project", 
+                        projects, 
+                        index=projects.index(task['project']) if task['project'] in projects else 0
+                    )
+                    if new_project == "Add New Project":
+                        new_project = st.text_input("New Project Name")
+                    
+                    new_status = st.selectbox(
+                        "Status", 
+                        ["Pending", "In Progress", "Completed"], 
+                        index=["Pending", "In Progress", "Completed"].index(task['status'])
+                    )
+                    new_date = st.date_input(
+                        "Due Date", 
+                        value=datetime.strptime(task['date'], "%Y-%m-%d").date()
+                    )
+                
+                # Attachment Management
+                st.subheader("📎 Attachments")
+                attachments = task.get('attachments', [])  # Safely get attachments
+                if attachments:
+                    for att in attachments:
+                        col1, col2 = st.columns([4,1])
+                        with col1:
+                            st.markdown(f"""
+                            <div style="margin:0.5rem 0; padding:0.5rem; border-radius:5px; background:#f0f2f6;">
+                                <div style="display:flex; align-items:center; gap:1rem;">
+                                    <span style="flex:1;">📄 {att['name']}</span>
+                                    <a href="{att['url']}" target="_blank" style="text-decoration:none;">
+                                        🔗 View
+                                    </a>
+                                </div>
+                                <small style="color:#666;">{att['type']} | {att['size'] // 1024} KB</small>
                             </div>
-                            <small style="color:#666;">{att['type']} | {att['size'] // 1024} KB</small>
-                        </div>
-                        """, unsafe_allow_html=True)
-                    with col2:
-                        if st.button("🗑️", key=f"del_att_{att['name']}"):
-                            # Remove attachment from task
-                            updated_attachments = [a for a in attachments if a['name'] != att['name']]
-                            db.collection("tasks").document(task['id']).update({
-                                "attachments": updated_attachments
-                            })
-                            st.rerun()
-            
+                            """, unsafe_allow_html=True)
+                        with col2:
+                            if st.button("🗑️", key=f"del_att_{att['name']}"):
+                                # Remove attachment from task
+                                updated_attachments = [a for a in attachments if a['name'] != att['name']]
+                                db.collection("tasks").document(task['id']).update({
+                                    "attachments": updated_attachments
+                                })
+                                st.rerun()
+                
             # New Attachments
             new_attachments = st.file_uploader(
                 "Add New Attachments",
@@ -591,7 +591,7 @@ def main():
             )
             
             # Submit Button
-            submitted = st.form_submit_button("💾 Save Changes")
+                    submitted = st.form_submit_button("💾 Save Changes")
             
             if submitted:
                 try: