Update app.py

app.py

@@ -18,11 +18,12 @@ face_shape_descriptions = {
 
 # Frame images path
 glasses_images = {
-    "Heart": "Glasses/RimlessFrame.jpg",
-    "Oblong": "Glasses/RectangleFrame.jpg",
-    "Oval": "Glasses/RoundFrame.jpg",
-    "Round": "Glasses/SquareFrame.jpg",
-    "Square": "Glasses/OvalFrame.jpg"
+    "Oval": "glasses/oval.jpg",
+    "Round": "glasses/round.jpg",
+    "Square": "glasses/square.jpg",
+    "Octagon": "glasses/octagon.jpg",
+    "Cat Eye": "glasses/cat eye.jpg",
+    "Pilot (Aviator)": "glasses/aviator.jpg"
 }
 
 id2label = {0: 'Heart', 1: 'Oblong', 2: 'Oval', 3: 'Round', 4: 'Square'}
@@ -105,7 +106,14 @@ def predict(image):
         frame_recommendations = recommend_glasses_tree(pred_label)
 
         description = face_shape_descriptions[pred_label]
-        frame_image_path = glasses_images.get(pred_label)
+        frame_images = []
+
+        # Load images for all recommended frames
+        for frame in frame_recommendations:
+            frame_image_path = glasses_images.get(frame)
+            if frame_image_path and os.path.exists(frame_image_path):
+                frame_image = Image.open(frame_image_path)
+                frame_images.append(frame_image)
 
         # Build explanation text
         if frame_recommendations:
@@ -117,16 +125,10 @@ def predict(image):
             explanation = (f"Bentuk wajah kamu adalah {pred_label} ({pred_prob:.2f}%). "
                            f"Tidak ada rekomendasi frame untuk bentuk wajah ini.")
 
-        # Load frame image if available
-        if frame_image_path and os.path.exists(frame_image_path):
-            frame_image = Image.open(frame_image_path)
-        else:
-            frame_image = None
-
-        return pred_label, explanation, frame_image
+        return pred_label, explanation, frame_images
 
     except Exception as e:
-        return "Error", str(e), None
+        return "Error", str(e), []
 
 # Gradio Interface
 iface = gr.Interface(
@@ -135,7 +137,7 @@ iface = gr.Interface(
     outputs=[
         gr.Textbox(label="Bentuk Wajah Terdeteksi"),
         gr.Textbox(label="Rekomendasi dan Penjelasan"),
-        gr.Image(label="Gambar Frame Rekomendasi")
+        gr.Gallery(label="Gambar Frame Rekomendasi", columns=3)
     ],
     title="Rekomendasi Kacamata Berdasarkan Bentuk Wajah",
     description="Upload foto wajahmu untuk mendapatkan rekomendasi bentuk kacamata yang sesuai!"