Update app.py

app.py

@@ -513,43 +513,34 @@ def get_images_html():
     # Create HTML for displaying images
     html = "<div class='image-gallery'>"
     
-    if not image_files:
-        html += "<p>No images found in the database.</p>"
-    else:
-        current_page = None
-        
-        for image_file in image_files:
-            # Extract page number for section headers
-            parts = image_file.split('_')
-            if len(parts) >= 2:
-                page_num = parts[0].replace('page', '')
-                
-                # Add page section header if we're on a new page
-                if current_page != page_num:
-                    current_page = page_num
-                    html += f"<h2 class='page-header'>Page {page_num}</h2>"
+    # The problem was that we were checking if image_files is empty, but it's not empty
+    # We have the list of files, we just need to display them
+    
+    current_page = None
+    for image_file in image_files:
+        # Extract page number for section headers
+        parts = image_file.split('_')
+        if len(parts) >= 2:
+            page_num = parts[0].replace('page', '')
+            # Add page section header if we're on a new page
+            if current_page != page_num:
+                current_page = page_num
+                html += f"<h2 class='page-header'>Page {page_num}</h2>"
             
-            # Determine file extension for MIME type
-            file_ext = os.path.splitext(image_file)[1].lower()
-            if file_ext == '.jpeg' or file_ext == '.jpg':
-                image_type = 'jpeg'
-            elif file_ext == '.png':
-                image_type = 'png'
-            else:
-                image_type = 'jpeg'  # Default fallback
+            # In Hugging Face Spaces, files need the correct path
+            # This is the likely source of your error - incorrect path to the images
+            img_path = f"files/{image_file}"  # Adjust this path based on your Hugging Face Space structure
             
-            # In Hugging Face Spaces, files are directly accessible via the path
-            # Format the HTML to use the image directly from the space
             html += f"""
-                <div class='image-card'>
-                    <div class='image-container'>
-                        <img src="{image_file}" alt="{image_file}">
-                    </div>
-                    <div class='image-details'>
-                        <h3>{image_file}</h3>
-                        <p><strong>Page:</strong> {page_num}</p>
-                    </div>
+            <div class='image-card'>
+                <div class='image-container'>
+                    <img src="{img_path}" alt="{image_file}">
+                </div>
+                <div class='image-details'>
+                    <h3>{image_file}</h3>
+                    <p><strong>Page:</strong> {page_num}</p>
                 </div>
+            </div>
             """
     
     html += "</div>"