feat(classes): optimize teacher dashboard ui and implement grade management

src/app/(dashboard)/student/learning/assignments/[assignmentId]/page.tsx

@@ -2,6 +2,7 @@ import { notFound } from "next/navigation"
 
 import { getDemoStudentUser, getStudentHomeworkTakeData } from "@/modules/homework/data-access"
 import { HomeworkTakeView } from "@/modules/homework/components/homework-take-view"
+import { HomeworkReviewView } from "@/modules/homework/components/student-homework-review-view"
 import { formatDate } from "@/shared/lib/utils"
 
 export const dynamic = "force-dynamic"
@@ -18,6 +19,23 @@ export default async function StudentAssignmentTakePage({
   const data = await getStudentHomeworkTakeData(assignmentId, student.id)
   if (!data) return notFound()
 
+  // If status is graded or submitted, use the review view
+  const status = data.submission?.status
+  if (status === "graded" || status === "submitted") {
+    return (
+      <div className="flex h-full flex-col space-y-4 p-6">
+        <div className="flex flex-col gap-1">
+          <h2 className="text-2xl font-bold tracking-tight">{data.assignment.title}</h2>
+          <div className="text-sm text-muted-foreground">
+             <span>Due: {data.assignment.dueAt ? formatDate(data.assignment.dueAt) : "-"}</span>
+          </div>
+        </div>
+
+        <HomeworkReviewView initialData={data} />
+      </div>
+    )
+  }
+
   return (
     <div className="flex h-full flex-col space-y-4 p-6">
       <div className="flex flex-col gap-1">