Update app.py

app.py

@@ -1,25 +1,16 @@
-import os
 import openai
-from openai import OpenAI
+import os
 import streamlit as st
-from PIL import Image
-import pytesseract
-import subprocess  # ✅ 添加这一行！
-
-# ✅ 使用默认 tesseract 路径，Hugging Face 会自动识别
-client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
 
-def extract_text_from_images(uploaded_files):
-    all_text = ""
-    for uploaded_file in uploaded_files:
-        image = Image.open(uploaded_file).convert("RGB")
-        text = pytesseract.image_to_string(image)
-        all_text += text + "\n"
-    return all_text.strip()
+# ========== 设置 OpenAI API Key ==========
+client = openai.Client(api_key=os.getenv("OPENAI_API_KEY"))
 
+# ========== 生成个性化邮件的函数 ==========
 def generate_email(profile1_text, profile2_text):
+    """根据用户输入的 LinkedIn 信息生成个性化邮件"""
     prompt = f"""
-    根据以下两人的 LinkedIn Profile，提取共同点，并生成一封自然、有温度的 Coffee Chat 邀请邮件（避免模板化）：
+    你是一个 AI 助手，专门帮助职场人士通过电子邮件邀请别人进行 coffee chat。  
+    现在有两个人的 LinkedIn 信息，你需要根据以下两个信息生成一封个性化且高质量的 Coffee Chat 邀请邮件：
 
     📌 Profile 1:
     {profile1_text}
@@ -27,54 +18,39 @@ def generate_email(profile1_text, profile2_text):
     📌 Profile 2:
     {profile2_text}
 
-    ✉️ 邮件内容如下：
+    生成的邮件内容必须具有以下特点：
+    1. 礼貌、简洁且专业；
+    2. 清楚表达双方的共同点并礼貌地提出邀请；
+    3. 语气亲切自然，避免模板化，内容更具个性化。
+
+    ✉️ 生成的邮件内容如下：
     """
 
-    response = client.chat.completions.create(
+    response = client.completions.create(
         model="gpt-4",
-        messages=[
-            {"role": "system", "content": "你是一位 AI Networking 邮件助手，擅长生成 Coffee Chat 邮件"},
-            {"role": "user", "content": prompt}
-        ],
+        prompt=prompt,
         temperature=0.7,
-        max_tokens=600
+        max_tokens=350
     )
-
-    return response.choices[0].message.content.strip()
-
-def main():
-    st.set_page_config(page_title="AI Cold Email Generator", layout="centered")
-    st.title("📬 AI Cold Email Generator")
-    st.write("上传两位 LinkedIn 截图（每人可多张），系统将识别并生成 Coffee Chat 邀请邮件")
-
-    col1, col2 = st.columns(2)
-    with col1:
-        files1 = st.file_uploader("上传你的 LinkedIn 截图（支持多张）", type=["png", "jpg", "jpeg"], accept_multiple_files=True, key="profile1")
-    with col2:
-        files2 = st.file_uploader("上传目标联系人的截图（支持多张）", type=["png", "jpg", "jpeg"], accept_multiple_files=True, key="profile2")
-
-    if st.button("🚀 生成邮件"):
-        if files1 and files2:
-            with st.spinner("识别图像内容 + 生成邮件中..."):
-                profile1_text = extract_text_from_images(files1)
-                profile2_text = extract_text_from_images(files2)
-                email = generate_email(profile1_text, profile2_text)
-                st.success("✅ 邮件生成成功")
-                st.text_area("📨 Coffee Chat 邮件内容", email, height=300)
-        else:
-            st.warning("请上传双方截图")
-
-import subprocess
-
-with st.expander("🛠️ 查看 Tesseract 安装状态（调试用）"):
-    try:
-        version = subprocess.check_output(["tesseract", "--version"])
-        st.success("✅ Tesseract 已安装")
-        st.code(version.decode(), language="bash")
-    except Exception as e:
-        st.error("❌ Tesseract 未安装")
-        st.text(str(e))
-
-
-if __name__ == "__main__":
-    main()
+    
+    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 信息。")