Spaces:

gitdeem
/

sdxl

Sleeping

App Files Files Community

gitdeem commited on Sep 19, 2024

Commit

d17b633

verified ·

1 Parent(s): e85cd64

Upload 2 files

Browse files

Files changed (2) hide show

Dockerfile +3 -0
app.js +57 -57

Dockerfile CHANGED Viewed

@@ -6,4 +6,7 @@ COPY . /app
 RUN npm install
 CMD ["npm", "start"]

 RUN npm install
+RUN mkdir /app/images \
+  && chmod 777 /app/images
 CMD ["npm", "start"]

app.js CHANGED Viewed

@@ -2,17 +2,29 @@ import express from "express";
 import bodyParser from "body-parser";
 import dotenv from "dotenv";
 import fetch from "node-fetch";
 dotenv.config();
 const app = express();
 app.use(bodyParser.json());
 app.get("/", (req, res) => {
   res.send(`
     <html>
       <head>
         <title>OPENAI</title>
-      </head
       <body>
         <h1>OpenAI</h1>
         <p>Congratulations! Your project has been successfully deployed.</p>
@@ -22,7 +34,7 @@ app.get("/", (req, res) => {
 });
 app.post("/deem/v1/chat/completions", async (req, res) => {
-  let token; // 将 token 的定义提到函数的开头
   const authHeader =
     req.headers["authorization"] || req.headers["Authorization"];
   if (!authHeader) {
@@ -31,7 +43,7 @@ app.post("/deem/v1/chat/completions", async (req, res) => {
       errmsg: "Unauthorized.",
     });
   } else {
-    token = authHeader.split(" ")[1]; // 删除 const 定义
     if (!token) {
       return res.status(401).json({
         code: 401,
@@ -45,32 +57,24 @@ app.post("/deem/v1/chat/completions", async (req, res) => {
     const messages = data.messages;
     const lastMessage = messages[messages.length - 1];
     const messagesContent = lastMessage.content;
-    //const messagesContent = messages.map(m => m.content).join(' ');
-    //const resp = await fetch(`https://aihh.api.ecylt.top/?key=${token}&prompt=${messagesContent}`);
-    //const imgUrl = resp.url;//直接获取图片地址
-    //或者 imgUrl = https://aihh.api.ecylt.top/?key=${token}&prompt=${messagesContent}
-    const apiUrl = `https://aihh.api.ecylt.top/?key=${token}&prompt=${messagesContent}`
     const apiResponse = await fetch(apiUrl);
     if (!apiResponse.ok) {
       throw new Error("文生图服务失效：" + apiResponse.status);
     }
-    const imageBlob = await apiResponse.blob();
-    const formData = new FormData();
-    formData.append("file", imageBlob, "image.jpg");
-    const uploadResponse = await fetch("https://img.deem.love/upload", {
-      method: 'POST',
-      body: formData
-    });
-    if (!uploadResponse.ok) {
-      throw new Error("图片上传失败");
-    }
-    const uploadResult = await uploadResponse.json();
-    const imgUrl = "https://img.deem.love" + uploadResult[0].src;
     console.log(imgUrl);
@@ -100,15 +104,16 @@ app.post("/deem/v1/chat/completions", async (req, res) => {
       res.write("data: [DONE]\n\n");
       res.end();
     } else {
-      res.status(500).json({ error: "地址获取失效" });
     }
   } catch (error) {
     console.error("Error:", error);
   }
 });
 app.post("/deem/v1/images/generations", async (req, res) => {
-  let token; // 将 token 的定义提到函数的开头
   const authHeader =
     req.headers["authorization"] || req.headers["Authorization"];
   if (!authHeader) {
@@ -117,7 +122,7 @@ app.post("/deem/v1/images/generations", async (req, res) => {
       errmsg: "Unauthorized.",
     });
   } else {
-    token = authHeader.split(" ")[1]; // 删除 const 定义
     if (!token) {
       return res.status(401).json({
         code: 401,
@@ -130,49 +135,44 @@ app.post("/deem/v1/images/generations", async (req, res) => {
     const data = req.body;
     const messages = data.prompt;
-    const apiUrl = `https://aihh.api.ecylt.top/?key=${token}&prompt=${messages}`
     const apiResponse = await fetch(apiUrl);
     if (!apiResponse.ok) {
       throw new Error("文生图服务失效：" + apiResponse.status);
     }
-    const imageBlob = await apiResponse.blob();
-    const formData = new FormData();
-    formData.append("file", imageBlob, "image.jpg");
-    const uploadResponse = await fetch("https://img.deem.love/upload", {
-      method: 'POST',
-      body: formData
-    });
-    if (!uploadResponse.ok) {
-      throw new Error("图片上传失败");
-    }
-    const uploadResult = await uploadResponse.json();
-    const imgUrl = "https://img.deem.love" + uploadResult[0].src;
     console.log(imgUrl);
     if (imgUrl) {
       const chunkCreated = Math.floor(Date.now() / 1000);
-      res.write(
-        JSON.stringify({
-          created: chunkCreated,
-          data: [
-            {
-              url: imgUrl,
-            },
-          ],
-        })
-      );
-res.end();
     } else {
-  res.status(500).json({ error: "地址获取失效" });
-}
   } catch (error) {
-  console.error("Error:", error);
-}
 });
-app.listen(process.env.PORT || 3000);

 import bodyParser from "body-parser";
 import dotenv from "dotenv";
 import fetch from "node-fetch";
+import fs from "fs";
+import path from "path";
 dotenv.config();
 const app = express();
 app.use(bodyParser.json());
+// 创建图片保存目录
+const imageDir = path.join(process.cwd(), 'images');
+if (!fs.existsSync(imageDir)) {
+  fs.mkdirSync(imageDir);
+}
+// 静态文件服务
+app.use('/images', express.static(imageDir));
 app.get("/", (req, res) => {
   res.send(`
     <html>
       <head>
         <title>OPENAI</title>
+      </head>
       <body>
         <h1>OpenAI</h1>
         <p>Congratulations! Your project has been successfully deployed.</p>
 });
 app.post("/deem/v1/chat/completions", async (req, res) => {
+  let token;
   const authHeader =
     req.headers["authorization"] || req.headers["Authorization"];
   if (!authHeader) {
       errmsg: "Unauthorized.",
     });
   } else {
+    token = authHeader.split(" ")[1];
     if (!token) {
       return res.status(401).json({
         code: 401,
     const messages = data.messages;
     const lastMessage = messages[messages.length - 1];
     const messagesContent = lastMessage.content;
+    const apiUrl = `https://aihh.api.ecylt.top/?key=${token}&prompt=${messagesContent}`;
     const apiResponse = await fetch(apiUrl);
     if (!apiResponse.ok) {
       throw new Error("文生图服务失效：" + apiResponse.status);
     }
+    const imageBuffer = await apiResponse.buffer();
+    // 生成唯一的文件名
+    const fileName = `image_${Date.now()}.jpg`;
+    const filePath = path.join(imageDir, fileName);
+    // 保存图片到本地
+    fs.writeFileSync(filePath, imageBuffer);
+    // 构建本地图片URL
+    const imgUrl = `${req.protocol}://${req.get('host')}/images/${fileName}`;
     console.log(imgUrl);
       res.write("data: [DONE]\n\n");
       res.end();
     } else {
+      res.status(500).json({ error: "图片保存失败" });
     }
   } catch (error) {
     console.error("Error:", error);
+    res.status(500).json({ error: error.message });
   }
 });
 app.post("/deem/v1/images/generations", async (req, res) => {
+  let token;
   const authHeader =
     req.headers["authorization"] || req.headers["Authorization"];
   if (!authHeader) {
       errmsg: "Unauthorized.",
     });
   } else {
+    token = authHeader.split(" ")[1];
     if (!token) {
       return res.status(401).json({
         code: 401,
     const data = req.body;
     const messages = data.prompt;
+    const apiUrl = `https://aihh.api.ecylt.top/?key=${token}&prompt=${messages}`;
     const apiResponse = await fetch(apiUrl);
     if (!apiResponse.ok) {
       throw new Error("文生图服务失效：" + apiResponse.status);
     }
+    const imageBuffer = await apiResponse.buffer();
+    // 生成唯一的文件名
+    const fileName = `image_${Date.now()}.jpg`;
+    const filePath = path.join(imageDir, fileName);
+    // 保存图片到本地
+    fs.writeFileSync(filePath, imageBuffer);
+    // 构建本地图片URL
+    const imgUrl = `${req.protocol}://${req.get('host')}/images/${fileName}`;
     console.log(imgUrl);
     if (imgUrl) {
       const chunkCreated = Math.floor(Date.now() / 1000);
+      res.json({
+        created: chunkCreated,
+        data: [
+          {
+            url: imgUrl,
+          },
+        ],
+      });
     } else {
+      res.status(500).json({ error: "图片保存失败" });
+    }
   } catch (error) {
+    console.error("Error:", error);
+    res.status(500).json({ error: error.message });
+  }
 });
+app.listen(process.env.PORT || 3000);