import os
import re
from dotenv import load_dotenv, find_dotenv
from openai import OpenAI
from concurrent.futures import ThreadPoolExecutor


loaded = load_dotenv(find_dotenv(), override=True)
# 从环境变量中获取 OpenAI API Key 或者直接赋值
API_KEY = os.getenv("API_KEY")

# 如果您使用的是官方 API，就直接用 https://api.siliconflow.cn/v1 就行。
BASE_URL = "https://api.siliconflow.cn/v1"
# 如果您使用的不是官方 API，而是通过代理进行请求，请设置您的代理 URL。
# BASE_URL = "https://api.xxx.../v1"

# 实例化 OpenAI 对象
# 传入参数：OpenAI API Key（必需）、Base URL 和最大重试次数
client = OpenAI(api_key=API_KEY, base_url=BASE_URL, max_retries=3)

# 参数 n，整数或 Null，可选项，默认为 1。为每条输入信息生成多少个聊天完成选项。
# 参数 temperature，实数值或 Null，可选项，默认为 1。使用的采样温度，介于 0 和 2 之间。0.8 等较高值会使输出更加随机，而 0.2 等较低值会使输出更加集中和确定。

def get_completions(llm_prompt, model_endpoint):

    extra_body = {}
    if "Qwen3" in model_endpoint:
        extra_body={
            "enable_thinking": False
        }
    
    response = client.chat.completions.create(model=model_endpoint,
                                              messages=[
                                                        {"role": "user", "content": llm_prompt}
                                                       ],
                                              n=1, temperature=0.20, seed=42,
                                              presence_penalty=0, frequency_penalty=0,
                                              max_tokens=4096, extra_body = extra_body
                                             )

    return response.choices[0].message.content.strip()

llm = "Qwen/Qwen3-8B"

question_Chinese = "经常锻炼有什么好处呢？"

skeleton_prompt_Chinese = f"""您是一位组织者，只负责提供回答问题的骨架（而非完整内容）。
以要点列表（编号为 1.、2.、3.等）的形式提供骨架，以回答问题。
每个要点应非常简短，只有 2∼8 个字，而不是写一个完整的句子。
一般来说，骨架应包含 3∼10 个要点。

问题：
中国菜有哪些典型类型？
骨架：
1. 饺子。
2. 面条。
3. 点心。
4. 火锅。
5. 云吞。
6. 麻婆豆腐。
7. 叉烧。
8. 炒饭。

问题：
对于个体有哪些减少碳排放的实用建议？
骨架：
1. 节约能源。
2. 高效交通。
3. 家庭能效。
4. 减少用水量。
5. 可持续饮食。
6. 可持续的旅行。

现在，请为下面的问题提供骨架。
{question_Chinese}
骨架：
"""

skeleton_result_Chinese = get_completions(skeleton_prompt_Chinese, llm)

print(f"生成的思维骨架-中文：\n{skeleton_result_Chinese}")

llm = "Qwen/Qwen3-8B"

question_English = "What are the benefits of regular exercise?"

skeleton_prompt_English = f"""You are an organizer responsible for only giving the skeleton (not the full content) for answering the question.\
Provide the skeleton in a list of points (numbered 1., 2., 3., etc.) to answer the question. \
Instead of writing a full sentence, each skeleton point should be very short with only 2∼6 words. \
Generally, the skeleton should have 3∼10 points.

Question:
What are the typical types of Chinese dishes?
Skeleton:
1. Dumplings.
2. Noodles.
3. Dim Sum.
4. Hot Pot.
5. Wonton.
6. Ma Po Tofu.
7. Char Siu.
8. Fried Rice.

Question:
What are some practical tips for individuals to reduce their carbon emissions?
Skeleton:
1. Energy conservation.
2. Efficient transportation.
3. Home energy efficiency.
4. Reduce water consumption.
5. Sustainable diet.
6. Sustainable travel.

Now, please provide the skeleton for the following question.
{question_English}
Skeleton:
"""

