Fix image_utils.py to handle all images in single directory

image_utils.py

@@ -1,22 +1,33 @@
-import base64
+import base64
 import os
 
 def get_base64_images():
     images = {}
     static_dir = 'static/images'
     
-    # Logo and banner
-    with open(os.path.join(static_dir, 'logo.png'), 'rb') as f:
-        images['logo'] = base64.b64encode(f.read()).decode()
-    
-    with open(os.path.join(static_dir, 'banner.png'), 'rb') as f:
-        images['banner'] = base64.b64encode(f.read()).decode()
-    
-    # Crop images
-    crops_dir = os.path.join(static_dir, 'crops')
-    for image_file in os.listdir(crops_dir):
-        if image_file.endswith(('.jpg', '.png')):
-            with open(os.path.join(crops_dir, image_file), 'rb') as f:
-                images[image_file.split('.')[0]] = base64.b64encode(f.read()).decode()
-    
-    return images 
+    try:
+        # Create static/images directory if it doesn't exist
+        os.makedirs(static_dir, exist_ok=True)
+        
+        print(f"Looking for images in: {static_dir}")
+        
+        # Load all image files in the directory
+        if os.path.exists(static_dir):
+            for image_file in os.listdir(static_dir):
+                if image_file.lower().endswith(('.png', '.jpg', '.jpeg')):
+                    try:
+                        with open(os.path.join(static_dir, image_file), 'rb') as f:
+                            # Get filename without extension as the key
+                            name = os.path.splitext(image_file)[0].lower()
+                            images[name] = base64.b64encode(f.read()).decode()
+                            print(f"Loaded: {image_file}")
+                    except Exception as e:
+                        print(f"Error loading {image_file}: {e}")
+        else:
+            print(f"Warning: Directory not found: {static_dir}")
+                    
+    except Exception as e:
+        print(f"Error in get_base64_images: {e}")
+        print(f"Current working directory: {os.getcwd()}")
+        
+    return images