Upload 2 files

rag_engine.py

@@ -1,3 +1,5 @@
+import os
+import zipfile
 import time
 import json
 import re
@@ -14,6 +16,14 @@ from utils import normalize_arabic, detect_lang
 
 class ENAEngine:
     def __init__(self, groq_token=None):
+        # Auto-extract DB if missing
+        if os.path.exists("chroma_ena_db.zip") and not os.path.exists(CHROMA_PATH):
+            try:
+                with zipfile.ZipFile("chroma_ena_db.zip", 'r') as zip_ref:
+                    zip_ref.extractall(".")
+            except:
+                pass
+
         self.embeddings = HuggingFaceEmbeddings(
             model_name=EMBED_MODEL, 
             model_kwargs={"device":"cpu"},