Write python function to fix error

api/api.py

@@ -37,6 +37,12 @@ def get_adalflow_default_root_path():
     """
     Get the adalflow root path. In containerized environments, use a writable location.
     """
+    # First check if ADALFLOW_CACHE_DIR environment variable is set
+    env_cache_dir = os.environ.get("ADALFLOW_CACHE_DIR")
+    if env_cache_dir:
+        os.makedirs(env_cache_dir, exist_ok=True)
+        return env_cache_dir
+    
     # Check if we're in a containerized environment or if file system is read-only
     default_path = os.path.expanduser(os.path.join("~", ".adalflow"))
     

api/config.py

@@ -4,6 +4,29 @@ import logging
 import re
 from pathlib import Path
 from typing import List, Union, Dict, Any
+from dotenv import load_dotenv
+
+# Load environment variables from .env file at module import time
+load_dotenv()
+
+# Ensure adalflow cache directory is set for containerized environments
+# This prevents permission errors when running in Hugging Face Spaces or similar environments
+if not os.environ.get("ADALFLOW_CACHE_DIR"):
+    try:
+        # Check if we can use the default home directory
+        default_path = os.path.expanduser(os.path.join("~", ".adalflow"))
+        os.makedirs(default_path, exist_ok=True)
+        # Test write permissions
+        test_file = os.path.join(default_path, "test_write")
+        with open(test_file, "w") as f:
+            f.write("test")
+        os.remove(test_file)
+        os.environ["ADALFLOW_CACHE_DIR"] = default_path
+    except (PermissionError, OSError):
+        # Use /tmp if home directory is not writable
+        fallback_path = "/tmp/adalflow"
+        os.makedirs(fallback_path, exist_ok=True)
+        os.environ["ADALFLOW_CACHE_DIR"] = fallback_path
 
 logger = logging.getLogger(__name__)
 
@@ -12,7 +35,10 @@ from api.openrouter_client import OpenRouterClient
 from api.bedrock_client import BedrockClient
 from api.azureai_client import AzureAIClient
 from api.dashscope_client import DashscopeClient
-from adalflow import GoogleGenAIClient, OllamaClient
+# Import adalflow after environment setup to prevent path permission errors
+import adalflow
+GoogleGenAIClient = adalflow.GoogleGenAIClient
+OllamaClient = adalflow.OllamaClient
 
 # Get API keys from environment variables
 OPENAI_API_KEY = os.environ.get('OPENAI_API_KEY')

api/main.py

@@ -7,6 +7,80 @@ from dotenv import load_dotenv
 # Load environment variables from .env file
 load_dotenv()
 
+def setup_adalflow_paths():
+    """
+    Set up adalflow paths for containerized environments like Hugging Face Spaces.
+    This prevents permission errors when adalflow tries to create directories in restricted locations.
+    """
+    try:
+        # Check if we're in a containerized environment with restricted filesystem
+        default_path = os.path.expanduser(os.path.join("~", ".adalflow"))
+        
+        # Try to create the default path first
+        try:
+            os.makedirs(default_path, exist_ok=True)
+            # Test if we can write to it
+            test_file = os.path.join(default_path, "test_write")
+            with open(test_file, "w") as f:
+                f.write("test")
+            os.remove(test_file)
+            # If successful, set the environment variables to this path
+            os.environ["ADALFLOW_CACHE_DIR"] = default_path
+            os.environ["ADALFLOW_HOME"] = default_path
+            # Set XDG variables which many libraries use for cache directories
+            os.environ["XDG_DATA_HOME"] = default_path
+            os.environ["XDG_CACHE_HOME"] = default_path
+            return default_path
+        except (PermissionError, OSError):
+            # Fall back to using a writable location in /tmp
+            fallback_path = "/tmp/adalflow"
+            try:
+                os.makedirs(fallback_path, exist_ok=True)
+                os.environ["ADALFLOW_CACHE_DIR"] = fallback_path
+                os.environ["ADALFLOW_HOME"] = fallback_path
+                # Set XDG variables for fallback
+                os.environ["XDG_DATA_HOME"] = fallback_path
+                os.environ["XDG_CACHE_HOME"] = fallback_path
+                # Also override HOME if we're in a restricted environment
+                if not os.path.exists(os.path.expanduser("~")) or not os.access(os.path.expanduser("~"), os.W_OK):
+                    os.environ["HOME"] = "/tmp"
+                return fallback_path
+            except (PermissionError, OSError):
+                # Last resort: use current working directory
+                current_dir_path = os.path.join(os.getcwd(), ".adalflow")
+                os.makedirs(current_dir_path, exist_ok=True)
+                os.environ["ADALFLOW_CACHE_DIR"] = current_dir_path
+                os.environ["ADALFLOW_HOME"] = current_dir_path
+                # Set XDG variables for last resort
+                os.environ["XDG_DATA_HOME"] = current_dir_path
+                os.environ["XDG_CACHE_HOME"] = current_dir_path
+                return current_dir_path
+    except Exception as e:
+        print(f"Error setting up adalflow paths: {e}")
+        # Set a default fallback
+        fallback_path = "/tmp/adalflow"
+        os.makedirs(fallback_path, exist_ok=True)
+        os.environ["ADALFLOW_CACHE_DIR"] = fallback_path
+        os.environ["ADALFLOW_HOME"] = fallback_path
+        # Set XDG variables for emergency fallback
+        os.environ["XDG_DATA_HOME"] = fallback_path
+        os.environ["XDG_CACHE_HOME"] = fallback_path
+        # If we can't access home, set it to /tmp
+        try:
+            test_home = os.path.expanduser("~")
+            if not os.path.exists(test_home) or not os.access(test_home, os.W_OK):
+                os.environ["HOME"] = "/tmp"
+        except:
+            os.environ["HOME"] = "/tmp"
+        return fallback_path
+
+# Set up adalflow paths before any imports that might use adalflow
+adalflow_path = setup_adalflow_paths()
+print(f"Adalflow path configured to: {adalflow_path}")
+print(f"HOME environment variable: {os.environ.get('HOME')}")
+print(f"ADALFLOW_HOME: {os.environ.get('ADALFLOW_HOME')}")
+print(f"ADALFLOW_CACHE_DIR: {os.environ.get('ADALFLOW_CACHE_DIR')}")
+
 from api.logging_config import setup_logging
 
 # Configure logging