Update app.py

app.py

@@ -17,28 +17,59 @@ def get_examples(examples_dir: str = "assets/examples/ghibli-fine-tuned-sd-2.1")
     Each example is a subdirectory containing a config.json and an image file.
     Returns a list of [prompt, height, width, num_inference_steps, guidance_scale, seed, image_path].
     """
-    examples = Path(examples_dir)
+    # Check if the directory exists
+    if not os.path.exists(examples_dir) or not os.path.isdir(examples_dir):
+        raise ValueError(f"Directory {examples_dir} does not exist or is not a directory")
+    
+    # Get list of subfolder paths (e.g., 1, 2, etc.)
+    all_examples_dir = [os.path.join(examples_dir, d) for d in os.listdir(examples_dir) 
+                        if os.path.isdir(os.path.join(examples_dir, d))]
+    
     ans = []
-    for example in examples.iterdir():
-        if not example.is_dir():
+    for example_dir in all_examples_dir:
+        config_path = os.path.join(example_dir, "config.json")
+        image_path = os.path.join(example_dir, "result.png")
+        
+        # Check if config.json and result.png exist
+        if not os.path.isfile(config_path):
+            print(f"Warning: config.json not found in {example_dir}")
+            continue
+        if not os.path.isfile(image_path):
+            print(f"Warning: result.png not found in {example_dir}")
+            continue
+        
+        try:
+            with open(config_path, 'r') as f:
+                example_dict = json.load(f)
+        except (json.JSONDecodeError, IOError) as e:
+            print(f"Error reading or parsing {config_path}: {e}")
             continue
-        with open(example / "config.json") as f:
-            example_dict = json.load(f)
         
+        # Required keys for the config
         required_keys = ["prompt", "height", "width", "num_inference_steps", "guidance_scale", "seed", "image"]
         if not all(key in example_dict for key in required_keys):
+            print(f"Warning: Missing required keys in {config_path}")
+            continue
+        
+        # Verify that the image key in config.json matches 'result.png'
+        if example_dict["image"] != "result.png":
+            print(f"Warning: Image key in {config_path} does not match 'result.png'")
+            continue
+        
+        try:
+            example_list = [
+                example_dict["prompt"],
+                example_dict["height"],
+                example_dict["width"],
+                example_dict["num_inference_steps"],
+                example_dict["guidance_scale"],
+                example_dict["seed"],
+                image_path  # Use verified image path
+            ]
+            ans.append(example_list)
+        except KeyError as e:
+            print(f"Error processing {config_path}: Missing key {e}")
             continue
-
-        example_list = [
-            example_dict["prompt"],
-            example_dict["height"],
-            example_dict["width"],
-            example_dict["num_inference_steps"],
-            example_dict["guidance_scale"],
-            example_dict["seed"],
-            str(example / example_dict["image"])  # Path to the image file
-        ]
-        ans.append(example_list)
     
     if not ans:
         ans = [