skeleton_result_English = get_completions(skeleton_prompt_English, llm)

print(f"生成的思维骨架-英文：\n{skeleton_result_English}")

# 我们可以使用一个专门设计的正则表达式 "(\d+)\.\s?([\s\S]+?)(?=\n|\n*$)" 从骨架响应中提取要点索引和要点骨架。
key_point_skeleton_English = re.findall(r"(\d+)\.\s?([\s\S]+?)(?=\n|\n*$)", skeleton_result_English)
key_point_skeleton_Chinese = re.findall(r"(\d+)\.\s?([\s\S]+?)(?=\n|\n*$)", skeleton_result_Chinese)
print(key_point_skeleton_English)
print(key_point_skeleton_Chinese)

def expand_point_English(point_data):
    point_index, key_point = point_data
    point_expanding_prompt = f"""You are responsible for continuing the writing of one and only one point \
in the overall answer to the following question.

{question_English}

The skeleton of the answer is:
{skeleton_result_English}

Continue and only continue the writing of point {point_index}. \
Write it accurately and concisely in 2∼4 sentences, and do not continue with other points!
"""
    point_expanding = get_completions(point_expanding_prompt, llm)

    return (point_index, point_expanding)


def expand_point_Chinese(point_data):
    point_index, key_point = point_data
    point_expanding_prompt = f"""您有责任在下面问题的总答案中续写一个且仅一个要点。

{question_Chinese}

答案的骨架是：
{skeleton_result_Chinese}

继续且仅继续书写点 {point_index}。用 4∼6 句话准确、简洁地写完，不要续写其他要点！
"""
    point_expanding = get_completions(point_expanding_prompt, llm)

    return (point_index, point_expanding)

llm = "Qwen/Qwen3-8B"

question_Chinese = "经常锻炼有什么好处呢？"
final_result_Chinese = ""

# For 循环顺序执行
# for point_index, key_point in key_point_skeleton_Chinese:
#     point_expanding_prompt = f"""您有责任在下面问题的总答案中续写一个且仅一个要点。

# {question_Chinese}

# 答案的骨架是：
# {skeleton_result_Chinese}

# 继续且仅继续书写点 {point_index}。用 3∼6 句话准确、简洁地写完，不要续写其他要点！
# """
#     point_expanding = get_completions(point_expanding_prompt, llm)
#     if point_index == key_point_skeleton[-1][0]:
#         final_result_Chinese += f"{point_index}. {point_expanding}"
#     else:
#         final_result_Chinese += f"{point_index}. {point_expanding}\n"

# 使用 ThreadPoolExecutor 来并发执行 expand_point_Chinese 函数
results = []
max_workers = len(key_point_skeleton_Chinese)
with ThreadPoolExecutor(max_workers=max_workers) as executor:
    results = list(executor.map(expand_point_Chinese, key_point_skeleton_Chinese))

for point_index, point_expanding in results:
    # print(f"Point Index: {point_index}, Point Expanding: {point_expanding}")
    if point_expanding[0].isnumeric():      # 判断有没有数字编号
        pass
    else:
        point_expanding = f"{point_index}. {point_expanding}"
    if point_index == key_point_skeleton_Chinese[-1][0]:
        final_result_Chinese += f"{point_expanding}"
    else:
        final_result_Chinese += f"{point_expanding}\n"

print(f"要点扩展后的内容-中文：\n{final_result_Chinese}")

llm = "Qwen/Qwen3-8B"

question_English = "What are the benefits of regular exercise?"
final_result_English = ""

# For 循环顺序执行
# for point_index, key_point in key_point_skeleton_English:
#     point_expanding_prompt = f"""You are responsible for continuing the writing of one and only one point \
# in the overall answer to the following question.

# {question_English}

# The skeleton of the answer is:
# {skeleton_result_English}

