Update app.py

app.py

@@ -4,8 +4,6 @@ from firebase_admin import credentials, firestore, auth
 import pandas as pd
 import plotly.express as px
 from datetime import datetime
-import hashlib
-from fpdf import FPDF
 
 # Initialize Firebase
 if not firebase_admin._apps:
@@ -34,34 +32,6 @@ def authenticate_user(email, password):
     except Exception:
         return False
 
-# Function to generate PDF with tabular format
-def generate_pdf(df):
-    pdf = FPDF()
-    pdf.set_auto_page_break(auto=True, margin=15)
-    pdf.add_page()
-    pdf.set_font("Arial", size=12)
-    pdf.cell(200, 10, "Task Status Report", ln=True, align='C')
-    pdf.ln(10)
-    
-    pdf.set_font("Arial", style='B', size=10)
-    col_widths = [50, 40, 50, 30, 30]
-    headers = ["Task", "Type", "Project", "Status", "Date"]
-    for i in range(len(headers)):
-        pdf.cell(col_widths[i], 10, headers[i], border=1, align='C')
-    pdf.ln()
-    pdf.set_font("Arial", size=10)
-    
-    for _, row in df.iterrows():
-        pdf.cell(col_widths[0], 10, row['task'], border=1)
-        pdf.cell(col_widths[1], 10, row['type'], border=1)
-        pdf.cell(col_widths[2], 10, row['project'], border=1)
-        pdf.cell(col_widths[3], 10, row['status'], border=1)
-        pdf.cell(col_widths[4], 10, row['date'], border=1)
-        pdf.ln()
-    
-    # Return PDF as bytes (critical fix)
-    return pdf.output(dest="S").encode("latin1")
-
 # UI Layout
 st.set_page_config(page_title="Construction To-Do List", layout="wide")
 
@@ -93,13 +63,11 @@ if not st.session_state["authenticated"]:
 
 # Main Dashboard After Login
 st.sidebar.title(f"Welcome, {st.session_state['email']}")
-menu = st.sidebar.radio("Go to", ["Dashboard View", "Task Entry", "View Tasks by Status", "Add New Project"])
+menu = st.sidebar.radio("Go to", ["Dashboard View", "Task Entry", "View Tasks", "Add New Project", "Edit Task"])
 email = st.session_state['email']
 
 if menu == "Dashboard View":
     st.title("📊 Dashboard Overview")
-    
-    # Fetch tasks for the logged-in user
     tasks_ref = db.collection("tasks").where("user", "==", email)
     tasks = [task.to_dict() for task in tasks_ref.stream()]
     
@@ -107,29 +75,17 @@ if menu == "Dashboard View":
         st.info("No tasks found.")
     else:
         df = pd.DataFrame(tasks)
-        
-        # Key Stats
         total_tasks = len(df)
         completed_tasks = len(df[df['status'] == "Completed"])
-        pending_tasks = len(df[df['status'] == "Pending"])
         
-        col1, col2, col3 = st.columns(3)
+        col1, col2 = st.columns(2)
         col1.metric("Total Tasks", total_tasks)
         col2.metric("Completed Tasks", completed_tasks)
-        col3.metric("Pending Tasks", pending_tasks)
         
-        # Pie Chart: Tasks by Status
         st.subheader("Tasks by Status")
         status_counts = df['status'].value_counts().reset_index()
-        status_counts.columns = ['Status', 'Count']
-        fig = px.pie(status_counts, values='Count', names='Status', hole=0.3)
+        fig = px.pie(status_counts, values='count', names='status', hole=0.3)
         st.plotly_chart(fig)
-        
-        # Tasks by Project (Optional)
-        st.subheader("Tasks by Project")
-        project_counts = df['project'].value_counts().reset_index()
-        project_counts.columns = ['Project', 'Count']
-        st.dataframe(project_counts)
 
 elif menu == "Task Entry":
     st.title("📝 Add New Task")
@@ -140,16 +96,15 @@ elif menu == "Task Entry":
     with st.form("task_form"):
         task = st.text_input("Task Description:")
         task_type = st.selectbox("Task Type:", ["Design", "Procurement", "Construction", "Testing", "Other"])
