Update utils.py

utils.py

@@ -7,31 +7,49 @@ import os
 from PIL import Image
 from collections import Counter
 
+# File paths for data storage (compatible with pre-existing folders)
+DATA_DIR = 'data'
+USERS_FILE = os.path.join(DATA_DIR, 'users.json')
+LABELS_FILE = os.path.join(DATA_DIR, 'labels.json')
+IMAGES_DIR = os.path.join(DATA_DIR, 'images')
+
 def load_users():
-    """Load users from JSON file"""
-    try:
-        with open('data/users.json', 'r') as f:
-            return json.load(f)
-    except (FileNotFoundError, json.JSONDecodeError):
-        return {}
+    """Load users from JSON file with error handling for pre-existing files"""
+    if os.path.exists(USERS_FILE):
+        try:
+            with open(USERS_FILE, 'r') as f:
+                return json.load(f)
+        except (FileNotFoundError, json.JSONDecodeError, PermissionError):
+            pass
+    return {}
 
 def save_users(users):
-    """Save users to JSON file"""
-    with open('data/users.json', 'w') as f:
-        json.dump(users, f, indent=2)
+    """Save users to JSON file with error handling for read-only environments"""
+    if os.path.exists(DATA_DIR):
+        try:
+            with open(USERS_FILE, 'w') as f:
+                json.dump(users, f, indent=2)
+        except (OSError, PermissionError):
+            pass
 
 def load_labels():
-    """Load labels from JSON file"""
-    try:
-        with open('data/labels.json', 'r') as f:
-            return json.load(f)
-    except (FileNotFoundError, json.JSONDecodeError):
-        return {}
+    """Load labels from JSON file with error handling for pre-existing files"""
+    if os.path.exists(LABELS_FILE):
+        try:
+            with open(LABELS_FILE, 'r') as f:
+                return json.load(f)
+        except (FileNotFoundError, json.JSONDecodeError, PermissionError):
+            pass
+    return {}
 
 def save_labels(labels):
-    """Save labels to JSON file"""
-    with open('data/labels.json', 'w') as f:
-        json.dump(labels, f, indent=2)
+    """Save labels to JSON file with error handling for read-only environments"""
+    if os.path.exists(DATA_DIR):
+        try:
+            with open(LABELS_FILE, 'w') as f:
+                json.dump(labels, f, indent=2)
+        except (OSError, PermissionError):
+            pass
 
 def validate_image(uploaded_file):
     """Validate uploaded image file"""