Critical fixes for Docker deployment

1. Fixed cache directory permissions:
- Added cache directory setup in Dockerfile with proper permissions
- Added HF_HOME and TRANSFORMERS_CACHE environment variables
- Added cache directory initialization in app startup

2. Fixed type annotation compatibility:
- Removed @st .cache_resource decorators from class methods
- Fixed Python 3.9 compatibility issues with union types
- Updated TranslationService and ABSAProcessor model loading

These fixes should resolve the permission denied and type annotation errors.

Dockerfile

@@ -9,6 +9,11 @@ RUN apt-get update && apt-get install -y \
     git \
     && rm -rf /var/lib/apt/lists/*
 
+# Create cache directory with proper permissions
+RUN mkdir -p /.cache && chmod 777 /.cache
+ENV HF_HOME=/.cache/huggingface
+ENV TRANSFORMERS_CACHE=/.cache/huggingface/transformers
+
 # Copy requirements first for better caching
 COPY requirements-docker.txt ./requirements.txt
 

app_enhanced.py

@@ -8,6 +8,33 @@ import pandas as pd
 import sys
 import os
 
+# Setup cache directories for Docker environment
+def setup_cache_directories():
+    """Setup cache directories with proper permissions for containerized environment."""
+    cache_dirs = [
+        os.path.expanduser("~/.cache"),
+        os.path.expanduser("~/.cache/huggingface"),
+        os.path.expanduser("~/.cache/huggingface/transformers"),
+        "/.cache",
+        "/.cache/huggingface", 
+        "/.cache/huggingface/transformers"
+    ]
+    
+    for cache_dir in cache_dirs:
+        try:
+            os.makedirs(cache_dir, exist_ok=True)
+            # Try to create a test file to check write permissions
+            test_file = os.path.join(cache_dir, "test_write.tmp")
+            with open(test_file, 'w') as f:
+                f.write("test")
+            os.remove(test_file)
+        except (PermissionError, OSError):
+            # If we can't write to this directory, continue with others
+            continue
+
+# Setup cache directories at startup
+setup_cache_directories()
+
 # Add src directory to path for imports
 sys.path.append(os.path.join(os.path.dirname(__file__), 'src'))
 

src/utils/data_processor.py

@@ -86,15 +86,14 @@ class TranslationService:
         self.tokenizer = None
         self._load_model()
     
-    @st.cache_resource
-    def _load_model(_self):
+    def _load_model(self):
         """Load M2M100 model for translation."""
         try:
             from transformers import M2M100ForConditionalGeneration, M2M100Tokenizer
             
             model_name = "facebook/m2m100_418M"
-            _self.tokenizer = M2M100Tokenizer.from_pretrained(model_name)
-            _self.model = M2M100ForConditionalGeneration.from_pretrained(model_name)
+            self.tokenizer = M2M100Tokenizer.from_pretrained(model_name)
+            self.model = M2M100ForConditionalGeneration.from_pretrained(model_name)
             
             logger.info("Translation model loaded successfully")
         except Exception as e:
@@ -180,13 +179,12 @@ class ABSAProcessor:
         self.aspect_extractor = None
         self._load_model()
     
-    @st.cache_resource
-    def _load_model(_self):
+    def _load_model(self):
         """Load pyABSA model."""
         try:
             from pyabsa import ATEPCCheckpointManager
             
-            _self.aspect_extractor = ATEPCCheckpointManager.get_aspect_extractor(
+            self.aspect_extractor = ATEPCCheckpointManager.get_aspect_extractor(
                 checkpoint='multilingual',
                 auto_device=True,
                 task_code='ATEPC'