Update app.py

app.py

@@ -5,19 +5,25 @@ import streamlit as st
 from PIL import Image
 import pytesseract
 
+# 手动指定 tesseract 路径（非常关键）
+pytesseract.pytesseract.tesseract_cmd = "/usr/bin/tesseract"
+
 # 初始化 OpenAI 客户端
 client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
 
-# OCR识别
-def extract_text_from_image(uploaded_file):
-    image = Image.open(uploaded_file).convert("RGB")
-    text = pytesseract.image_to_string(image)
-    return text
+# 将多张图片 OCR 提取为一个字符串
+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()
 
-# 邮件生成函数
+# GPT 生成邮件
 def generate_email(profile1_text, profile2_text):
     prompt = f"""
-    请根据以下两人的 LinkedIn Profile，识别他们的共同点，并撰写一封自然、有礼貌的 Coffee Chat 邀请邮件，避免模板式语言。
+    请根据以下两人的 LinkedIn Profile，提取共同点，并撰写一封自然、有礼貌、不模板化的 Coffee Chat 邀请邮件。
 
     📌 Profile 1:
     {profile1_text}
@@ -40,28 +46,29 @@ def generate_email(profile1_text, profile2_text):
 
     return response.choices[0].message.content.strip()
 
-# 主界面
+# Streamlit 前端
 def main():
     st.set_page_config(page_title="AI Cold Email Generator", layout="centered")
     st.title("📬 AI Cold Email Generator")
-    st.write("上传两张 LinkedIn 截图，AI 自动识别并生成 Coffee Chat 邀请邮件。")
+    st.write("上传两个人的 LinkedIn 截图（每人可多图），AI 将识别并生成 Coffee Chat 邮件")
 
     col1, col2 = st.columns(2)
+
     with col1:
-        file1 = st.file_uploader("上传你的 LinkedIn 截图", type=["png", "jpg", "jpeg"], key="profile1")
+        files1 = st.file_uploader("你的 LinkedIn 截图（支持多张）", type=["png", "jpg", "jpeg"], accept_multiple_files=True, key="profile1")
     with col2:
-        file2 = st.file_uploader("上传目标联系人的截图", type=["png", "jpg", "jpeg"], key="profile2")
+        files2 = st.file_uploader("目标联系人的截图（支持多张）", type=["png", "jpg", "jpeg"], accept_multiple_files=True, key="profile2")
 
     if st.button("🚀 生成邮件"):
-        if file1 and file2:
-            with st.spinner("正在识别和撰写邮件..."):
-                profile1_text = extract_text_from_image(file1)
-                profile2_text = extract_text_from_image(file2)
+        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("📨 邮件内容", email, height=300)
+                st.success("✅ 邮件生成成功！")
+                st.text_area("📨 Coffee Chat 邮件", email, height=300)
         else:
-            st.warning("请上传两张截图")
+            st.warning("请上传两个 Profile 的图像（至少一张）")
 
 if __name__ == "__main__":
     main()