img server

server/api/drawing/[id].ts

@@ -2,7 +2,11 @@ export default defineEventHandler(async event => {
 
     // 获取任务(普通用户只能看到自己的)
     if (event.node.req.method === 'GET') {
-        return await useStorage().getItem(`task:${event.context.params.id}`)
+        const task = await useStorage().getItem(`task:${event.context.params.id}`)
+        task.data = task.data.map((image_path: any) => {
+            return '/api/img/' + image_path.match(/[^\/]+$/)[0]
+        })
+        return task
     }
 
     // 修改任务(只允许BOT写入)

server/api/img/[id].ts

@@ -0,0 +1,14 @@
+import fs from 'fs'
+export default defineEventHandler(async event => {
+
+    // 获取图片
+    if (event.node.req.method === 'GET') {
+        // 判断当前执行目录是否存在 outputs 文件夹, 否则在上一级文件夹
+        let path = `outputs/outputs/txt2img-samples/samples/${event.context.params.id}`
+        if (!fs.existsSync('outputs')) {
+            path = `../outputs/outputs/txt2img-samples/samples/${event.context.params.id}`
+        }
+        return await useStorage().getItem(`img:${event.context.params.id}`)
+    }
+
+})