Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,7 +1,7 @@
 import streamlit as st
 from openai import OpenAI
 import time
-from datasets import load_dataset # 导入Hugging Face datasets库
+import json # 导入json库
 
 # =======================================================================
 # 1. 导入您的类和数据
@@ -10,25 +10,29 @@ from datasets import load_dataset # 导入Hugging Face datasets库
 # 请将 'your_agent_file' 替换为包含 MsPatient 类的实际文件名
 from ms_patient import MsPatient 
 
-# 从Hugging Face加载数据集的函数（移除了缓存）
-def load_hf_dataset():
+# 从本地JSON文件加载数据集的函数
+def load_data_from_json(filepath="Anna-CPsyCounD.json"):
     """
-    从Hugging Face Hub加载数据集。
-    这将返回一个字典列表，每个字典代表一个病人数据。
+    从本地的JSON文件加载数据集。
+    请确保您已将数据集文件上传到与此应用相同的目录中。
     """
     try:
-        # 加载'train'分割部分的数据
-        # cache_dir=None 可以尝试避免写入受限目录，但最可靠的方式是完全移除缓存
-        dataset = load_dataset("sci-m-wang/Anna-CPsyCounD", split='train', cache_dir=None)
-        # 转换为pandas DataFrame再转为字典列表，方便处理
-        return dataset.to_pandas().to_dict('records')
+        with open(filepath, 'r', encoding='utf-8') as f:
+            # 假设JSON文件的根是一个包含病人记录的列表
+            return json.load(f)
+    except FileNotFoundError:
+        st.error(f"错误：找不到数据文件 '{filepath}'。请确保您已将该文件上传到Hugging Face Space。")
+        return []
+    except json.JSONDecodeError:
+        st.error(f"错误：无法解析 '{filepath}'。请检查文件是否为有效的JSON格式。")
+        return []
     except Exception as e:
-        st.error(f"从Hugging Face加载数据集失败: {e}")
+        st.error(f"加载数据时发生未知错误: {e}")
         return []
 
 # 加载数据
-# 注意：现在每次脚本重新运行时都会调用此函数
-ALL_PATIENTS = load_hf_dataset()
+# 注意：现在每次脚本重新运行时都会从本地JSON文件加载
+ALL_PATIENTS = load_data_from_json()
 
 
 # =======================================================================
@@ -38,7 +42,7 @@ ALL_PATIENTS = load_hf_dataset()
 # --- 页面配置 ---
 st.set_page_config(
     page_title="与Anna对话",
-    page_icon="👩",
+    page_icon="�",
     layout="wide"
 )
 
@@ -115,7 +119,7 @@ with st.sidebar:
 
     # 病人选择
     if not ALL_PATIENTS:
-        st.error("无法加载病人数据，请检查网络连接或数据集名称。")
+        st.error("无法加载病人数据。请检查JSON文件是否已上传且格式正确。")
     else:
         patient_options = {p["id"]: f"{p['portrait']['gender']}，{p['portrait']['age']}岁 - {p['portrait']['symptom']}" for p in ALL_PATIENTS}
         selected_id = st.selectbox(