upload Ai

app.js

@@ -47,6 +47,7 @@ const goodsRoutes = require('./routes/goods')
 const userRoutes = require('./routes/users')
 const uploadRoutes = require('./routes/upload')
 const stockRoutes = require('./routes/stock')
+const aiRoutes = require('./routes/ai')
 
 router.use('/api/orders', orderRoutes)
 router.use('/api/categories', categoryRoutes)
@@ -54,6 +55,7 @@ router.use('/api/goods', goodsRoutes)
 router.use('/api/users', userRoutes)
 router.use('/api/upload', uploadRoutes)
 router.use('/api/stock', stockRoutes)
+router.use('/api/ai', aiRoutes)
 
 app.use(router.routes())
 app.use(router.allowedMethods())

package.json

@@ -15,7 +15,8 @@
     "koa-router": "^10.1.1",
     "koa-static": "^5.0.0",
     "multer": "^2.1.1",
-    "mysql2": "^3.22.3"
+    "mysql2": "^3.22.3",
+    "node-fetch": "^2.6.7"
   },
   "keywords": [],
   "author": "",

routes/ai.js

@@ -0,0 +1,110 @@
+const Router = require('koa-router');
+const fetch = require('node-fetch');
+
+const router = new Router();
+
+// NVIDIA API 配置
+const NVIDIA_API_KEY = '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';
+const NVIDIA_API_URL = 'https://integrate.api.nvidia.com/v1/chat/completions';
+
+// 生成商品信息的 API
+router.post('/generate-product', async (ctx) => {
+  try {
+    const { imageUrl, keywords } = ctx.request.body;
+    
+    if (!imageUrl && !keywords) {
+      ctx.status = 400;
+      ctx.body = {
+        code: 400,
+        message: '请提供商品图片或关键词'
+      };
+      return;
+    }
+
+    // 构建提示词
+    let prompt = '你是一个专业的便利店商品管理助手。';
+    
+    if (imageUrl) {
+      prompt += `\n请分析这张商品图片：${imageUrl}`;
+    }
+    
+    if (keywords) {
+      prompt += `\n关键词：${keywords}`;
+    }
+    
+    prompt += `
+请生成商品的详细信息，返回JSON格式，不要包含其他内容：
+{
+  "name": "商品名称（简洁明了，2-10字）",
+  "category": "商品分类（请从以下选择：饮料,零食,日用品,食品,烟酒,其他）",
+  "description": "商品详细描述（50-100字，突出产品特点）",
+  "suggestedPrice": 建议售价（数字）
+}`;
+
+    // 调用 NVIDIA API
+    const response = await fetch(NVIDIA_API_URL, {
+      method: 'POST',
+      headers: {
+        'Authorization': `Bearer ${NVIDIA_API_KEY}`,
+        'Content-Type': 'application/json'
+      },
+      body: JSON.stringify({
+        model: 'deepseek-ai/deepseek-v4-pro',
+        messages: [
+          {
+            role: 'user',
+            content: prompt
+          }
+        ],
+        temperature: 0.7,
+        max_tokens: 500
+      })
+    });
+
+    if (!response.ok) {
+      const errorText = await response.text();
+      console.error('NVIDIA API Error:', response.status, errorText);
+      throw new Error(`API 调用失败: ${response.status}`);
+    }
+
+    const data = await response.json();
+    const aiResponse = data.choices[0].message.content;
+    
+    // 解析 JSON 响应
+    let productInfo;
+    try {
+      // 清理响应文本，只保留 JSON 部分
+      const jsonMatch = aiResponse.match(/\{[\s\S]*\}/);
+      if (jsonMatch) {
+        productInfo = JSON.parse(jsonMatch[0]);
+      } else {
+        throw new Error('无法解析 AI 响应');
+      }
+    } catch (parseError) {
+      console.error('JSON 解析失败:', aiResponse);
+      // 如果解析失败，提供默认值
+      productInfo = {
+        name: keywords || '新商品',
+        category: '其他',
+        description: '商品描述',
+        suggestedPrice: 0
+      };
+    }
+
+    ctx.body = {
+      code: 200,
+      message: '生成成功',
+      data: productInfo
+    };
+
+  } catch (error) {
+    console.error('生成商品信息失败:', error);
+    ctx.status = 500;
+    ctx.body = {
+      code: 500,
+      message: error.message || '生成失败，请稍后重试'
+    };
+  }
+});
+
+module.exports = router;

scripts/fix-image-urls.js

@@ -0,0 +1,65 @@
+const mysql = require('mysql2/promise');
+const config = require('../config/database');
+
+async function fixImageUrls() {
+  try {
+    const connection = await mysql.createConnection(config);
+    console.log('✅ 数据库连接成功\n');
+
+    // 查找包含旧地址的记录
+    console.log('🔍 检查数据库中的图片URL...');
+    const [goods] = await connection.execute('SELECT id, images FROM goods WHERE images LIKE "%localhost%"');
+    
+    console.log(`找到 ${goods.length} 条包含 localhost 的记录\n`);
+
+    // 更新图片URL
+    for (const item of goods) {
+      if (item.images) {
+        const oldImages = item.images;
+        // 替换所有 localhost:3000 和 localhost:3005
+        let newImages = oldImages
+          .replace(/http:\/\/localhost:3000/g, 'http://110.42.255.239:3005')
+          .replace(/http:\/\/localhost:3005/g, 'http://110.42.255.239:3005');
+        
+        await connection.execute(
+          'UPDATE goods SET images = ? WHERE id = ?',
+          [newImages, item.id]
+        );
+        
+        console.log(`商品ID ${item.id}:`);
+        console.log(`  旧: ${oldImages}`);
+        console.log(`  新: ${newImages}`);
+        console.log('---');
+      }
+    }
+
+    // 检查其他表（如有需要）
+    const [pointsGoods] = await connection.execute('SELECT id, images FROM points_goods WHERE images LIKE "%localhost%"');
+    if (pointsGoods.length > 0) {
+      console.log(`\n找到 ${pointsGoods.length} 条积分商品记录...`);
+      for (const item of pointsGoods) {
+        if (item.images) {
+          const oldImages = item.images;
+          let newImages = oldImages
+            .replace(/http:\/\/localhost:3000/g, 'http://110.42.255.239:3005')
+            .replace(/http:\/\/localhost:3005/g, 'http://110.42.255.239:3005');
+          
+          await connection.execute(
+            'UPDATE points_goods SET images = ? WHERE id = ?',
+            [newImages, item.id]
+          );
+          
+          console.log(`积分商品ID ${item.id}: 已更新`);
+        }
+      }
+    }
+
+    await connection.end();
+    console.log('\n✅ 所有图片URL更新完成！');
+  } catch (error) {
+    console.error('❌ 更新失败:', error);
+    process.exit(1);
+  }
+}
+
+fixImageUrls();