Update src/streamlit_app.py

src/streamlit_app.py

@@ -2,6 +2,7 @@ import streamlit as st
 from openai import OpenAI
 import time
 import json # 导入json库
+import os   # 导入os库用于读取环境变量
 
 # =======================================================================
 # 1. 导入您的类和数据
@@ -11,7 +12,7 @@ import json # 导入json库
 from ms_patient import MsPatient 
 
 # 从本地JSON文件加载数据集的函数
-def load_data_from_json(filepath="/app/src/CPsyCounS-3134.json"):
+def load_data_from_json(filepath="Anna-CPsyCounD.json"):
     """
     从本地的JSON文件加载数据集。
     请确保您已将数据集文件上传到与此应用相同的目录中。
@@ -31,7 +32,6 @@ def load_data_from_json(filepath="/app/src/CPsyCounS-3134.json"):
         return []
 
 # 加载数据
-# 注意：现在每次脚本重新运行时都会从本地JSON文件加载
 ALL_PATIENTS = load_data_from_json()
 
 
@@ -79,49 +79,40 @@ st.markdown("""
 """, unsafe_allow_html=True)
 
 
-# --- 初始化 Session State ---
+# --- 初始化 Session State 和 OpenAI Client ---
+# 仅在会话状态中不存在时，才从环境变量初始化客户端
+if "openai_client" not in st.session_state:
+    api_key = os.getenv("OPENAI_API_KEY")
+    if not api_key:
+        st.session_state.openai_client = None
+        st.session_state.model_name = None
+    else:
+        try:
+            st.session_state.openai_client = OpenAI(api_key=api_key, base_url=os.getenv("OPENAI_BASE_URL"))
+            st.session_state.model_name = os.getenv("OPENAI_MODEL_NAME", "gpt-4o-mini")
+        except Exception as e:
+            st.error(f"初始化OpenAI客户端失败: {e}")
+            st.session_state.openai_client = None
+            st.session_state.model_name = None
+
 if "patient_agent" not in st.session_state:
     st.session_state.patient_agent = None
 if "messages" not in st.session_state:
     st.session_state.messages = []
 if "selected_patient_id" not in st.session_state:
     st.session_state.selected_patient_id = None
-if "openai_client" not in st.session_state:
-    st.session_state.openai_client = None
-if "model_name" not in st.session_state:
-    st.session_state.model_name = "gpt-4o-mini" # 默认模型
+
 
 # --- 侧边栏 ---
 with st.sidebar:
     st.title("👩 AnnaAgent 设置")
     st.markdown("---")
 
-    # API Key 输入
-    with st.expander("🔑 API 设置", expanded=True):
-        api_key = st.text_input("输入您的 OpenAI API Key", type="password", help="您的API Key将仅用于本次会话，不会被储存。")
-        base_url = st.text_input("API Base URL (可选)", value="https://api.openai.com/v1")
-        model_name = st.text_input("模型名称", value=st.session_state.model_name)
-
-        if st.button("连接模型"):
-            if api_key:
-                try:
-                    st.session_state.openai_client = OpenAI(api_key=api_key, base_url=base_url)
-                    st.session_state.model_name = model_name
-                    st.success("连接成功！")
-                    if st.session_state.patient_agent:
-                        st.session_state.patient_agent.client = st.session_state.openai_client
-                except Exception as e:
-                    st.error(f"连接失败: {e}")
-            else:
-                st.warning("请输入API Key。")
-
-    st.markdown("---")
-
     # 病人选择
     if not ALL_PATIENTS:
         st.error("无法加载病人数据。请检查JSON文件是否已上传且格式正确。")
     else:
-        patient_options = {p["id"]: f"{p['portrait']['gender']}，{p['portrait']['age']}岁 - {p['portrait']['symptoms']}" for p in ALL_PATIENTS}
+        patient_options = {p["id"]: f"{p['portrait']['gender']}，{p['portrait']['age']}岁 - {p['portrait']['symptom']}" for p in ALL_PATIENTS}
         selected_id = st.selectbox(
             "选择一位病人进行对话",
             options=list(patient_options.keys()),
@@ -162,24 +153,28 @@ with st.sidebar:
 st.title("💬 与 Anna 对话")
 st.caption("这是一个模拟心理咨询的AI Agent。由 `MsPatient` 类驱动。")
 
-# 显示聊天记录
-for message in st.session_state.messages:
-    avatar = "👩" if message["role"] == "assistant" else "🧑‍⚕️"
-    with st.chat_message(message["role"], avatar=avatar):
-        st.markdown(message["content"])
-
-# 获取用户输入
-if prompt := st.chat_input("请输入您想说的话..."):
-    st.session_state.messages.append({"role": "user", "content": prompt})
-    with st.chat_message("user", avatar="🧑‍⚕️"):
-        st.markdown(prompt)
-
-    if st.session_state.patient_agent:
-        with st.chat_message("assistant", avatar="👩"):
-            with st.spinner("Anna正在思考..."):
-                response = st.session_state.patient_agent.chat(prompt)
-                st.markdown(response)
-        
-        st.session_state.messages.append({"role": "assistant", "content": response})
-    else:
-        st.warning("请先在左侧选择一位病人并配置API Key。")
+# 检查API Key是否已在后台设置
+if not st.session_state.openai_client:
+    st.error("后台未设置 OPENAI_API_KEY。请在Hugging Face Space的'Settings' -> 'Secrets'中进行设置后刷新页面。")
+else:
+    # 显示聊天记录
+    for message in st.session_state.messages:
+        avatar = "👩" if message["role"] == "assistant" else "🧑‍⚕️"
+        with st.chat_message(message["role"], avatar=avatar):
+            st.markdown(message["content"])
+
+    # 获取用户输入
+    if prompt := st.chat_input("请输入您想说的话..."):
+        st.session_state.messages.append({"role": "user", "content": prompt})
+        with st.chat_message("user", avatar="🧑‍⚕️"):
+            st.markdown(prompt)
+
+        if st.session_state.patient_agent:
+            with st.chat_message("assistant", avatar="👩"):
+                with st.spinner("Anna正在思考..."):
+                    response = st.session_state.patient_agent.chat(prompt)
+                    st.markdown(response)
+            
+            st.session_state.messages.append({"role": "assistant", "content": response})
+        else:
+            st.warning("请先在左侧选择一位病人。")