# Continue and only continue the writing of point {point_index}. \
# Write it accurately and concisely in 3∼5 sentences, and do not continue with other points!
# """
#     point_expanding = get_completions(point_expanding_prompt, llm)
#     if point_index == key_point_skeleton[-1][0]:
#         final_result_English += f"{point_index}. {point_expanding}"
#     else:
#         final_result_English += f"{point_index}. {point_expanding}\n"

# 使用 ThreadPoolExecutor 来并发执行 expand_point_English 函数
results = []
max_workers = len(key_point_skeleton_English)
with ThreadPoolExecutor(max_workers=4) as executor:
    results = list(executor.map(expand_point_English, key_point_skeleton_English))

for point_index, point_expanding in results:
    # print(f"Point Index: {point_index}, Point Expanding: {point_expanding}")
    if point_expanding[0].isnumeric():      # 判断有没有数字编号
        pass
    else:
        point_expanding = f"{point_index}. {point_expanding}"
    if point_index == key_point_skeleton_English[-1][0]:
        final_result_English += f"{point_expanding}"
    else:
        final_result_English += f"{point_expanding}\n"

print(f"要点扩展后的内容-英文：\n{final_result_English}")

llm = "Qwen/Qwen3-8B"

question_Chinese = "人工智能专业的研究生应该养成那些良好的科研习惯？"

prompt_SoT_Chinese = f"""您的任务是分两步回答下面的问题。\
首先，用一个要点列表（编号为 1.、2.、3.等）列出简明扼要的答案要点。\
每个要点应非常简短，只有 2∼8 个字。一般来说，骨架应包含 3∼10 个要点。\
提供骨架后，用 3∼6 句话简要扩展每个要点。

示例问题：
经常锻炼有什么好处呢？
骨架：
1. 增强心肺功能。
2. 控制体重。
3. 提高免疫力。
4. 改善睡眠质量。
5. 减少患疾病风险。
6. 提升心情和情绪。
7. 延缓衰老。
要点扩展后的答案：
1. 经常锻炼可以增强心肺功能，使心脏更加强壮，提高肺活量，增加氧气的吸收量。这有助于提高身体的耐力和抵抗力，减少疲劳感，让我们在日常生活中更加精力充沛和活力十足。
2. 控制体重。经常锻炼可以帮助消耗多余的热量，从而帮助控制体重。此外，锻炼还可以增加肌肉质量，提高新陈代谢率，进一步帮助维持健康的体重。保持适当的体重有助于预防肥胖相关疾病，如心血管疾病和糖尿病。因此，通过坚持锻炼，可以更好地控制体重，保持身体健康。
3. 经常锻炼可以提高免疫力，使身体更加抵抗疾病。锻炼可以促进血液循环，增加白细胞数量，从而帮助身体更快地应对病毒和细菌的入侵。此外，锻炼还可以减少患感冒和流感等呼吸道疾病的风险，让身体更加健康。
4. 锻炼可以帮助改善睡眠质量，使人更容易入睡并保持深度睡眠状态。适当的运动可以调节身体的生物钟，帮助人们建立规律的作息时间。此外，锻炼也有助于减轻焦虑和压力，使人们在睡前更加放松，从而提高睡眠质量。良好的睡眠对身体健康和心理健康都至关重要。
5. 经常锻炼可以减少患疾病的风险，包括心脏病、中风、糖尿病和某些癌症。运动有助于控制体重，减少肥胖引发的健康问题。此外，锻炼还可以提高身体的新陈代谢，促进身体内部的废物排出，保持身体健康。
6. 锻炼可以释放身体内的内啡肽和多巴胺等神经递质，提升心情和情绪。这些化学物质可以帮助减轻焦虑和抑郁，让人感到更加愉快和放松。通过锻炼，人们可以更好地处理压力和情绪波动，保持心理健康。
7. 锻炼还可以增加身体的灵活性和力量，减少关节疼痛和肌肉疲劳。此外，定期锻炼还可以改善身体姿势和平衡能力，降低摔倒和受伤的风险。最重要的是，锻炼可以提高自信心和自尊心，让人更加积极乐观地面对生活中的挑战。
您的任务是回答：```{question_Chinese}```

骨架：
1. 
2. 
...

要点扩展后的答案：
1. 
2. 
...
"""

