fix bug

Dockerfile

@@ -28,7 +28,10 @@ COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
 COPY . .
-RUN mkdir -p logs uploads chroma_data
+
+# Tạo thư mục với permissions đúng cho ChromaDB và logs
+RUN mkdir -p logs uploads chroma_data chroma_db && \
+    chmod -R 777 logs uploads chroma_data chroma_db
 
 EXPOSE 7860
 CMD ["python", "app.py"]

config.py

@@ -9,8 +9,8 @@ GEMINI_API_KEY = os.getenv("GEMINI_API_KEY", "YOUR_API_KEY_HERE")
 # Embedding model
 EMBEDDING_MODEL = "intfloat/multilingual-e5-base"
 
-# ChromaDB settings
-CHROMA_PERSIST_DIRECTORY = "chroma_db"
+# ChromaDB settings - sử dụng absolute path
+CHROMA_PERSIST_DIRECTORY = os.path.join(os.getcwd(), "chroma_db")
 COLLECTION_NAME = "nutrition_data"
 
 # RAG settings

core/embedding_model.py

@@ -3,6 +3,7 @@ from sentence_transformers import SentenceTransformer
 import chromadb
 from chromadb.config import Settings
 import uuid
+import os
 from config import EMBEDDING_MODEL, CHROMA_PERSIST_DIRECTORY, COLLECTION_NAME
 
 # Cấu hình logging
@@ -35,20 +36,42 @@ class EmbeddingModel:
         except Exception as e:
             logger.error(f"Lỗi khởi tạo model: {e}")
             # Thử với cache_folder explicit
-            import os
             cache_dir = os.getenv('SENTENCE_TRANSFORMERS_HOME', '/app/.cache/sentence-transformers')
             self.model = SentenceTransformer(EMBEDDING_MODEL, cache_folder=cache_dir, trust_remote_code=True)
             logger.info("Đã tải sentence transformer model với cache folder explicit")
         
+        # Đảm bảo thư mục ChromaDB tồn tại và có quyền ghi
+        try:
+            os.makedirs(CHROMA_PERSIST_DIRECTORY, exist_ok=True)
+            # Test ghi file để kiểm tra permission
+            test_file = os.path.join(CHROMA_PERSIST_DIRECTORY, 'test_permission.tmp')
+            with open(test_file, 'w') as f:
+                f.write('test')
+            os.remove(test_file)
+            logger.info(f"Thư mục ChromaDB đã sẵn sàng: {CHROMA_PERSIST_DIRECTORY}")
+        except Exception as e:
+            logger.error(f"Lỗi tạo/kiểm tra thư mục ChromaDB: {e}")
+            # Fallback to /tmp directory
+            import tempfile
+            CHROMA_PERSIST_DIRECTORY = os.path.join(tempfile.gettempdir(), 'chroma_db')
+            os.makedirs(CHROMA_PERSIST_DIRECTORY, exist_ok=True)
+            logger.warning(f"Sử dụng thư mục tạm thời: {CHROMA_PERSIST_DIRECTORY}")
+        
         # Khởi tạo ChromaDB client với persistent storage
-        self.chroma_client = chromadb.PersistentClient(
-            path=CHROMA_PERSIST_DIRECTORY,
-            settings=Settings(
-                anonymized_telemetry=False,
-                allow_reset=True
+        try:
+            self.chroma_client = chromadb.PersistentClient(
+                path=CHROMA_PERSIST_DIRECTORY,
+                settings=Settings(
+                    anonymized_telemetry=False,
+                    allow_reset=True
+                )
             )
-        )
-        logger.info(f"Đã kết nối ChromaDB tại: {CHROMA_PERSIST_DIRECTORY}")
+            logger.info(f"Đã kết nối ChromaDB tại: {CHROMA_PERSIST_DIRECTORY}")
+        except Exception as e:
+            logger.error(f"Lỗi kết nối ChromaDB: {e}")
+            # Fallback to in-memory client
+            logger.warning("Fallback to in-memory ChromaDB client")
+            self.chroma_client = chromadb.Client()
         
         # Lấy hoặc tạo collection
         try:
@@ -207,7 +230,6 @@ class EmbeddingModel:
             logger.error(f"Lỗi thêm documents: {e}")
             return False
     
-    # Các phương thức khác giữ nguyên...
     def index_chunks(self, chunks):
         """
         Index các chunks dữ liệu vào ChromaDB