Update app.py

app.py

@@ -1,3 +1,4 @@
+
 import gradio as gr
 import pandas as pd
 import numpy as np
@@ -56,7 +57,7 @@ def load_model(model_path):
 
 def deck_to_ids(deck, mapping):
     """Convert card names to IDs based on the mapping."""
-    return [mapping.get(card, 0) - 1 for card in deck]  # Zero-based indices
+    return [mapping.get(card, 0) - 1 for card in deck]
 
 def preprocess_deck(deck):
     """Prepare the selected deck for the model."""
@@ -81,13 +82,22 @@ model = load_model(MODEL_PATH)
 with gr.Blocks() as interface:
     gr.Markdown("## Clash Royale Prediction")
     gr.Markdown("Select 8 cards from the opponent's deck to predict the probability of winning!")
-    
-    opponent_deck = gr.CheckboxGroup(
-        choices=list(card_images.keys()),
-        label="Opponent's Deck",
-        info="Select exactly 8 cards.",
-        item_text=lambda name: f'<img src="{card_images[name]}" alt="{name}" width="50" height="50" /> {name}'
-    )
+
+    selected_deck = []
+    for card_name, image_url in card_images.items():
+        def toggle_card(card_name):
+            if card_name in selected_deck:
+                selected_deck.remove(card_name)
+            else:
+                selected_deck.append(card_name)
+
+        gr.Image(
+            value=image_url,
+            label=card_name,
+            tool="select",
+            interactive=True,
+            click=toggle_card(card_name),
+        )
     
     result = gr.Textbox(label="Prediction Result:", interactive=False)
 
@@ -97,6 +107,6 @@ with gr.Blocks() as interface:
         return predict_outcome(deck)
 
     predict_button = gr.Button("Make Prediction")
-    predict_button.click(validate_and_predict, inputs=[opponent_deck], outputs=[result])
+    predict_button.click(validate_and_predict, inputs=[selected_deck], outputs=[result])
 
 interface.launch()