result_SoT_Chinese = get_completions(prompt_SoT_Chinese, llm)

print(f"SoT生成结果-中文：\n{result_SoT_Chinese}")

llm = "Qwen/Qwen3-8B"

question_English = "What good research habits should graduate students in Artificial Intelligence Major?"

prompt_SoT_English = f"""You are tasked with answering the following question in a two-step process. \
First, create a concise skeleton of the answer in a list of points (numbered 1., 2., 3., etc.). \
Each skeleton point should be very short with only 2~6 words. \
Generally, the skeleton should have 3∼10 points. \
After providing the skeleton, expand on each point briefly in 2∼4 sentences.

Example Question:
What are the benefits of regular exercise?
Skeleton:
1. Weight management.
2. Improved mood.
3. Increased energy levels.
4. Better sleep quality.
5. Stronger immune system.
6. Reduced risk of chronic diseases.
7. Improved cognitive function.
8. Better overall physical health.
Expanded Answer:
1. Regular exercise helps with weight management by burning calories and increasing metabolism. It can also help build muscle mass, which in turn can help increase the body's ability to burn calories even at rest. Additionally, exercise can help regulate appetite and improve overall body composition.
2. Regular exercise has been shown to improve mood by releasing endorphins, which are known as "feel-good" hormones. This can help reduce feelings of anxiety, stress, and depression, leading to a more positive outlook on life. Additionally, exercise can provide a sense of accomplishment and boost self-esteem, further enhancing overall emotional well-being.
3. Regular exercise can lead to increased energy levels by improving cardiovascular health, increasing muscle strength, and enhancing overall endurance. This boost in energy can help individuals feel more alert and focused throughout the day, leading to increased productivity and a greater sense of well-being.
4. Regular exercise can lead to better sleep quality by helping to regulate sleep patterns and improve overall sleep efficiency. Physical activity can also reduce symptoms of insomnia and promote deeper, more restful sleep. Additionally, exercise has been shown to decrease the time it takes to fall asleep and increase the amount of time spent in the restorative stages of sleep.
5. Regular exercise helps to strengthen the immune system by promoting healthy circulation, which allows immune cells to move through the body more efficiently. This can help the body fight off infections and illnesses more effectively, reducing the frequency and severity of colds, flu, and other common ailments. Additionally, exercise can also help to reduce inflammation in the body, further supporting immune function.
6. Regular exercise can help reduce the risk of chronic diseases such as heart disease, diabetes, and certain types of cancer. It can also improve cardiovascular health, lower blood pressure, and decrease the likelihood of developing conditions like osteoporosis. Additionally, exercise can help manage and improve symptoms of existing chronic diseases, leading to a better quality of life.
7. Regular exercise has been shown to improve cognitive function by increasing blood flow to the brain, promoting the growth of new brain cells, and enhancing overall brain health. This can lead to better memory, sharper focus, and improved decision-making skills, ultimately contributing to a higher quality of life.
8. Regular exercise also helps to improve overall physical health by strengthening the cardiovascular system, increasing muscle strength and endurance, improving flexibility and balance, and enhancing overall physical performance. This can lead to a decreased risk of injury, improved posture, and a greater ability to perform daily tasks with ease. Additionally, exercise can help to maintain bone density and reduce the risk of osteoporosis as we age.
Your task is to answer: ```{question_English}```

Skeleton:
1. 
2. 
...

Expanded Answer:
1. 
2. 
...
"""

result_SoT_English = get_completions(prompt_SoT_English, llm)

print(f"SoT生成结果-英文：\n{result_SoT_English}")