Update app.py

app.py

@@ -1,13 +1,13 @@
 import streamlit as st
-import openai
 import logging
-import os
+import requests
+import json
 from typing import Optional
 
 # 配置日志
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
 
-def generate_email(
+def generate_email_with_vveai(
     profile1: str, 
     profile2: str, 
     style: str = "friendly", 
@@ -15,14 +15,14 @@ def generate_email(
     api_key: Optional[str] = None
 ) -> str:
     """
-    基于两个LinkedIn简介生成个性化咖啡聊天邀请邮件。
+    使用VVEAI API基于两个LinkedIn简介生成个性化咖啡聊天邀请邮件。
     
     参数:
         profile1 (str): 发件人的LinkedIn简介文本
         profile2 (str): 收件人的LinkedIn简介文本
         style (str): 邮件风格，可选项为 "formal", "friendly", "humorous"
         temperature (float): 生成文本的创造性程度 (0.0-1.0)
-        api_key (str, optional): OpenAI API密钥，如果未提供则使用环境变量
+        api_key (str, optional): VVEAI API密钥
         
     返回:
         str: 生成的个性化咖啡聊天邮件
@@ -41,9 +41,9 @@ def generate_email(
             logging.warning(f"温度 {temperature} 超出范围，调整至 0.7")
             temperature = 0.7
         
-        # 设置API密钥
-        if api_key:
-            openai.api_key = api_key
+        # 验证API密钥
+        if not api_key:
+            raise ValueError("请提供VVEAI API密钥")
             
         # 构建提示词
         prompt = f"""
@@ -80,27 +80,57 @@ def generate_email(
         Format the response as a complete, ready-to-send email without explanations.
         """
         
-        # 调用OpenAI API
-        response = openai.chat.completions.create(
-            model="gpt-4",
-            messages=[
-                {"role": "system", "content": "You are an expert email writer who helps professionals create personalized networking messages."},
+        # 准备API请求
+        system_message = "You are an expert email writer who helps professionals create personalized networking messages."
+        
+        # VVEAI API端点
+        url = "https://api.vveai.com/v1/chat/completions"
+        
+        # 构建请求
+        headers = {
+            "Content-Type": "application/json",
+            "Authorization": f"Bearer {api_key}"
+        }
+        
+        payload = {
+            "model": "gpt-4",  # 可能需要根据VVEAI支持的模型调整
+            "messages": [
+                {"role": "system", "content": system_message},
                 {"role": "user", "content": prompt}
             ],
-            temperature=temperature,
-            max_tokens=1000
-        )
+            "temperature": temperature,
+            "max_tokens": 1000
+        }
+        
+        # 发送请求
+        response = requests.post(url, headers=headers, data=json.dumps(payload), timeout=60)
         
-        # 提取并返回生成的邮件
-        email_text = response.choices[0].message.content.strip()
-        return email_text
+        # 添加详细日志便于调试
+        logging.info(f"API响应状态码: {response.status_code}")
+        logging.info(f"API响应头: {response.headers}")
+        
+        # 处理响应
+        if response.status_code == 200:
+            result = response.json()
+            # 记录完整响应便于调试
+            logging.info(f"API响应内容: {result}")
+            try:
+                email_text = result["choices"][0]["message"]["content"].strip()
+                return email_text
+            except KeyError as e:
+                logging.error(f"响应格式错误: {e}. 完整响应: {result}")
+                return f"API响应格式错误: {e}。请联系VVEAI支持获取帮助。"
+        else:
+            error_message = f"API请求失败: {response.status_code} - {response.text}"
+            logging.error(error_message)
+            return f"API错误: {response.status_code}。请检查您的API密钥是否正确。详细信息: {response.text}"
         
     except ValueError as e:
         logging.error(f"输入错误: {str(e)}")
         return f"发生错误: {str(e)}"
-    except openai.OpenAIError as e:
-        logging.error(f"OpenAI API错误: {str(e)}")
-        return f"OpenAI API错误: {str(e)}"
+    except requests.RequestException as e:
+        logging.error(f"请求错误: {str(e)}")
+        return f"网络请求错误: {str(e)}"
     except Exception as e:
         logging.error(f"意外错误: {str(e)}")
         return f"发生未预期的错误: {str(e)}"
@@ -117,19 +147,50 @@ st.title("☕ 个性化咖啡聊天邮件生成器")
 st.markdown("""
 此应用可以帮助求职者生成个性化的咖啡聊天邀请邮件。
 只需输入您的简介和目标联系人的简介，系统将分析共同点并生成一封真诚的邀请邮件。
+
+**注意：此应用使用 VVEAI API 服务，请确保输入正确的 API 密钥。**
 """)
 
-# 侧边栏 - API密钥设置
+# 侧边栏 - API设置
 with st.sidebar:
-    st.header("设置")
-    api_key = st.text_input("OpenAI API密钥", type="password", key="api_key")
-    if api_key:
-        os.environ["OPENAI_API_KEY"] = api_key
-    else:
-        # 尝试从环境变量获取
-        api_key = os.environ.get("OPENAI_API_KEY")
+    st.header("API设置")
+    
+    api_key = st.text_input("VVEAI API密钥", type="password", key="api_key", 
+                           help="输入以 sk- 开头的 VVEAI API 密钥")
+    
+    # 添加API状态检测按钮
+    if st.button("测试API连接"):
         if not api_key:
-            st.warning("请输入OpenAI API密钥以使用此应用")
+            st.error("请先输入API密钥")
+        else:
+            try:
+                # 简单的API测试请求
+                test_url = "https://api.vveai.com/v1/chat/completions"
+                test_headers = {
+                    "Content-Type": "application/json",
+                    "Authorization": f"Bearer {api_key}"
+                }
+                test_payload = {
+                    "model": "gpt-4",
+                    "messages": [
+                        {"role": "user", "content": "Hello"}
+                    ],
+                    "max_tokens": 5
+                }
+                
+                test_response = requests.post(
+                    test_url, 
+                    headers=test_headers, 
+                    data=json.dumps(test_payload),
+                    timeout=10
+                )
+                
+                if test_response.status_code == 200:
+                    st.success("API连接成功！")
+                else:
+                    st.error(f"API连接失败: {test_response.status_code} - {test_response.text}")
+            except Exception as e:
+                st.error(f"测试API时出错: {str(e)}")
     
     st.subheader("邮件风格")
     style = st.radio(
@@ -169,14 +230,14 @@ with col2:
 submit_disabled = not api_key or not profile1 or not profile2
 if submit_disabled:
     if not api_key:
-        reason = "请先输入OpenAI API密钥"
+        reason = "请先输入VVEAI API密钥"
     else:
         reason = "请填写两个简介"
     st.warning(reason)
 
 if st.button("生成邮件", type="primary", disabled=submit_disabled):
     with st.spinner("正在分析简介并创建个性化邮件..."):
-        email = generate_email(
+        email = generate_email_with_vveai(
             profile1=profile1,
             profile2=profile2,
             style=style,
@@ -184,17 +245,33 @@ if st.button("生成邮件", type="primary", disabled=submit_disabled):
             api_key=api_key
         )
         
-    st.success("邮件已生成！")
+    # 检查是否发生错误
+    if email.startswith("发生错误") or email.startswith("API错误") or email.startswith("网络请求错误"):
+        st.error(email)
+    else:
+        st.success("邮件已生成！")
+        
+        # 显示结果
+        st.header("您的个性化咖啡聊天邮件")
+        st.text_area("可复制的邮件内容", email, height=400)
+        st.download_button(
+            label="下载邮件",
+            data=email,
+            file_name="咖啡聊天邀请.txt",
+            mime="text/plain"
+        )
+
+# 调试信息（可选，帮助解决API问题）
+if st.checkbox("显示调试信息"):
+    st.subheader("API信息")
+    st.markdown("""
+    - API端点: `https://api.vveai.com/v1/chat/completions`
+    - 请求方法: POST
+    - 内容类型: application/json
+    - 授权方式: Bearer Token
     
-    # 显示结果
-    st.header("您的个性化咖啡聊天邮件")
-    st.text_area("可复制的邮件内容", email, height=400)
-    st.download_button(
-        label="下载邮件",
-        data=email,
-        file_name="咖啡聊天邀请.txt",
-        mime="text/plain"
-    )
+    如果遇到持续的API问题，请联系VVEAI支持团队并提供以上信息。
+    """)
 
 # 使用提示
 st.markdown("---")
@@ -204,4 +281,5 @@ st.markdown("""
 - **尝试不同风格**以适应不同的行业和联系人
 - 生成后可以**微调邮件内容**，添加您认为合适的个人细节
 - 适合用于**金融、咨询和科技行业**的专业人士
+- 如果API连接测试失败，请确认您的**VVEAI API密钥**是否正确
 """)