import requests,os
import json
from flask import Flask, request, jsonify

app = Flask(__name__)

# 请替换为你的 DeepSeek API 地址和 API Key
API_URL = "https://api.deepseek.com"
API_KEY = os.getenv('DeepSeek_API_KEY')

def call_deepseek(prompt, model="deepseek-chat", max_tokens=2048, temperature=0.7):
    """
    调用 DeepSeek API 的函数
    
    参数:
    - prompt: 输入的提示文本
    - model: 使用的模型名称(默认为 deepseek-chat)
    - max_tokens: 生成的最大token数
    - temperature: 控制生成随机性的参数(0-1)
    
    返回:
    - API的响应内容
    """
    # API端点(请根据实际API文档替换)
    api_url = "https://api.deepseek.com/v1/chat/completions"
    
    # 你的API密钥(请替换为实际的密钥)
    api_key = "sk-b36c4122bb874a3f83f250d2e9c8a705"
    
    # 请求头
    headers = {
        "Content-Type": "application/json",
        "Authorization": f"Bearer {api_key}"
    }
    
    # 请求体
    payload = {
        "model": model,
        "messages": [{"role": "user", "content": prompt}],
        "max_tokens": max_tokens,
        "temperature": temperature
    }
    
    try:
        # 发送POST请求
        response = requests.post(api_url, headers=headers, json=payload)
        response.raise_for_status()  # 检查是否有错误
        
        # 返回API响应
        return response.json()
    
    except requests.exceptions.RequestException as e:
        print(f"请求出错: {e}")
        return None

@app.route('/generate', methods=['POST'])
def generate_text():
    data = request.get_json()
    prompt = data.get('prompt')
    max_new_tokens = data.get('max_new_tokens', 150)

    if not prompt:
        return jsonify({"error": "Prompt is required"}), 400

    response = call_deepseek(prompt, max_new_tokens)
    if response:
        return jsonify({"generated_text": response})
    else:
        return jsonify({"error": "Failed to generate text"}), 500

if __name__ == "__main__":
    app.run(debug=True)