import openai
import os
import streamlit as st

# ========== 设置 OpenAI API Key ==========
client = openai.Client(api_key=os.getenv("OPENAI_API_KEY"))

# ========== 生成个性化邮件的函数 ==========
def generate_email(profile1_text, profile2_text):
    """根据用户输入的 LinkedIn 信息生成个性化邮件"""
    prompt = f"""
    你是一个 AI 助手，专门帮助职场人士通过电子邮件邀请别人进行 coffee chat。  
    现在有两个人的 LinkedIn 信息，你需要根据以下两个信息生成一封个性化且高质量的 Coffee Chat 邀请邮件：

    📌 Profile 1:
    {profile1_text}

    📌 Profile 2:
    {profile2_text}

    生成的邮件内容必须具有以下特点：
    1. 礼貌、简洁且专业；
    2. 清楚表达双方的共同点并礼貌地提出邀请；
    3. 语气亲切自然，避免模板化，内容更具个性化。

    ✉️ 生成的邮件内容如下：
    """

    response = client.completions.create(
        model="gpt-4",
        prompt=prompt,
        temperature=0.7,
        max_tokens=350
    )
    
    return response['choices'][0]['text'].strip()

# ========== 页面设计 ==========
st.set_page_config(page_title="AI Cold Email Generator", layout="centered")
st.title("📬 AI Cold Email Generator")

# ========== 用户输入 LinkedIn 信息 ==========
st.write("请输入两个人的 LinkedIn 信息，系统将生成个性化的邮件。")

profile1_text = st.text_area("输入第一个人的 LinkedIn 信息", height=250)
profile2_text = st.text_area("输入第二个人的 LinkedIn 信息", height=250)

# ========== 生成邮件 ==========
if st.button("🚀 生成邮件"):
    if profile1_text and profile2_text:
        with st.spinner("正在生成个性化邮件..."):
            email = generate_email(profile1_text, profile2_text)
            st.success("✅ 邮件生成成功！")
            st.text_area("📨 Coffee Chat 邮件内容", email, height=300)
    else:
        st.warning("请完整输入两个人的 LinkedIn 信息。")