Update app.py

app.py

@@ -1,50 +1,67 @@
+import os
 import openai
+from openai import OpenAI
 import streamlit as st
+from PIL import Image
+import pytesseract
 
-# 设置 OpenAI API 密钥（推荐使用 secrets.toml 或 st.secrets）
-openai.api_key = st.secrets["OPENAI_API_KEY"] if "OPENAI_API_KEY" in st.secrets else "sk-..."  # ← 改为你自己的 API Key
+# 初始化 OpenAI 客户端
+client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
 
-def generate_email(profile1, profile2):
+# OCR识别
+def extract_text_from_image(uploaded_file):
+    image = Image.open(uploaded_file).convert("RGB")
+    text = pytesseract.image_to_string(image)
+    return text
+
+# 邮件生成函数
+def generate_email(profile1_text, profile2_text):
     prompt = f"""
-    根据以下两位的 LinkedIn 资料，提取共同点，并写一封自然、有礼貌的 Coffee Chat 邀请邮件。避免模板感，适度专业即可。
+    请根据以下两人的 LinkedIn Profile，识别他们的共同点，并撰写一封自然、有礼貌的 Coffee Chat 邀请邮件，避免模板式语言。
 
     📌 Profile 1:
-    {profile1}
+    {profile1_text}
 
     📌 Profile 2:
-    {profile2}
+    {profile2_text}
 
-    ✉️ 请生成一封个性化 Coffee Chat 邮件：
+    ✉️ 请生成邮件：
     """
 
-    response = openai.ChatCompletion.create(
+    response = client.chat.completions.create(
         model="gpt-4",
         messages=[
-            {"role": "system", "content": "你是一位专业、友好的冷邮件助手，擅长帮用户写好Coffee Chat邮件。"},
+            {"role": "system", "content": "你是一位专业邮件写作助手，擅长撰写冷邮件。"},
             {"role": "user", "content": prompt}
         ],
         temperature=0.7,
-        max_tokens=400
+        max_tokens=500
     )
 
     return response.choices[0].message.content.strip()
 
+# 主界面
 def main():
     st.set_page_config(page_title="AI Cold Email Generator", layout="centered")
-    st.title("🤖 AI Networking Email Generator")
-    st.write("输入两个人的 LinkedIn Profile，生成个性化 Coffee Chat 邀请邮件 ☕")
+    st.title("📬 AI Cold Email Generator")
+    st.write("上传两张 LinkedIn 截图，AI 自动识别并生成 Coffee Chat 邀请邮件。")
 
-    profile1 = st.text_area("🧑‍💼 请输入你的 LinkedIn Profile", height=150)
-    profile2 = st.text_area("🎯 请输入目标联系人的 LinkedIn Profile", height=150)
+    col1, col2 = st.columns(2)
+    with col1:
+        file1 = st.file_uploader("上传你的 LinkedIn 截图", type=["png", "jpg", "jpeg"], key="profile1")
+    with col2:
+        file2 = st.file_uploader("上传目标联系人的截图", type=["png", "jpg", "jpeg"], key="profile2")
 
     if st.button("🚀 生成邮件"):
-        if profile1 and profile2:
-            with st.spinner("AI 正在撰写邮件中，请稍候..."):
-                email = generate_email(profile1, profile2)
-                st.success("✅ 邮件生成成功！可复制使用：")
-                st.text_area("📬 Coffee Chat 邮件", email, height=200)
+        if file1 and file2:
+            with st.spinner("正在识别和撰写邮件..."):
+                profile1_text = extract_text_from_image(file1)
+                profile2_text = extract_text_from_image(file2)
+                email = generate_email(profile1_text, profile2_text)
+                st.success("✅ 邮件生成成功")
+                st.text_area("📨 邮件内容", email, height=300)
         else:
-            st.warning("请确保两个 Profile 均已填写！")
+            st.warning("请上传两张截图")
 
 if __name__ == "__main__":
     main()