Update app.py

app.py

@@ -152,7 +152,10 @@ async def validate_user(request: Request, username: str = Form(...), password: s
     status,role = verify_user(username, password)
     if status == 'success':
         logging.info(f"user role {role}is rerturned")
-        return templates.TemplateResponse("dashboard.html", {"request": request, "username": username,"role": role})
+        response = RedirectResponse(url="/dashboard", status_code=302)
+        response.set_cookie(key="role", value=role)
+        response.set_cookie(key="username", value=username)
+        return response
        
     else:
         return templates.TemplateResponse("index.html", {"request": request})
@@ -200,7 +203,32 @@ async def get_companies():
         raise HTTPException(status_code=500, detail="Internal Server Error")
 @app.get("/dashboard")
 async def dashboard(request: Request):
-    return templates.TemplateResponse("dashboard.html", {"request": request,"title":"Dashboard"})
+    try:
+        cnx = get_db_connection()
+        cursor = cnx.cursor()
+
+        cursor.execute("show tables")
+        all_tables = cursor.fetchall()
+        table_count_of_each_table = {}
+        # Fetch count of each table in the database
+        for table in all_tables:
+            query = f"SELECT COUNT(*) FROM {table[0]}"
+            cursor.execute(query)
+            table_count_of_each_table[table[0]] = cursor.fetchone()[0]
+
+        cursor.close()
+        cnx.close() 
+        
+        logging.info(table_count_of_each_table)
+       # return templates.TemplateResponse("dashboard.html", {"request": request,"title":"Dashboard" , "table_count_of_each_table": table_count_of_each_table})
+        return templates.TemplateResponse("dashboard.html", {
+            "request": request,
+            "title": "Dashboard",
+            "table_count_of_each_table": table_count_of_each_table
+        })
+    except mysql.connector.Error as err:
+        logging.error(f"Database error: {err}")
+        raise HTTPException(status_code=500, detail="Internal Server Error")
 @app.get("/company_profile")
 async def company_profile(request: Request):
     return templates.TemplateResponse("company_profile.html", {"request": request,"title":"Company Profile"})