import streamlit as st
import os
import requests
from dotenv import load_dotenv

# 環境変数のロード
load_dotenv()

# OpenAI APIキーの設定
api_key = os.getenv("OPENAI_API_KEY")

def generate_reply(message):
    headers = {
        "Authorization": f"Bearer {api_key}",
        "Content-Type": "application/json"
    }
    data = {
        "model": "gpt-4-turbo-preview",
        "messages": [
            {
                "role": "user",
                "content": message
            }
        ],
        "temperature": 1,
        "max_tokens": 256,
        "top_p": 1,
        "frequency_penalty": 0,
        "presence_penalty": 0
    }
    response = requests.post("https://api.openai.com/v1/chat/completions", headers=headers, json=data)
    if response.status_code == 200:
        return response.json()['choices'][0]['message']['content']
    else:
        return "エラーが発生しました。"

# StreamlitアプリケーションのUI
st.title('メール返信ジェネレーター')

user_input = st.text_area("メールの内容をここに入力してください:")

if st.button('返信を生成'):
    reply = generate_reply(user_input)
    st.text_area("AIによる返信:", value=reply, height=300, disabled=True)