Update app.py

app.py

@@ -15,18 +15,24 @@ with zipfile.ZipFile("images/stable_diffusion.zip","r") as zip_ref:
 
    
 def open_stable_ims(profession):
-    if len(profession) != 0:
-        dirname = 'images/stable_diffusion/'+ profession+'/'
-        images = [Image.open(os.path.join(dirname+im)).convert("RGB") for im in os.listdir(dirname)]
-        random.shuffle(images)
-        return images[:16]
+    if not profession:
+        return []
+    dirname = 'images/stable_diffusion/'+ profession+'/'
+    if not os.path.isdir(dirname):
+        return []
+    images = [Image.open(os.path.join(dirname+im)).convert("RGB") for im in os.listdir(dirname)]
+    random.shuffle(images)
+    return images[:16]
 
 def open_fair_ims(profession):
-    if len(profession) != 0:
-        dirname = 'images/fair_diffusion/' + profession+'/'
-        images = [Image.open(os.path.join(dirname+im)).convert("RGB") for im in os.listdir(dirname)]
-        random.shuffle(images)
-        return images[:16]
+    if not profession:
+        return []
+    dirname = f'images/fair_diffusion/{profession}/'
+    if not os.path.isdir(dirname):
+        return []
+    images = [Image.open(os.path.join(dirname+im)).convert("RGB") for im in os.listdir(dirname)]
+    random.shuffle(images)
+    return images[:16]