Update app.py

app.py

@@ -106,37 +106,40 @@ def predict(image):
         frame_recommendations = recommend_glasses_tree(pred_label)
 
         description = face_shape_descriptions[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)
+        frame_image_path = glasses_images.get(pred_label)
 
         # Build explanation text
-        explanation = (f"**Bentuk Wajah:** {pred_label} ({pred_prob:.2f}%)\n\n"
-                       f"Kamu memiliki bentuk wajah {description}.\n\n"
-                       f"Rekomendasi kacamata yang cocok dengan wajah kamu:\n")
-        
-        return explanation, frame_images
+        if frame_recommendations:
+            recommended_frames = ', '.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}.")
+        else:
+            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
 
     except Exception as e:
-        return "Error: " + str(e), []
+        return "Error", str(e), None
 
 # Gradio Interface
 iface = gr.Interface(
     fn=predict,
     inputs=gr.Image(type="pil"),
     outputs=[
-        gr.Markdown(label="Penjelasan dan Rekomendasi"),  # Display explanation in markdown
-        gr.Gallery(label="Frame Kacamata Rekomendasi", show_label=False)  # Display gallery without labels
+        gr.Textbox(label="Bentuk Wajah Terdeteksi"),
+        gr.Textbox(label="Rekomendasi dan Penjelasan"),
+        gr.Image(label="Gambar Frame Rekomendasi")
     ],
     title="Rekomendasi Kacamata Berdasarkan Bentuk Wajah",
-    description="Upload foto wajahmu untuk mendapatkan rekomendasi bentuk kacamata yang sesuai dengan wajah kamu!",
-    layout="vertical",  # Stack components vertically
-    theme="compact"  # Compact theme for a cleaner design
+    description="Upload foto wajahmu untuk mendapatkan rekomendasi bentuk kacamata yang sesuai!"
 )
 
 if __name__ == "__main__":