package servlet;

import java.io.IOException;
import java.io.PrintWriter;
import java.net.URI;
import java.net.http.HttpClient;
import java.net.http.HttpRequest;
import java.net.http.HttpResponse;
import javax.servlet.ServletException;
import javax.servlet.annotation.WebServlet;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import org.json.JSONObject;
import java.util.logging.Logger;
import java.util.logging.Level;

public class ChatServlet extends HttpServlet {
    private static final String API_KEY = "sk-0ad1eaf5194249439f82698893607bef"; // 请替换为您的实际 API 密钥
    private static final String APP_ID = "1d2b00751a914efd84ea4b354fffd962"; // 请替换为您的实际应用 ID
    private static final String API_URL = "https://dashscope.aliyuncs.com/api/v1/apps/" + APP_ID + "/completion";
    private static final Logger LOGGER = Logger.getLogger(ChatServlet.class.getName());

    protected void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
        request.setCharacterEncoding("UTF-8");
        response.setContentType("application/json;charset=UTF-8");
        response.setCharacterEncoding("UTF-8");

        String prompt = request.getParameter("prompt");
        
        try {
            HttpClient client = HttpClient.newHttpClient();
            JSONObject requestBody = new JSONObject();
            JSONObject input = new JSONObject();
            input.put("prompt", prompt);
            requestBody.put("input", input);
            requestBody.put("parameters", new JSONObject());

            HttpRequest apiRequest = HttpRequest.newBuilder()
                .uri(URI.create(API_URL))
                .header("Content-Type", "application/json")
                .header("Authorization", "Bearer " + API_KEY)
                .POST(HttpRequest.BodyPublishers.ofString(requestBody.toString()))
                .build();

            HttpResponse<String> apiResponse = client.send(apiRequest, HttpResponse.BodyHandlers.ofString());
            /*
            if (apiResponse.statusCode() == 200) {
                PrintWriter out = response.getWriter();
                out.print(apiResponse.body());
                out.flush();
              */  
            if (apiResponse.statusCode() == 200) {
                String responseBody = apiResponse.body();
                String formattedResponse = formatResponse(responseBody); // 格式化响应
                PrintWriter out = response.getWriter();
                out.print(formattedResponse);
                out.flush();    
            } else {
                LOGGER.log(Level.SEVERE, "API request failed with status code: " + apiResponse.statusCode());
                response.setStatus(HttpServletResponse.SC_INTERNAL_SERVER_ERROR);
                response.getWriter().write("{\"error\":\"API请求失败\"}");
            }
        } catch (Exception e) {
            LOGGER.log(Level.SEVERE, "Error in ChatServlet", e);
            response.setStatus(HttpServletResponse.SC_INTERNAL_SERVER_ERROR);
            response.getWriter().write("{\"error\":\"服务器错误: " + e.getMessage() + "\"}");
        }
    }
 // 格式化函数
    private String formatResponse(String responseBody) {
        // 假设responseBody是字符串形式的JSON，您可能需要先解析它
        // 这里简单示例对字符串进行替换和格式化
        return responseBody
            .replaceAll("\\*\\*(.*?)\\*\\*", "<strong>$1</strong>") // 加粗文本
            .replaceAll("\\n", "<br>") // 替换换行符
            .replaceAll("(\\d+\\. )", "<li>$1") // 列表项处理
            .replaceAll("\\s+", " "); // 去除多余空白
    }
}