import os
import sys

sys.path.append(os.getcwd())
import rag2_0.dify.dify_client.dify_api as DifyApi

import pandas as pd
pd_data = pd.read_excel("data/excel/2025年5月30日到6月10号对话记录_转工单.xlsx")


dify_api = DifyApi.DifyApi()
dataset_id = dify_api.get_or_create_dataset_by_name("工单问答数据")
document_id = dify_api.upload_text_to_document(text_name="5月30日到6月10号对话工单", text="", dataset_id=dataset_id)

segments_list=[]
for index, row in pd_data.iterrows():
    query = row["客户问题"]
    answer = row["解决方案"]
    if "存在抱怨" in answer:
        answer = answer.split("存在抱怨")[0]
    
    content = f"问题：{query}\n回答：{answer}"
    segments_list.append({
        "content": str(content),
        "answer": "",
        "keywords": []
    })

dify_api.add_document_segments(dataset_id=dataset_id, document_id=document_id, segments_list=segments_list)