import pandas as pd
from pandasai import SmartDataframe
from pandasai.llm.base import LLM
import requests

# 自定義DeepSeek大模型
class DeepSeekLLM(LLM):
    def __init__(self, api_url:str, api_key: str, model: str):
        super().__init__()
        self.api_url = api_url
        self.api_key = api_key
        self.model = model        


    def call(self, instruction, context: dict = None, **kwargs) -> str:  
        """
        PandasAI 用來(lái)向 LLM 模型發(fā)起請(qǐng)求的接口入口。

        Args:
            instruction (str): PandasAI 傳入的 prompt，可能是 string 或自定義對(duì)象，表示用戶的問(wèn)題。
            context (dict, optional): 包含上下文信息（例如 DataFrame 元數(shù)據(jù)等），在部分 LLM 中可用。
            **kwargs: 保留接口向前兼容（比如未來(lái)增加其他參數(shù)時(shí)也能傳入）。

        Returns:
            str: 執(zhí)行指令后的結(jié)果輸出。
        """

        # 把復(fù)雜的 Prompt 對(duì)象變成字符串
        if not isinstance(instruction, str):
            instruction = str(instruction)

        # 請(qǐng)求頭
        headers = {
            "Authorization": f"Bearer {self.api_key}",   # DeepSeek身份驗(yàn)證
            "Content-Type": "application/json"
        }

        # 兼容 OpenAI 的 Chat Completion API的對(duì)話消息格式
        messages = [
            {
                "role": "system", # 設(shè)定 LLM 的行為
                "content": "You are a helpful AI assistant for data analysis."
            },
            {
                "role": "user",  # 是實(shí)際的問(wèn)題
                "content": instruction
            }
        ]

        # 請(qǐng)求體
        payload = {
            "model": self.model,
            "messages": messages,
            "temperature": 0.0 # 值越低，回答的效果越穩(wěn)定
        }

        # 發(fā)送post請(qǐng)求
        response = requests.post(self.api_url, headers=headers, json=payload)

        # 檢查返回狀態(tài)碼，如果不是 200 則報(bào)錯(cuò)。
        if response.status_code != 200:
            raise Exception(f"DeepSeek API Error: {response.status_code}, {response.text}")

        # 解析返回結(jié)果
        result = response.json()
        return result["choices"][0]["message"]["content"]

    @property
    def type(self):
        return "deepseek-custom"

# Sample DataFrame
sales_by_country = pd.DataFrame({
    "country": ["United States", "United Kingdom", "France", "Germany", "Italy", "Spain", "Canada", "Australia", "Japan", "China"],
    "sales": [5000, 3200, 2900, 4100, 2300, 2100, 2500, 2600, 4500, 7000]
})

# 用自定義的 DeepSeek LLM
llm = DeepSeekLLM(
    api_url = "https://dashscope.aliyuncs.com/compatible-mode/v1/chat/completions",
    api_key="sk-xxx", 
    model="deepseek-r1")

df = SmartDataframe(sales_by_country, config={"llm": llm})

result = df.chat('列出銷(xiāo)售額最高的3個(gè)國(guó)家。')

print(result)

from openai import OpenAI

client = OpenAI(
    # 使用大模型對(duì)應(yīng)的Key
    api_key = "sk-xxx",
    base_url="https://dashscope.aliyuncs.com/compatible-mode/v1"
)

completion = client.chat.completions.create(
    model="deepseek-r1",  # 此處以 deepseek-r1 為例，可按需更換模型名稱(chēng)。
    messages=[
        {'role': 'user', 'content': '想快速入門(mén)AI大模型，給我推薦一下具體的學(xué)習(xí)方案。'}
    ]
)

# 通過(guò)reasoning_content字段打印思考過(guò)程
print("思考過(guò)程：")
print(completion.choices[0].message.reasoning_content)

# 通過(guò)content字段打印最終答案
print("最終答案：")
print(completion.choices[0].message.content)

# 初始化模型
chat = ChatOpenAI(
    model_name="deepseek-r1",
    temperature=0,
    base_url="https://dashscope.aliyuncs.com/compatible-mode/v1", #服務(wù)地址
    api_key="sk-xxx"  #API密鑰
)

# 發(fā)送請(qǐng)求
response = chat.invoke([HumanMessage(content="請(qǐng)用中文介紹你自己。")])

# 輸出結(jié)果
print(response.content)

from langchain_openai import ChatOpenAI

llm_client = ChatOpenAI(
    temperature=0.0,
    model_name="deepseek-r1",
    base_url="https://dashscope.aliyuncs.com/compatible-mode/v1",
    api_key="sk-852da921b11545c99de697e584210fc7"
)

# 假設(shè)你的多標(biāo)簽集合如下：
total_class = {"正面", "負(fù)面", "價(jià)格問(wèn)題", "物流問(wèn)題", "推薦", "外觀設(shè)計(jì)", "用戶體驗(yàn)"}

# 系統(tǒng)提示詞
system_prompt = """你是一個(gè)多標(biāo)簽分類(lèi)助手，請(qǐng)從以下標(biāo)簽中選出適用于文本的所有標(biāo)簽（可以多選）：
正面, 負(fù)面, 價(jià)格問(wèn)題, 物流問(wèn)題, 推薦, 外觀設(shè)計(jì), 用戶體驗(yàn)

只輸出標(biāo)簽，多個(gè)標(biāo)簽之間用英文逗號(hào)分隔。如果無(wú)法判斷，則返回空字符串。
"""

def llm_labeling(texts: list[str]) -> list[list[str]]:
    """
    對(duì)輸入文本列表進(jìn)行多標(biāo)簽打標(biāo)，返回每條文本對(duì)應(yīng)的標(biāo)簽列表。
    """
    results = []
    batch_inputs = [system_prompt + '\n' + t for t in texts] 
    res = llm_client.batch(batch_inputs)
    for item in res:
        content = item.content.strip()
        if not content:
            results.append([])
            continue
        # 解析逗號(hào)分隔標(biāo)簽，清洗一下
        tags = [t.strip() for t in content.split(',')]
        # 只保留在 total_class 中的合法標(biāo)簽
        tags = [t for t in tags if t in total_class]
        results.append(tags)
    return results

texts = [
    "這個(gè)產(chǎn)品非常好用，值得推薦",
    "物流速度太慢了，體驗(yàn)不好",
    "外觀漂亮，使用方便，就是價(jià)格稍貴"
]

result = llm_labeling(texts)
for i, tags in enumerate(result):
    print(f"第{i+1}條: 標(biāo)簽 = 
PandasAI
LLM
數(shù)據(jù)分析
")