Update Astronomy_BH_hybrid_RAG.py

Astronomy_BH_hybrid_RAG.py

@@ -23,50 +23,48 @@ def download_blob(bucket_name, source_blob_name, destination_file_name):
     blob.download_to_filename(destination_file_name)
     print(f"Downloaded storage object {source_blob_name} from bucket {bucket_name} to local file {destination_file_name}.")
 
-if not (os.path.exists("storage/bm25") and os.path.exists("storag/kg")):
-    # List of file names to download
-    file_names = [
-        "default__vector_store.json",
-        "docstore.json",
-        "graph_store.json",
-        "image__vector_store.json",
-        "index_store.json"
-    ]
-
-    # Bucket name
-    bucket_name = "title_tailors_bucket"
-
-    # Create the destination directory if it doesn't exist
-    os.makedirs("storage/bm25", exist_ok=True)
-
-    # Loop through the file names and download each one
-    for file_name in file_names:
-        source_blob_name = f"storage/bm25/{file_name}"
-        destination_file_name = f"storage/bm25/{file_name}"
-        download_blob(bucket_name, source_blob_name, destination_file_name)
-        
-    # List of file names to download
-    file_names = [
-        "default__vector_store.json",
-        "docstore.json",
-        "graph_store.json",
-        "image__vector_store.json",
-        "index_store.json"
-    ]
-
-    # Bucket name
-    bucket_name = "title_tailors_bucket"
-
-    # Create the destination directory if it doesn't exist
-    os.makedirs("storage/kg", exist_ok=True)
-
-    # Loop through the file names and download each one
-    for file_name in file_names:
-        source_blob_name = f"storage/kg/{file_name}"
-        destination_file_name = f"storage/kg/{file_name}"
-        download_blob(bucket_name, source_blob_name, destination_file_name)
-else:
-    print("Files already exist in the storage_file directory.")
+# List of file names to download
+file_names = [
+    "default__vector_store.json",
+    "docstore.json",
+    "graph_store.json",
+    "image__vector_store.json",
+    "index_store.json"
+]
+
+# Bucket name
+bucket_name = "title_tailors_bucket"
+
+# Create the destination directory if it doesn't exist
+os.makedirs("storage/bm25", exist_ok=True)
+
+# Loop through the file names and download each one
+for file_name in file_names:
+    source_blob_name = f"storage/bm25/{file_name}"
+    destination_file_name = f"storage/bm25/{file_name}"
+    download_blob(bucket_name, source_blob_name, destination_file_name)
+    
+# List of file names to download
+file_names = [
+    "default__vector_store.json",
+    "docstore.json",
+    "graph_store.json",
+    "image__vector_store.json",
+    "index_store.json"
+]
+
+# Bucket name
+bucket_name = "title_tailors_bucket"
+
+# Create the destination directory if it doesn't exist
+os.makedirs("storage/kg", exist_ok=True)
+
+# Loop through the file names and download each one
+for file_name in file_names:
+    source_blob_name = f"storage/kg/{file_name}"
+    destination_file_name = f"storage/kg/{file_name}"
+    download_blob(bucket_name, source_blob_name, destination_file_name)
+
 
 
 HF_TOKEN = os.environ.get("HF_TOKEN", None)