Update app.py

app.py

@@ -451,9 +451,9 @@ def create_empty_df():
     """Create an empty DataFrame with the required columns"""
     return pd.DataFrame(columns=[
         'ID', 'Title', 'Description', 'Assignee', 'Status', 
-        'Date Started', 'Date to Finish', 'Due Date'  # Added new date fields
+        'Date Started', 'Date to Finish', 'Due Date', 'Documents'  # Added Documents field
     ])
-
+    
 def load_tasks():
     """Load tasks from JSON file with better error handling"""
     # First check if we have in-memory tasks from a failed save
@@ -873,19 +873,119 @@ if selected_task_id:
             </div>
         """, unsafe_allow_html=True)
         
-        # Description section
+        # Description section - Fix formatting
         st.markdown('<div class="section-divider"></div>', unsafe_allow_html=True)
         st.markdown('<div class="task-details-section">', unsafe_allow_html=True)
         st.markdown('<h3>Description</h3>', unsafe_allow_html=True)
         st.markdown('<div class="task-description">', unsafe_allow_html=True)
-        st.markdown(f"{task['Description'] if task['Description'] else 'No description provided.'}", unsafe_allow_html=True)
+        # Ensure proper encoding and sanitization of description text
+        description_text = task['Description'] if task['Description'] else 'No description provided.'
+        st.markdown(f"<p>{description_text}</p>", unsafe_allow_html=True)
         st.markdown('</div>', unsafe_allow_html=True)
         st.markdown('</div>', unsafe_allow_html=True)
         
-        # Actions section
+        # Task actions
         st.markdown('<div class="section-divider"></div>', unsafe_allow_html=True)
         st.markdown('<div class="task-details-section">', unsafe_allow_html=True)
-        st.markdown('<h3>Actions</h3>', unsafe_allow_html=True)
+        st.markdown('<h3>Edit Task</h3>', unsafe_allow_html=True)
+
+        # Edit description
+        new_description = st.text_area("Edit Description", 
+                              value=task['Description'] if task['Description'] else '',
+                              height=150,
+                              key=f"edit_desc_{selected_task_id}")
+
+        # File upload section
+        uploaded_files = st.file_uploader("Upload Documents", 
+                                 accept_multiple_files=True,
+                                 key=f"upload_{selected_task_id}")
+
+        # Save changes button for description
+        if st.button("Save Description", key=f"save_desc_{selected_task_id}"):
+            task_idx = task_df.index[0]
+    
+            # Only update if description changed
+            if new_description != task['Description']:
+                update_task(task_idx, 'Description', new_description)
+                st.success("Description updated successfully!")
+                st.rerun()
+
+        # Handle file uploads
+        if uploaded_files:
+            # Create a container to show upload status
+            upload_status = st.empty()
+    
+            # Process each uploaded file
+            for uploaded_file in uploaded_files:
+                # Display upload progress
+                upload_status.info(f"Processing {uploaded_file.name}...")
+        
+                try:
+                    # Read file content
+                    file_content = uploaded_file.read()
+            
+                    # Create documents directory if it doesn't exist
+                    doc_dir = os.path.join(os.path.dirname(TASKS_FILE), "task_documents")
+                    os.makedirs(doc_dir, exist_ok=True)
+            
+                    # Create task-specific directory
+                    task_doc_dir = os.path.join(doc_dir, f"task_{selected_task_id.replace('#', '')}")
+                    os.makedirs(task_doc_dir, exist_ok=True)
+            
+                    # Save file
+                    file_path = os.path.join(task_doc_dir, uploaded_file.name)
+                    with open(file_path, "wb") as f:
+                        f.write(file_content)
+            
+                    # Update task with document reference (optional)
+                    if 'Documents' not in task or not isinstance(task['Documents'], list):
+                        task_df.at[task_df.index[0], 'Documents'] = [uploaded_file.name]
+                    else:
+                        docs = task['Documents']
+                        if uploaded_file.name not in docs:
+                            docs.append(uploaded_file.name)
+                            task_df.at[task_df.index[0], 'Documents'] = docs
+            
+                    # Save the updated task data
+                    st.session_state.tasks.loc[task_df.index[0]] = task_df.iloc[0]
+                    save_tasks()
+            
+                    upload_status.success(f"Uploaded {uploaded_file.name} successfully!")
+                except Exception as e:
+                    upload_status.error(f"Error uploading {uploaded_file.name}: {str(e)}")
+    
+            # Clear status after all files processed
+            st.success("All files processed. Refresh to see attached documents.")
+
+        # Display attached documents if any
+        if 'Documents' in task and isinstance(task['Documents'], list) and len(task['Documents']) > 0:
+            st.markdown('<div class="section-divider"></div>', unsafe_allow_html=True)
+            st.markdown('<div class="task-details-section">', unsafe_allow_html=True)
+            st.markdown('<h3>Attached Documents</h3>', unsafe_allow_html=True)
+    
+            for doc in task['Documents']:
+                # Create the file path
+                doc_dir = os.path.join(os.path.dirname(TASKS_FILE), "task_documents")
+                task_doc_dir = os.path.join(doc_dir, f"task_{selected_task_id.replace('#', '')}")
+                file_path = os.path.join(task_doc_dir, doc)
+        
+                # Display document with download option
+                if os.path.exists(file_path):
+                    col1, col2 = st.columns([3, 1])
+                    with col1:
+                        st.markdown(f"📄 **{doc}**")
+                    with col2:
+                        with open(file_path, "rb") as file:
+                            st.download_button(
+                                label="Download",
+                                data=file,
+                                file_name=doc,
+                                key=f"download_{doc}_{selected_task_id}"
+                            )
+    
+            st.markdown('</div>', unsafe_allow_html=True)
+
+        st.markdown('</div>', unsafe_allow_html=True)
         
         # Create columns for actions
         col1, col2 = st.columns(2)