Update app.py

app.py

@@ -12,6 +12,20 @@ import xml.etree.ElementTree as ET
 
 load_dotenv()
 
+MEMORY_STORAGE_TYPE = "HF_DATASET" 
+# --- DEMO --- #
+# If using HF_DATASET, specify the repository names here.
+# These will override the .env file settings.
+HF_DATASET_MEMORY_REPO = "broadfield-dev/ai-brain"
+HF_DATASET_RULES_REPO = "broadfield-dev/ai-rules"
+
+# Set environment variables based on the toggles above BEFORE importing other modules
+os.environ['STORAGE_BACKEND'] = MEMORY_STORAGE_TYPE
+if MEMORY_STORAGE_TYPE == "HF_DATASET":
+    os.environ['HF_MEMORY_DATASET_REPO'] = HF_DATASET_MEMORY_REPO
+    os.environ['HF_RULES_DATASET_REPO'] = HF_DATASET_RULES_REPO
+# --- END CONFIGURATION ---
+
 from model_logic import (
     get_available_providers, get_model_display_names_for_provider,
     get_default_model_display_name_for_provider, call_model_stream, MODELS_BY_PROVIDER