Update app.py

app.py

@@ -60,10 +60,18 @@ def recommend_mask(image):
         # Predict
         prediction = model.predict([[face_encoded, skin_encoded, size_encoded]])[0]
         return encoders["mask_style"].classes_[prediction]
+
+        # Get recommended mask image path
+        mask_image_path = encoders['mask_images'][prediction]
+        return (
+            encoders["mask_style"].classes_[prediction],  # Text
+            mask_image_path                               # Image
+        )
         
     except Exception as e:
         print(f"Prediction error: {str(e)}")
-        return f"Recommended: Basic Mask (Fallback)"
+        return "Error", "default_mask.png"  # Fallback
+
 
 # Initialize model and encoders
 model, encoders = safe_load_model()
@@ -71,8 +79,11 @@ model, encoders = safe_load_model()
 # Create Gradio interface
 demo = gr.Interface(
     fn=recommend_mask,
-    inputs=gr.Image(label="Upload Your Face", type="filepath"),
-    outputs=gr.Textbox(label="Recommended Mask Style"),
+    inputs=gr.Image(type="filepath"),
+    outputs=[
+        gr.Textbox(label="Recommended Style"),
+        gr.Image(label="Mask Preview")  # Add image output
+    ],
     title="🎭 AI Party Mask Recommender",
     description="Upload a photo to get a personalized mask recommendation!",
 )