Update src/streamlit_app.py

src/streamlit_app.py

@@ -33,7 +33,24 @@ except ImportError:
 st.set_page_config(page_title="Cybersecurity AI Assistant (Hugging Face RAG & Batch Analysis)", page_icon="🛡️", layout="wide")
 st.title("🛡️ Meta-Llama-3-8B-Instruct with FAISS RAG & Batch Analysis (Inference Client)")
 st.markdown("已啟用：**IndexFlatIP** + **L2 正規化** + **Hugging Face Inference Client (API)**。上傳 JSON 執行批量分析，上傳其他檔案作為 RAG 參考庫。")
+if 'execute_batch_analysis' not in st.session_state:
+    st.session_state.execute_batch_analysis = False
+    
+if 'batch_results' not in st.session_state:
+    st.session_state.batch_results = None # 確保初始化
 
+if 'rag_current_file_key' not in st.session_state:
+    st.session_state.rag_current_file_key = None
+    
+if 'json_current_file_key' not in st.session_state:
+    st.session_state.json_current_file_key = None
+    
+# 確保所有用於存儲狀態的變量都已初始化，例如：
+if 'vector_store' not in st.session_state:
+    st.session_state.vector_store = None
+    
+if 'json_data_for_batch' not in st.session_state:
+    st.session_state.json_data_for_batch = None
 # 設定模型 ID (替換為您指定的模型)
 MODEL_ID = "Zennar/Deecon-SecurityAnalyst-1.5B" ## meta-llama/Llama-4-Scout-17B-16E-Instruct
 WINDOW_SIZE = 8