改写下上面程序 按json文件名生成

好的，改写如下：

import openai
import json

with open("openai_api_key.json") as f:
    openai.api_key = json.load(f)["api_key"]

def generate_text(prompt, model, temperature=0.5, max_tokens=50):
    completions = openai.Completion.create(
        engine=model,
        prompt=prompt,
        temperature=temperature,
        max_tokens=max_tokens,
        n=1,
        stop=None,
        timeout=10,
    )

    message = completions.choices[0].text.strip()
    return message

def main():
    with open("chitgpt_config.json") as f:
        config = json.load(f)

    prompt = input("You: ")
    response = generate_text(prompt, config["model"], config["temperature"], config["max_tokens"])
    print("ChitGPT: " + response)

if __name__ == "__main__":
    main()

其中，我们将 API key 存储在一个名为 `openai_api_key.json` 的 JSON 文件中，并使用 `json.load()` 函数将其读取。而模型和其他参数则存储在一个名为 `chitgpt_config.json` 的 JSON 文件中，并同样使用 `json.load()` 函数将其读取。