Update config/settings.py

config/settings.py

@@ -9,8 +9,8 @@ IS_HF_SPACE = os.getenv('SPACE_ID') is not None
 
 # Base directories
 if IS_HF_SPACE:
-    # Use streamlit's cache directory for persistent storage in HF Spaces
-    BASE_DIR = Path(st.runtime.get_instance().cache_path)
+    # Use a directory in the HF space
+    BASE_DIR = Path('/data')
 else:
     BASE_DIR = Path(__file__).resolve().parent.parent
 
@@ -21,10 +21,19 @@ CHAT_IMAGES_DIR = DATA_DIR / "chat_images"
 
 # Create directories if they don't exist
 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)
+    try:
+        directory.mkdir(parents=True, exist_ok=True)
+        # Ensure directories are writable
+        if directory.exists():
+            os.chmod(directory, 0o777)
+    except Exception as e:
+        st.warning(f"Could not create or modify directory {directory}: {str(e)}")
+        # Fallback to temporary directory if needed
+        if str(directory).startswith(str(BASE_DIR)):
+            new_path = Path('/tmp') / directory.relative_to(BASE_DIR)
+            new_path.mkdir(parents=True, exist_ok=True)
+            if IS_HF_SPACE:
+                directory = new_path
 
 # API Settings - Get from HuggingFace secrets
 ANTHROPIC_API_KEY = os.getenv('ANTHROPIC_API_KEY')