const fs = require('fs');
const path = require('path');

// 读取server.js文件
const serverFilePath = path.join(__dirname, '..', 'server', 'server.js');
let content = fs.readFileSync(serverFilePath, 'utf8');

// 替换Gemini的API端点URL (单图片处理)
content = content.replace(
  /apiUrl = 'YOUR_GEMINI_API_ENDPOINT';/g, 
  "apiUrl = 'https://generativelanguage.googleapis.com/v1beta/models/gemini-pro-vision:generateContent';"
);

// 替换GPT-4o的API端点URL (单图片处理)
content = content.replace(
  /apiUrl = 'YOUR_GPT4O_API_ENDPOINT';/g, 
  "apiUrl = 'https://api.openai.com/v1/chat/completions';"
);

// 替换Gemini的API端点URL (多图片处理)
content = content.replace(
  /apiUrl = 'YOUR_GEMINI_API_ENDPOINT'; \/\/ Replace with actual Gemini endpoint/g, 
  "apiUrl = 'https://generativelanguage.googleapis.com/v1beta/models/gemini-pro-vision:generateContent';"
);

// 替换GPT-4o的API端点URL (多图片处理)
content = content.replace(
  /apiUrl = 'YOUR_GPT4O_API_ENDPOINT'; \/\/ Replace with actual GPT-4o endpoint/g, 
  "apiUrl = 'https://api.openai.com/v1/chat/completions';"
);

// 写回文件
fs.writeFileSync(serverFilePath, content, 'utf8');

console.log('API端点URL已成功修复！');