Update app.py

app.py

@@ -1,4 +1,7 @@
 import streamlit as st
+from datetime import datetime
+import pandas as pd
+import plotly.express as px
 
 # Configure page first
 st.set_page_config(
@@ -8,7 +11,7 @@ st.set_page_config(
     menu_items={
         'Get Help': 'https://github.com',
         'Report a bug': "https://github.com",
-        'About': "# Task Management System v3.2"
+        'About': "# Task Management System v3.3"
     }
 )
 
@@ -21,7 +24,6 @@ st.markdown("""
         max-width: 1400px;
     }
     
-    /* Gradient Header */
     .header {
         background: linear-gradient(135deg, #2c3e50, #3498db);
         color: white;
@@ -31,7 +33,6 @@ st.markdown("""
         box-shadow: 0 4px 6px rgba(0,0,0,0.1);
     }
     
-    /* Status Badges */
     .status-badge {
         padding: 6px 14px;
         border-radius: 20px;
@@ -44,20 +45,6 @@ st.markdown("""
     .pending { background: #fff3cd; color: #856404; }
     .in-progress { background: #cce5ff; color: #004085; }
     .completed { background: #d4edda; color: #155724; }
-    
-    /* Progress Bars */
-    .progress-container {
-        background: #f0f0f0;
-        border-radius: 10px;
-        height: 10px;
-        overflow: hidden;
-    }
-    
-    .progress-bar {
-        height: 100%;
-        background: #3498db;
-        transition: width 0.3s ease;
-    }
 </style>
 """, unsafe_allow_html=True)
 
@@ -87,7 +74,7 @@ def create_default_admin(auth):
         st.error(f"Admin setup error: {str(e)}")
 
 def main():
-    # Initialize Firebase within Streamlit context
+    # Initialize Firebase
     db, auth = initialize_firebase()
     create_default_admin(auth)
 
@@ -114,12 +101,14 @@ def main():
                     if st.form_submit_button("🚀 Login"):
                         try:
                             user = auth.get_user_by_email(email)
-                            user_doc = db.collection("users").document(user.uid).get().to_dict()
+                            user_doc = db.collection("users").document(user.uid).get()
+                            custom_claims = user.custom_claims or {}
+                            
                             st.session_state.update({
                                 "authenticated": True,
                                 "email": user.email,
                                 "user_uid": user.uid,
-                                "is_admin": user.custom_claims.get('admin', False) if user.custom_claims else user_doc.get('is_admin', False)
+                                "is_admin": custom_claims.get('admin', False) or user_doc.get('is_admin') or False
                             })
                             st.rerun()
                         except Exception as e:
@@ -143,7 +132,7 @@ def main():
                             st.error(f"⚠️ Registration failed: {str(e)}")
         return
 
-    # Authenticated User Flow
+    # Main Application
     st.markdown(f"""
     <div class='header'>
         <h1>📋 To-Do List Management System</h1>
@@ -341,7 +330,7 @@ def main():
         else:
             st.info("📭 No tasks to edit")
 
-    # Project Management (Admin Only)
+    # Project Management
     elif menu == "🏗️ Project Management" and st.session_state.is_admin:
         st.subheader("🏗️ Project Management")
         projects = [p.id for p in db.collection("projects").stream()]