-        selected_project = st.selectbox("Project Name:", projects, key="select_project")
+        selected_project = st.selectbox("Project Name:", projects)
         if selected_project == "Add New Project":
-            new_project = st.text_input("Enter New Project Name:", key="new_project")
+            new_project = st.text_input("Enter New Project Name:")
             if new_project:
                 db.collection("projects").document(new_project).set({"created_by": email})
                 selected_project = new_project
         status = st.selectbox("Status:", ["Pending", "In Progress", "Completed"])
         date = st.date_input("Task Date:")
-        submit = st.form_submit_button("Add Task")
-        if submit and task:
+        if st.form_submit_button("Add Task"):
             db.collection("tasks").add({
                 "user": email,
                 "task": task,
@@ -165,27 +120,52 @@ elif menu == "Add New Project":
     st.title("🏗️ Add New Project")
     with st.form("project_form"):
         new_project_name = st.text_input("Enter New Project Name:")
-        submit_project = st.form_submit_button("Add Project")
-        if submit_project and new_project_name:
+        if st.form_submit_button("Add Project"):
             db.collection("projects").document(new_project_name).set({"created_by": email})
             st.success("Project Added Successfully!")
             st.rerun()
 
-elif menu == "View Tasks by Status":
-    st.subheader("📂 View Tasks by Status")
-    statuses = ["Pending", "In Progress", "Completed"]
-    for status in statuses:
-        st.subheader(f"📌 {status} Tasks")
-        tasks_ref = db.collection("tasks").where("user", "==", email).where("status", "==", status)
-        tasks = [task.to_dict() for task in tasks_ref.stream()]
-        if tasks:
-            df = pd.DataFrame(tasks)
-            st.dataframe(df[['task', 'type', 'project', 'status', 'date']].rename(columns={
-                'task': 'Task', 'type': 'Type', 'project': 'Project', 'status': 'Status', 'date': 'Date'
-            }))
-            
-            # Button to download PDF (error fixed)
-            pdf_data = generate_pdf(df)
-            st.download_button(label="Download Report", data=pdf_data, file_name="task_report.pdf", mime="application/pdf")
-        else:
-            st.info("No tasks found.")
+elif menu == "View Tasks":
+    st.subheader("📋 All Tasks")
+    tasks_ref = db.collection("tasks").where("user", "==", email)
+    tasks = [task.to_dict() for task in tasks_ref.stream()]
+    
+    if tasks:
+        df = pd.DataFrame(tasks)
+        st.dataframe(df[['task', 'type', 'project', 'status', 'date']])
+    else:
+        st.info("No tasks found.")
+
+elif menu == "Edit Task":
+    st.title("✏️ Edit Task")
+    tasks_ref = db.collection("tasks").where("user", "==", email)
+    tasks = [{"id": task.id, **task.to_dict()} for task in tasks_ref.stream()]
+    
+    if not tasks:
+        st.info("No tasks to edit")
+    else:
+        task_options = [f"{task['task']} ({task['id']})" for task in tasks]
+        selected_task = st.selectbox("Select Task to Edit", task_options)
+        task_id = selected_task.split("(")[-1].strip(")")
+        
+        # Find the selected task
+        task_to_edit = next((t for t in tasks if t['id'] == task_id), None)
+        
+        if task_to_edit:
+            with st.form("edit_form"):
+                new_task = st.text_input("Task Description", value=task_to_edit['task'])
+                new_type = st.selectbox("Task Type", ["Design", "Procurement", "Construction", "Testing", "Other"], 
+                                      index=["Design", "Procurement", "Construction", "Testing", "Other"].index(task_to_edit['type']))
+                new_status = st.selectbox("Status", ["Pending", "In Progress", "Completed"], 
+                                       index=["Pending", "In Progress", "Completed"].index(task_to_edit['status']))
+                new_date = st.date_input("Date", value=datetime.strptime(task_to_edit['date'], "%Y-%m-%d"))
+                
+                if st.form_submit_button("Update Task"):
+                    db.collection("tasks").document(task_id).update({
+                        "task": new_task,
+                        "type": new_type,
+                        "status": new_status,
+                        "date": str(new_date)
+                    })
+                    st.success("Task updated successfully!")
+                    st.rerun()