from flask import Flask, render_template, request, jsonify import openai import time

设置代理网址

openai.api_base = 'https://api.openai-proxy.com/v1'

替换为您自己的OpenAI API密钥

api_key = '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'

初始化OpenAI

openai.api_key = api_key

创建Flask应用程序

app = Flask(name)

定义可供选择的模型

available_models = { 'gpt-3.5-turbo-0301': 'GPT-3.5-Turbo-0301', 'gpt-3.5-turbo-16k': 'GPT-3.5-Turbo-16k', 'gpt-3.5-turbo': 'GPT-3.5-Turbo', 'gpt-3.5-turbo-0613': 'GPT-3.5-Turbo-0613', 'gpt-3.5-turbo-16k-0613': 'GPT-3.5-Turbo-16k-0613' }

messages = []

@app.route('/') def index(): return render_template('index4.html', models=available_models)

@app.route('/get_response', methods=['POST']) def get_response(): user_input = request.form.get('user_input') selected_model = request.form.get('selected_model') system_message = request.form.get('system_message') temperature = float(request.form.get('temperature')) max_tokens = int(request.form.get('max_tokens')) continuous_chat = request.form.get('continuous_chat') # 获取连续对话参数

try:
    if continuous_chat == 'true' and messages:
        messages.append({'role': 'user', 'content': user_input})
    else:
        messages.clear()  # 清空消息列表
        messages.append({'role': 'system', 'content': system_message})
        messages.append({'role': 'user', 'content': user_input})

    # 调整消息历史记录的长度
    if len(messages) > 3:
        messages.pop(0)

    response = openai.ChatCompletion.create(
        model=selected_model,
        messages=messages,
        temperature=temperature,
        max_tokens=max_tokens,
        stream=True
    )
    collected_chunks = []
    collected_messages = []
    for chunk in response:
        collected_chunks.append(chunk)  # save the event response
        generated_text = chunk['choices'][0]['delta']  # extract the message
        collected_messages.append(generated_text)
        time.sleep(0.03)  # 暂停30毫秒

    full_reply_content = ''.join([m.get('content', '') for m in collected_messages])

    messages.append({'role': 'assistant', 'content': full_reply_content})  # 将助手回复添加到消息中
    print('用户输入内容：', user_input)
    print('GPT回复内容：', full_reply_content)
    return jsonify({'response': full_reply_content})
except Exception as e:
    return jsonify({'error': f'发生错误: {str(e)}'})

if name == 'main': app.run('0.0.0.0', 8000)