Add code to unzip data in HF space

.gitignore

@@ -1,3 +1,4 @@
 data/
 .DS_Store
-__pycache__
+__pycache__
+data.zip

app.py

@@ -2,17 +2,21 @@ import gradio as gr
 import numpy as np
 from PIL import Image
 from engine.search import ImageSearchModule
+from engine.utils import unzip_file
 import os
 from pathlib import Path
 
 PROJECT_ROOT = Path(__file__).resolve().parent
 
+
 def check_dirs():
     dirs = {
         "Data": (PROJECT_ROOT / "data"),
         "Images": (PROJECT_ROOT / "data" / "images"),
         "Features": (PROJECT_ROOT / "data" / "features")
     }
+    if not dirs['Data'].exists():
+        unzip_file(zip_file_path=str(PROJECT_ROOT / "data.zip"), destination_folder=str(PROJECT_ROOT ))
     for dir_name, dir_path in dirs.items():
         if not dir_path.exists():
             raise FileNotFoundError(f"{dir_name} directory not found: {dir_path}")

engine/utils.py

@@ -0,0 +1,29 @@
+import zipfile
+import os
+def unzip_file(zip_file_path: str, destination_folder: str) -> None:
+    """Unzips a ZIP file to the specified destination folder.
+
+    Args:
+        zip_file_path: The path to the ZIP file.
+        destination_folder: The path to the folder where the contents should be extracted.
+    """
+    
+    try:
+        # Check if the ZIP file exists
+        if not os.path.isfile(zip_file_path):
+            raise FileNotFoundError(f"ZIP file not found: {zip_file_path}")
+
+        # Check if the destination folder exists and create it if necessary
+        os.makedirs(destination_folder, exist_ok=True)
+
+        # Extract the ZIP file
+        with zipfile.ZipFile(zip_file_path, 'r') as zip_ref:
+            zip_ref.extractall(destination_folder)
+        print(f"Successfully unzipped '{zip_file_path}' to '{destination_folder}'")
+    
+    except zipfile.BadZipFile:
+        print(f"Error: '{zip_file_path}' is not a valid ZIP file.")
+    except FileNotFoundError as e:
+        print(e) # Print the error message for FileNotFoundError
+    except Exception as e:  # Catch any other unexpected errors
+        print(f"An unexpected error occurred: {e}")