自动生成流程图

src/admin/api.ts

@@ -68,6 +68,10 @@ export function toggleProblemVisible(problemID: number) {
   return http.put("admin/problem/visible", { id: problemID })
 }
 
+export function generateFlowchartFromPythonCode(python: string) {
+  return http.post("admin/problem/flowchart", { python })
+}
+
 export function editContestProblem(problem: AdminProblem | BlankProblem) {
   return http.put("admin/contest/problem", problem)
 }

src/admin/problem/detail.vue

@@ -14,6 +14,7 @@ import {
   createProblem,
   editContestProblem,
   editProblem,
+  generateFlowchartFromPythonCode,
   getProblem,
   uploadTestcases,
 } from "../api"
@@ -46,6 +47,8 @@ const title = computed(
     })[<string>route.name],
 )
 
+const isAIGenerating = ref(false)
+
 const problem = useLocalStorage<BlankProblem>(STORAGE_KEY.ADMIN_PROBLEM, {
   _id: "",
   title: "",
@@ -405,8 +408,14 @@ function clear() {
   location.reload()
 }
 
-function generateMermaid() {
-  message.info("还在开发中，敬请期待！")
+async function generateMermaid() {
+  isAIGenerating.value = true
+  const res = await generateFlowchartFromPythonCode(
+    problem.value.answers.filter((a) => a.language === "Python3")[0].code,
+  )
+  isAIGenerating.value = false
+  message.warning("如果渲染不成功，请复制到外部 AI 网站检查语法")
+  problem.value.mermaid_code = res.data.flowchart
 }
 
 onMounted(() => {
@@ -630,11 +639,15 @@ watch(
 
   <!-- 流程图相关设置 -->
   <n-form inline label-placement="left" :show-feedback="false">
-    <n-form-item label="根据上面的答案智能生成 Mermaid 代码">
+    <n-form-item label="根据上面的【Python答案】智能生成 Mermaid 代码">
       <n-button
         type="primary"
         size="small"
-        :disabled="!problem.answers.filter((a) => a.code).length"
+        :disabled="
+          !problem.answers.filter((a) => a.language === 'Python3')[0].code
+            .length
+        "
+        :loading="isAIGenerating"
         @click="generateMermaid"
       >
         AI 生成

src/admin/user/list.vue

@@ -295,9 +295,15 @@ watch(() => [query.page, query.limit, query.type, query.orderBy], listUsers)
         <n-form-item-gi :span="1" label="真名">
           <n-input v-model:value="userEditing.real_name" />
         </n-form-item-gi>
+        <n-form-item-gi v-if="!create" :span="1" label="班级">
+          <n-input v-model:value="userEditing.class_name" />
+        </n-form-item-gi>
         <n-form-item-gi :span="1" label="邮箱">
           <n-input v-model:value="userEditing.email" />
         </n-form-item-gi>
+        <n-form-item-gi v-if="!create" :span="1" label="类型">
+          <n-select v-model:value="userEditing.admin_type" :options="options" />
+        </n-form-item-gi>
         <n-form-item-gi
           :span="1"
           label="密码"
@@ -305,9 +311,6 @@ watch(() => [query.page, query.limit, query.type, query.orderBy], listUsers)
         >
           <n-input v-model:value="password" />
         </n-form-item-gi>
-        <n-form-item-gi v-if="!create" :span="1" label="类型">
-          <n-select v-model:value="userEditing.admin_type" :options="options" />
-        </n-form-item-gi>
         <n-form-item-gi
           v-if="!create && userEditing.admin_type === USER_TYPE.ADMIN"
           :span="1"