Update app.py

app.py

@@ -10,21 +10,23 @@ import os
 LLAMA_INDEX_DATASET_ID = os.getenv("HF_INDEX_DATASET_ID", "alperensn/llamaIndexVectorBase_fda")
 LLAMA_INDEX_SUBDIR = os.getenv("HF_INDEX_SUBDIR", "").strip()  # dataset içinde alt klasör kullanıyorsan burada belirt
 
+# Eski ve yeni (default__) adlandırmaları birlikte kontrol edelim
+MARKERS_CLASSIC = {"index_store.json", "docstore.json", "graph_store.json", "default__vector_store.json", "image__vector_store.json" }
+MARKERS_DEFAULT = {"default__index_store.json", "default__docstore.json", "default__vector_store.json", "default_image__vector_store.json","default__graph_store.json"}
+
+
 def _persist_path(base_dir: str) -> str:
     return os.path.join(base_dir, LLAMA_INDEX_SUBDIR) if LLAMA_INDEX_SUBDIR else base_dir
 
 def llama_index_exists(base_dir: str) -> bool:
-    """LlamaIndex'in persist formatındaki belirgin dosyalardan birine bak."""
     path = _persist_path(base_dir)
-    markers = ["index_store.json", "docstore.json", "vector_store.json"]
-    return any(os.path.exists(os.path.join(path, m)) for m in markers)
+    if not os.path.isdir(path):
+        return False
+    files = set(os.listdir(path))
+    return (MARKERS_CLASSIC.issubset(files) or MARKERS_DEFAULT.issubset(files))
+
 
 def download_llama_index_if_needed(base_dir: str):
-    """
-    Persist edilmiş LlamaIndex dosyaları yerelde yoksa dataset'ten indir.
-    - Public dataset ise token gerekmiyor
-    - Private ise HUGGINGFACEHUB_API_TOKEN secret'ı otomatik kullanılır
-    """
     path = _persist_path(base_dir)
     os.makedirs(path, exist_ok=True)
     if llama_index_exists(base_dir):
@@ -34,10 +36,22 @@ def download_llama_index_if_needed(base_dir: str):
         repo_type="dataset",
         local_dir=path,
         local_dir_use_symlinks=False,
-        # gerekirse allow_patterns ile kısıtlayabilirsin:
-        # allow_patterns=["*.json", "*.bin", "*.pkl", "*.npy"]
     )
-    
+
+# İndirilen LlamaIndex persist klasörünü altlarda aramak gerekirse:
+def find_llama_index_dir(base_dir: str) -> str:
+    wanted_sets = [MARKERS_CLASSIC, MARKERS_DEFAULT]
+    if os.path.isdir(base_dir):
+        files = set(os.listdir(base_dir))
+        if any(ws.issubset(files) for ws in wanted_sets):
+            return base_dir
+    for root, _, files in os.walk(base_dir):
+        files = set(files)
+        if any(ws.issubset(files) for ws in wanted_sets):
+            return root
+    return base_dir
+
+
 # Load environment variables from .env file
 load_dotenv()
 
@@ -128,6 +142,9 @@ def main():
                 try:
                     status.write(f"Downloading persisted index from: {LLAMA_INDEX_DATASET_ID}")
                     download_llama_index_if_needed(config.LLAMA_INDEX_STORE_PATH)
+                    detected_dir = find_llama_index_dir(config.LLAMA_INDEX_STORE_PATH)
+                    if detected_dir != config.LLAMA_INDEX_STORE_PATH:
+                        config.LLAMA_INDEX_STORE_PATH = detected_dir
                     status.update(label="Index downloaded from dataset.", state="complete", expanded=False)
                     time.sleep(1)
                 except Exception as e: