Update config/settings.py

config/settings.py

@@ -1,20 +1,33 @@
-# src/config/settings.py
+# config/settings.py
 
 import os
 from pathlib import Path
+import streamlit as st
+
+# Check if we're running on HuggingFace Space
+IS_HF_SPACE = os.getenv('SPACE_ID') is not None
 
 # Base directories
-BASE_DIR = Path(__file__).resolve().parent.parent.parent
+if IS_HF_SPACE:
+    # Use streamlit's cache directory for persistent storage in HF Spaces
+    BASE_DIR = Path(st.runtime.get_instance().cache_path)
+else:
+    BASE_DIR = Path(__file__).resolve().parent.parent
+
+# Data directories
 DATA_DIR = BASE_DIR / "data"
 CHAT_HISTORIES_DIR = DATA_DIR / "chat_histories"
 CHAT_IMAGES_DIR = DATA_DIR / "chat_images"
 
 # Create directories if they don't exist
-CHAT_HISTORIES_DIR.mkdir(parents=True, exist_ok=True)
-CHAT_IMAGES_DIR.mkdir(parents=True, exist_ok=True)
+for directory in [DATA_DIR, CHAT_HISTORIES_DIR, CHAT_IMAGES_DIR]:
+    directory.mkdir(parents=True, exist_ok=True)
+    if IS_HF_SPACE:
+        # Ensure directories are writable in HF environment
+        os.chmod(directory, 0o777)
 
-# API Settings
-ANTHROPIC_API_KEY = os.getenv("ANTHROPIC_API_KEY")
+# API Settings - Get from HuggingFace secrets
+ANTHROPIC_API_KEY = os.getenv('ANTHROPIC_API_KEY')
 CLAUDE_MODEL = "claude-3-opus-20240229"
 
 # App Settings