from flask import Flask, request, jsonify
import math

app = Flask(__name__)

# 功能：：： 计算煤矿物料在多个地点经过不同时间后的最终温度
# initial_temperature：初始温度，物料开始时的温度（单位：摄氏度或华氏度）。
# k：热传递系数，表示热量交换的速度。该值越大，物料温度变化越快。
# times：一个列表，表示物料在每个地点停留的时间（单位：时间）。
# locations：一个列表，表示每个地点的环境温度（单位：摄氏度或华氏度）
def calculate_temperature(initial_temperature, k, times, locations):
    current_temperature = initial_temperature
    for i in range(len(locations)):
        t = times[i]
        T_env = locations[i]
        current_temperature = current_temperature + (T_env - current_temperature) * (1 - math.exp(-k * t))
    return current_temperature


@app.route('/calculate', methods=['POST'])
def calculate():
    # 获取 JSON 数据
    data = request.get_json()
    initial_temperature = data.get("initial_temperature")
    k = data.get("k")
    times = data.get("times")
    locations = data.get("locations")

    # 计算结果
    result = calculate_temperature(initial_temperature, k, times, locations)
    return jsonify({"final_temperature": result})


if __name__ == "__main__":
    # 启动 Flask 服务
    app.run(port=9997, debug=True)