package com.ai.controller;

import com.ai.service.Assist;
import com.ai.service.LangChainService;
import dev.langchain4j.community.model.dashscope.QwenChatModel;
import dev.langchain4j.community.model.zhipu.ZhipuAiImageModel;
import dev.langchain4j.data.image.Image;
import dev.langchain4j.memory.ChatMemory;
import dev.langchain4j.memory.chat.MessageWindowChatMemory;
import dev.langchain4j.model.chat.ChatLanguageModel;
import dev.langchain4j.model.output.Response;
import dev.langchain4j.service.AiServices;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;

import java.net.URI;

@RestController
@RequestMapping("/langchain")
public class LangChainController {

    @Autowired
    private LangChainService langChainService;

    @Autowired
    private Assist assist;

    @Autowired
    private ZhipuAiImageModel zhipuAiImageModel;

    /**
     * 处理用户输入并调用 LangChainService 获取响应
     * @param input 用户输入的内容
     * @return 大语言模型生成的响应
     */
    @GetMapping("/chat")
    public String chat(@RequestParam("input") String input) {
        System.out.println("start chat...");
        return langChainService.getResponse(input);
    }

    @GetMapping("/high/chat")
    public String highChat(@RequestParam("input") String input) {
        System.out.println("start highlevel chat...");
        return assist.chat(input);
    }

    @GetMapping("/high/memory-chat")
    public String memoryChat(@RequestParam("memoryId") String memoryId, @RequestParam("input") String input) {
        System.out.println("start highlevel memory chat...");
        return assist.memoryChat(memoryId, input);
    }

    @GetMapping("/zhipu/img")
    public String generateImg( @RequestParam("input") String input) {
        System.out.println("start generate img...");
        Response<Image> generate = zhipuAiImageModel.generate(input);
        URI url = generate.content().url();
        return "Your remote image is here: " + url;
    }

    @GetMapping("/high/call")
    public String functionCall(@RequestParam("input") String input) {
        System.out.println("start highlevel memory chat...");
        return assist.chat(input);
    }
}