Update app.py

app.py

@@ -1,9 +1,9 @@
 import gradio as gr
 import pandas as pd
-from xgboost import Booster, DMatrix
 import numpy as np
+from xgboost import Booster, DMatrix
 
-# Define the mapping of card names to IDs (placeholder example)
+# Generating the complete `card_images` dictionary
 card_numbers = {
     "Archers": 1, "Archer Queen": 2, "Baby Dragon": 3, "Balloon": 4, "Bandit": 5, "Barbarians": 6,
     "Bats": 7, "Battle Healer": 8, "Battle Ram": 9, "Bomber": 10, "Bowler": 11, "Bush Goblins": 12,
@@ -23,29 +23,36 @@ card_numbers = {
     "Royal Giant": 78, "Royal Hogs": 79, "Royal Recruits": 80, "Skeleton Army": 81,
     "Skeleton Barrel": 82, "Skeleton Dragons": 83, "Skeleton King": 84, "Skeletons": 85, "Sparky": 86,
     "Spear Goblins": 87, "Suspicious Bush": 88, "Three Musketeers": 89, "Valkyrie": 90,
-    "Wall Breakers": 91, "Witch": 92, "Wizard": 93, "Zappies": 94,"Bomb Tower": 95, "Cannon": 96, "Cannon Cart (broken)": 97, "Inferno Tower": 98, "Mortar": 99,
-        "Tesla": 100, "X-Bow": 101,"Barbarian Hut": 102, "Elixir Collector": 103, "Furnace": 104, "Goblin Cage": 105,
-        "Goblin Drill": 106, "Goblin Hut": 107, "Phoenix Egg": 108, "Tombstone": 109, "Arrows": 110, "Barbarian Barrel": 111, "Earthquake": 112, "Fireball": 113, "Freeze": 114,
-        "Giant Snowball": 115, "Goblin Curse": 116, "Lightning": 117, "Poison": 118, "Rage": 119, "Rocket": 120, "Royal Delivery": 121, "The Log": 122, "Tornado": 123, "Void": 124, "Zap": 125,
-        "Barbarian Barrel": 126, "Barbarian Hut": 127, "Battle Ram": 128, "Elixir Golem": 129,
-        "Elixir Golemite": 130, "Furnace": 131, "Goblin Barrel": 132, "Goblin Cage": 133, "Goblin Curse": 134,
-        "Goblin Drill": 135, "Goblin Giant": 136, "Goblin Hut": 137, "Golem": 138, "Graveyard": 139,
-        "Lava Hound": 140, "Little Prince": 141, "Mother Witch": 142, "Night Witch": 143,
-        "Phoenix Egg": 144, "Royal Delivery": 145, "Skeleton Barrel": 146, "Skeleton King": 147,
-        "Suspicious Bush": 148, "Tombstone": 149, "Witch": 150,
-        "Archers/Evolution": 155, "Barbarians/Evolution": 156, "Battle Ram/Evolution": 157,
-        "Bats/Evolution": 158, "Bomber/Evolution": 159, "Cannon/Evolution": 160,
-        "Electro Dragon/Evolution": 161, "Firecracker/Evolution": 162, "Giant Snowball/Evolution": 163,
-        "Goblin Barrel/Evolution": 164, "Goblin Cage/Evolution": 165, "Goblin Drill/Evolution": 166,
-        "Goblin Giant/Evolution": 167, "Ice Spirit/Evolution": 168, "Knight/Evolution": 169,
-        "Mega Knight/Evolution": 170, "Mortar/Evolution": 171, "Musketeer/Evolution": 172,
-        "P.E.K.K.A/Evolution": 173, "Royal Giant/Evolution": 174, "Royal Recruits/Evolution": 175,
-        "Skeletons/Evolution": 176, "Tesla/Evolution": 177, "Valkyrie/Evolution": 178,
-        "Wall Breakers/Evolution": 179, "Wizard/Evolution": 180, "Zap/Evolution": 181
+    "Wall Breakers": 91, "Witch": 92, "Wizard": 93, "Zappies": 94, "Bomb Tower": 95, "Cannon": 96, 
+    "Cannon Cart (broken)": 97, "Inferno Tower": 98, "Mortar": 99, "Tesla": 100, "X-Bow": 101,
+    "Barbarian Hut": 102, "Elixir Collector": 103, "Furnace": 104, "Goblin Cage": 105,
+    "Goblin Drill": 106, "Goblin Hut": 107, "Phoenix Egg": 108, "Tombstone": 109, "Arrows": 110, 
+    "Barbarian Barrel": 111, "Earthquake": 112, "Fireball": 113, "Freeze": 114, "Giant Snowball": 115, 
+    "Goblin Curse": 116, "Lightning": 117, "Poison": 118, "Rage": 119, "Rocket": 120, 
+    "Royal Delivery": 121, "The Log": 122, "Tornado": 123, "Void": 124, "Zap": 125,
+    "Archers/Evolution": 155, "Barbarians/Evolution": 156, "Battle Ram/Evolution": 157,
+    "Bats/Evolution": 158, "Bomber/Evolution": 159, "Cannon/Evolution": 160,
+    "Electro Dragon/Evolution": 161, "Firecracker/Evolution": 162, "Giant Snowball/Evolution": 163,
+    "Goblin Barrel/Evolution": 164, "Goblin Cage/Evolution": 165, "Goblin Drill/Evolution": 166,
+    "Goblin Giant/Evolution": 167, "Ice Spirit/Evolution": 168, "Knight/Evolution": 169,
+    "Mega Knight/Evolution": 170, "Mortar/Evolution": 171, "Musketeer/Evolution": 172,
+    "P.E.K.K.A/Evolution": 173, "Royal Giant/Evolution": 174, "Royal Recruits/Evolution": 175,
+    "Skeletons/Evolution": 176, "Tesla/Evolution": 177, "Valkyrie/Evolution": 178,
+    "Wall Breakers/Evolution": 179, "Wizard/Evolution": 180, "Zap/Evolution": 181
+}
 
+card_images = {
+    card_name: f"https://clashroyale.fandom.com/wiki/Arenas?file={card_name.replace(' ', '').replace('/', '')}Card.png"
+    for card_name in card_numbers.keys()
 }
 
-MODEL_PATH = "model.json"  
+MODEL_PATH = "model.json"
+
+def load_model(model_path):
+    """Load the saved XGBoost model."""
+    model = Booster()
+    model.load_model(model_path)
+    return model
 
 def deck_to_ids(deck, mapping):
     """Convert card names to IDs based on the mapping."""
@@ -53,68 +60,43 @@ def deck_to_ids(deck, mapping):
 
 def preprocess_deck(deck):
     """Prepare the selected deck for the model."""
-    # Convert cards to IDs
     deck_ids = deck_to_ids(deck, card_numbers)
-
-    # Perform one-hot encoding
-    num_choices = 181  # Total number of cards
+    num_choices = len(card_numbers)
     one_hot = np.zeros(num_choices, dtype=int)
-    one_hot[np.array(deck_ids)] = 1  # Set 1 for selected cards
-
-    # Add additional features (placeholder for now)
-    trophy_difference = 0  # Placeholder for trophy difference
-    elixir_leaked = 0  # Placeholder for leaked elixir
-
-    # Combine features
-    features = np.concatenate(([trophy_difference, elixir_leaked], one_hot))
+    one_hot[np.array(deck_ids)] = 1
+    features = np.concatenate(([0, 0], one_hot))
     return pd.DataFrame([features])
 
-def load_model(model_path):
-    """Load the saved XGBoost model."""
-    model = Booster()
-    model.load_model(model_path)
-    return model
-
-# Load the model at startup
-model = load_model(MODEL_PATH)
-
 def predict_outcome(opponent_deck):
     """Make a prediction based on the opponent's deck."""
-    # Prepare the opponent deck data
     deck_data = preprocess_deck(opponent_deck)
-
-    # Make the prediction
-    dmatrix = DMatrix(deck_data)  # Convert data to DMatrix format
+    dmatrix = DMatrix(deck_data)
     prediction = model.predict(dmatrix)
+    return f"Probability of Winning: {prediction[0] * 100:.2f}%"
 
-    # Interpret the prediction
-    result = f"Probability of Winning: {prediction[0] * 100:.2f}%"
-    return result
-
-# List of cards for selection
-card_list = list(card_numbers.keys())
+# Load the model
+model = load_model(MODEL_PATH)
 
-# Create the Gradio interface
+# Create Gradio Interface
 with gr.Blocks() as interface:
     gr.Markdown("## Clash Royale Prediction")
-    gr.Markdown("Select the 8 cards from the opponent's deck to predict the probability of winning!")
+    gr.Markdown("Select 8 cards from the opponent's deck to predict the probability of winning!")
     
     opponent_deck = gr.CheckboxGroup(
-        choices=card_list,
-        label="Select 8 cards from the opponent's deck:",
-        info="Select exactly 8 cards."
+        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}'
     )
     
     result = gr.Textbox(label="Prediction Result:", interactive=False)
 
     def validate_and_predict(deck):
-        """Validate the number of selected cards and make a prediction."""
         if len(deck) != 8:
             return "Error: Select exactly 8 cards."
         return predict_outcome(deck)
-    
+
     predict_button = gr.Button("Make Prediction")
     predict_button.click(validate_and_predict, inputs=[opponent_deck], outputs=[result])
 
-# Launch the interface
 interface.launch()