Update tools.py

tools.py

@@ -3,22 +3,12 @@ from sefaria import get_text as sefaria_get_text, get_commentaries as sefaria_ge
 from tantivy_search import TantivySearch
 from typing import Optional
 from pydantic import BaseModel, Field
+import os 
+import gdown 
+import zipfile
 
 from app import INDEX_PATH
 
-def download_index_from_gdrive() :
-            zip_path = "index.zip"
-            url = f"https://drive.google.com/uc?id={self.gdrive_index_id}"
-            gdown.download(url, zip_path, quiet=False)  
-            with zipfile.ZipFile(zip_path, 'r') as zip_ref:
-                zip_ref.extractall(".")
-            os.remove(zip_path) 
-
-
-
-if not os.path.exists(INDEX_PATH):
-    download_index_from_gdrive():
-                
                     
 class ReadTextArgs(BaseModel):
         reference: str = Field(description="The reference to retrieve the text for. examples: בראשית א פרק א, Genesis 1:1")
@@ -89,6 +79,17 @@ class SearchArgs(BaseModel):
 
 
 index_path = INDEX_PATH
+if not os.path.exists(index_path):
+    try:
+            zip_path = "index.zip"
+            url = f"https://drive.google.com/uc?id={self.gdrive_index_id}"
+            gdown.download(url, zip_path, quiet=False)  
+            with zipfile.ZipFile(zip_path, 'r') as zip_ref:
+                zip_ref.extractall(".")
+            os.remove(zip_path)
+    except Exception as e:
+        raise Exception(e)
+                
 try:
     tantivy = TantivySearch(index_path)
     tantivy.validate_index()