Update app.py

app.py

@@ -229,48 +229,14 @@ def classify_gallery(flux_loras):
         
         for item in sorted_gallery:
             if "image" in item and "title" in item:
-                image_url = item["image"]
+                image_path = item["image"]
                 title = item["title"]
                 
-                # Try to load local images with PIL
-                if isinstance(image_url, str) and image_url.startswith("examples/"):
-                    try:
-                        import os
-                        # Try different possible paths
-                        possible_paths = [
-                            image_url,
-                            os.path.join(os.getcwd(), image_url),
-                            f"/home/user/app/{image_url}"
-                        ]
-                        
-                        image_loaded = False
-                        for path in possible_paths:
-                            if os.path.exists(path):
-                                try:
-                                    pil_image = Image.open(path)
-                                    gallery_items.append((pil_image, title))
-                                    image_loaded = True
-                                    print(f"✓ Successfully loaded image from: {path}")
-                                    break
-                                except Exception as e:
-                                    print(f"Failed to open image at {path}: {e}")
-                        
-                        if not image_loaded:
-                            print(f"✗ Could not load image: {image_url}")
-                            # Use the original path as fallback
-                            gallery_items.append((image_url, title))
-                    except Exception as e:
-                        print(f"Error processing image {image_url}: {e}")
-                        gallery_items.append((image_url, title))
-                else:
-                    # For URLs or other paths, use as-is
-                    gallery_items.append((image_url, title))
-        
-        if not gallery_items:
-            print("No gallery items found")
-            return [], sorted_gallery
+                # Simply use the path as-is for Gradio to handle
+                gallery_items.append((image_path, title))
+                print(f"Added to gallery: {image_path} - {title}")
         
-        print(f"Gallery loaded with {len(gallery_items)} items")
+        print(f"Total gallery items: {len(gallery_items)}")
         return gallery_items, sorted_gallery
     except Exception as e:
         print(f"Error in classify_gallery: {e}")
@@ -726,4 +692,4 @@ with gr.Blocks(css=css) as demo:
     )
 
 demo.queue(default_concurrency_limit=None)
-demo.launch(allowed_paths=["examples/"])
+demo.launch()