Update app.py

app.py

@@ -92,42 +92,18 @@ 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"])
+menu = st.sidebar.radio("Go to", ["Dashboard View", "Task Entry", "View Tasks by Status", "Add New Project"])
 email = st.session_state['email']
 
-if menu == "Dashboard View":
-    st.title("📊 Dashboard Overview")
-    tasks_ref = db.collection("tasks").where(filter=firestore.FieldFilter("user", "==", email))
-    tasks = [task.to_dict() for task in tasks_ref.stream()]
-    df = pd.DataFrame(tasks) if tasks else pd.DataFrame()
-
-    if not df.empty and "status" in df.columns:
-        col1, col2 = st.columns(2)
-        col1.metric("Total Tasks", len(df))
-        col1.metric("Pending Tasks", len(df[df['status'] == "Pending"]))
-        col2.metric("In Progress Tasks", len(df[df['status'] == "In Progress"]))
-        col2.metric("Completed Tasks", len(df[df['status'] == "Completed"]))
-
-        fig = px.pie(df, names='status', title='Task Status Distribution')
-        st.plotly_chart(fig)
-    else:
-        st.warning("No tasks found for the user.")
-
-elif menu == "Task Entry":
+if menu == "Task Entry":
     st.title("📝 Add New Task")
     projects_ref = db.collection("projects")
     projects = [proj.id for proj in projects_ref.stream()]
-    projects.append("Add New Project")
     
     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)
-        if selected_project == "Add 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")
@@ -142,3 +118,13 @@ elif menu == "Task Entry":
             })
             st.success("Task Added Successfully!")
             st.rerun()
+
+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:
+            db.collection("projects").document(new_project_name).set({"created_by": email})
+            st.success("Project Added Successfully!")
+            st.rerun()