Update app.py

app.py

@@ -109,9 +109,9 @@ def classify_zip_and_analyze_color(zip_file):
                 except Exception:
                     continue
 
-                # Thumbnail for gallery (higher-quality)
+                # Thumbnail for gallery (keep aspect ratio)
                 thumb = image.copy()
-                thumb = thumb.resize((200, 200), Image.LANCZOS)
+                thumb.thumbnail((200, 200), Image.LANCZOS)
                 thumbnails.append(thumb)
 
                 # Classification
@@ -258,7 +258,14 @@ demo = gr.Interface(
     outputs=[
         gr.Dataframe(headers=["Filename", "Top 3 Predictions", "Confidence", "Dominant Color", "Basic Color", "Face Info"]),
         gr.File(label="Download XLSX"),
-        gr.Gallery(label="Thumbnails", show_label=True, elem_id="thumbnail-gallery", columns=5),
+        gr.Gallery(
+            label="Thumbnails",
+            show_label=True,
+            elem_id="thumbnail-gallery",
+            columns=5,
+            object_fit="contain",   # no warping
+            square=False            # keep aspect ratio
+        ),
         gr.Image(type="pil", label="Basic Color Frequency"),
         gr.Image(type="pil", label="Top Prediction Distribution"),
         gr.Image(type="pil", label="Gender Distribution (Weighted ≤90%)"),