Update app.py

app.py

@@ -119,7 +119,7 @@ def predict(image):
         frame_recommendations = recommend_glasses_tree(pred_label)
 
         description = face_shape_descriptions.get(pred_label, "tidak dikenali")
-        frame_images = []
+        gallery_items = []
 
         # Load images for all recommended frames
         for frame in frame_recommendations:
@@ -127,21 +127,21 @@ def predict(image):
             if frame_image_path and os.path.exists(frame_image_path):
                 try:
                     frame_image = Image.open(frame_image_path)
-                    frame_images.append(frame_image)
+                    gallery_items.append((frame_image, frame))  # Tambahkan nama frame
                 except Exception as e:
                     print(f"Error loading image for {frame}: {e}")
 
         # Build explanation text
         if frame_recommendations:
-            recommended_frames = ', '.join(frame_recommendations)
+            recommended_frames_text = ', '.join(frame_recommendations)
             explanation = (f"Bentuk wajah kamu adalah {pred_label} ({pred_prob:.2f}%). "
                            f"Kamu memiliki bentuk wajah {description} "
-                           f"Rekomendasi bentuk kacamata yang sesuai dengan wajah kamu adalah: {recommended_frames}.")
+                           f"Rekomendasi bentuk kacamata yang sesuai dengan wajah kamu adalah: {recommended_frames_text}.")
         else:
             explanation = (f"Bentuk wajah kamu adalah {pred_label} ({pred_prob:.2f}%). "
                            f"Tidak ada rekomendasi frame untuk bentuk wajah ini.")
 
-        return pred_label, explanation, frame_images
+        return pred_label, explanation, gallery_items
 
     except ValueError as ve:
         return "Error", str(ve), []
@@ -156,17 +156,15 @@ with gr.Blocks(theme=gr.themes.Soft()) as iface:
     with gr.Row():
         with gr.Column():
             image_input = gr.Image(type="pil")
-            with gr.Row():
-                upload_button = gr.UploadButton("Unggah Gambar", file_types=["image"])
-                clear_button = gr.Button("Ganti")
+            upload_button = gr.Button("Unggah Gambar")
+            clear_button = gr.Button("Ganti")
         with gr.Column():
             detected_shape = gr.Textbox(label="Bentuk Wajah Terdeteksi")
             explanation_output = gr.Textbox(label="Penjelasan")
-            recommendation_gallery = gr.Gallery(label="Rekomendasi Kacamata", columns=3)
+            recommendation_gallery = gr.Gallery(label="Rekomendasi Kacamata", columns=3, show_label=False)
 
-    upload_button.upload(predict, inputs=upload_button, outputs=[detected_shape, explanation_output, recommendation_gallery])
+    upload_button.click(predict, inputs=image_input, outputs=[detected_shape, explanation_output, recommendation_gallery])
     clear_button.click(lambda: (None, "", []), inputs=None, outputs=[image_input, detected_shape, recommendation_gallery])
-    image_input.upload(predict, inputs=image_input, outputs=[detected_shape, explanation_output, recommendation_gallery])
 
 if __name__ == "__main__":
     iface.launch()