Update app.py

app.py

@@ -2,21 +2,22 @@ import zipfile
 import sys
 import os
 
-# Paths to the ZIP file and the extraction directory
-zip_file_path = '/home/user/app/models.zip'
-extract_dir = '/home/user/app/models/'
-
-# Ensure the extraction directory exists (create it if it doesn't)
-if not os.path.exists(extract_dir):
-    os.makedirs(extract_dir)
-
-# Unzipping the file
-try:
-    with zipfile.ZipFile(zip_file_path, 'r') as zip_ref:
-        zip_ref.extractall(extract_dir)
-        print(f"Files in '{extract_dir}' directory:", os.listdir(extract_dir))
-except Exception as e:
-    print(f"An error occurred: {e}")
+
+def unzip_file(zip_path, extract_to=None):
+    # Check if extract_to is provided, otherwise extract in the current directory
+    if extract_to is None:
+        extract_to = os.path.dirname(zip_path)  # Extract in the same directory as the zip file
+    
+    # Open the zip file
+    with zipfile.ZipFile(zip_path, 'r') as zip_ref:
+        # Extract all the contents into the extract_to directory
+        zip_ref.extractall(extract_to)
+        print(f"Extracted all files to {extract_to}")
+
+
+zip_file_path = 'models.zip'  # Replace with your zip file path
+unzip_file(zip_file_path)  # This will extract to the same directory as the zip file
+
 
 # Check current directory and list files
 print("Current Directory:", os.getcwd())