app.py

import gradio as gr
import pandas as pd
import numpy as np
import onnxruntime as ort
import sys
from pathlib import Path
sys.path.append("rd2l_pred")
from feature_engineering import heroes, hero_information

# Define expected columns
EXPECTED_COLUMNS = ['mmr', 'p1', 'p2', 'p3', 'p4', 'p5', 'count', 'mean', 'std', 'min', 'max', 'sum',
                   'total_games_played', 'total_winrate']

# Add games columns
games_ids = list(range(1, 24)) + [25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40,
                                 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57,
                                 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74,
                                 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91,
                                 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104, 105, 106,
                                 107, 108, 109, 110, 111, 112, 113, 114, 119, 120, 121, 123, 126,
                                 128, 129, 131, 135, 136, 137, 138, 145]

EXPECTED_COLUMNS.extend([f'games_{i}' for i in games_ids])
EXPECTED_COLUMNS.extend([f'winrate_{i}' for i in games_ids])

def prepare_single_player_data(user_id, mmr, comf_1, comf_2, comf_3, comf_4, comf_5):
    """Creates a DataFrame in the expected format for the model"""
    try:
        # Extract player_id from URL if needed
        player_id = user_id.split("/")[-1] if "/" in user_id else user_id
        
        # Get hero statistics using OpenDota API
        hero_stats = hero_information(player_id)
        
        # Create initial data dictionary with zeros for all columns
        data = {col: 0 for col in EXPECTED_COLUMNS}
        
        # Fill in the basic features
        data.update({
            'mmr': float(mmr),
            'p1': int(comf_1),
            'p2': int(comf_2),
            'p3': int(comf_3),
            'p4': int(comf_4),
            'p5': int(comf_5),
        })
        
        # Add hero statistics
        if hero_stats is not None:
            data['total_games_played'] = hero_stats.get('total_games_played', 0)
            data['total_winrate'] = hero_stats.get('total_winrate', 0)
            
            # Fill in the games and winrate columns from hero_stats
            for key, value in hero_stats.items():
                if key in EXPECTED_COLUMNS:
                    data[key] = value
        
        # Add mock statistics for money-related columns
        # These would normally come from league_money function
        stats = {
            'count': 1,
            'mean': mmr / 200,  # rough approximation
            'std': mmr / 400,
            'min': mmr / 250,
            'max': mmr / 150,
            'sum': mmr / 200
        }
        data.update(stats)
        
        # Convert to DataFrame
        df = pd.DataFrame([data])
        
        # Ensure columns are in correct order
        df = df[EXPECTED_COLUMNS]
        
        print(f"DataFrame shape: {df.shape}")
        print("Missing columns:", set(EXPECTED_COLUMNS) - set(df.columns))
        
        return df
        
    except Exception as e:
        print(f"Error in data preparation: {e}")
        raise e

def predict_cost(user_id, mmr, comf_1, comf_2, comf_3, comf_4, comf_5):
    """Main prediction function for Gradio interface"""
    try:
        # Prepare the player data
        processed_data = prepare_single_player_data(user_id, mmr, comf_1, comf_2, comf_3, comf_4, comf_5)
        
        # Load and use the model
        model_path = Path("model/rd2l_forest.onnx")
        if not model_path.exists():
            return f"Model file not found at: {model_path}"
            
        session = ort.InferenceSession(str(model_path))
        
        # Debug information
        print("Processed data shape:", processed_data.shape)
        print("Processed data columns:", processed_data.columns.tolist())
        
        # Make prediction
        input_name = session.get_inputs()[0].name
        prediction = session.run(None, {input_name: processed_data.values.astype(np.float32)})[0]
        predicted_cost = round(float(prediction[0]), 2)
        
        hero_stats = processed_data.iloc[0]
        total_games = hero_stats.get('total_games_played', 'N/A')
        total_winrate = hero_stats.get('total_winrate', 'N/A')
        
        return f"""Predicted Cost: {predicted_cost}

Player Details:
- MMR: {mmr}
- Position Comfort:
  * Pos 1: {comf_1}
  * Pos 2: {comf_2}
  * Pos 3: {comf_3}
  * Pos 4: {comf_4}
  * Pos 5: {comf_5}

Player Statistics:
- Total Games: {total_games}
- Overall Winrate: {total_winrate:.1%} if isinstance(total_winrate, float) else 'N/A'

Note: This prediction is based on historical data and player statistics from OpenDota."""
        
    except Exception as e:
        return f"Error in prediction pipeline: {str(e)}\n\nDebug info:\n{type(e).__name__}: {str(e)}"

# Create Gradio interface
demo = gr.Interface(
    fn=predict_cost,
    inputs=[
        gr.Textbox(label="Player ID or Link to OpenDota/Dotabuff",
                  placeholder="Enter player ID or full profile URL"),
        gr.Number(label="MMR", value=3000),
        gr.Slider(1, 5, value=3, step=1, label="Comfort (Pos 1)"),
        gr.Slider(1, 5, value=3, step=1, label="Comfort (Pos 2)"),
        gr.Slider(1, 5, value=3, step=1, label="Comfort (Pos 3)"),
        gr.Slider(1, 5, value=3, step=1, label="Comfort (Pos 4)"),
        gr.Slider(1, 5, value=3, step=1, label="Comfort (Pos 5)")
    ],
    examples=[
        ["https://www.dotabuff.com/players/188649776", 6812, 5, 5, 4, 2, 1]
    ],
    outputs=gr.Textbox(label="Prediction Results"),
    title="RD2L Player Cost Predictor",
    description="""This tool predicts the auction cost for RD2L players based on their MMR, 
                   position comfort levels, and historical performance data from OpenDota. 
                   Enter a player's OpenDota ID or profile URL along with their current stats 
                   to get a predicted cost.""",
    article="""### How it works
                - The predictor uses machine learning trained on historical RD2L draft data
                - Player statistics are fetched from OpenDota API
                - Position comfort levels range from 1 (least comfortable) to 5 (most comfortable)
                - Predictions are based on both current stats and historical performance
                
                ### Notes
                - MMR should be the player's current solo MMR
                - Position comfort should reflect actual role experience
                - Predictions are estimates and may vary from actual draft results"""
)

if __name__ == "__main__":
    demo.launch()