import streamlit as st
st.set_page_config(layout="wide")
import numpy as np
import pandas as pd
import pymongo
import time

@st.cache_resource
def init_conn():
        
        uri = st.secrets['mongo_uri']
        client = pymongo.MongoClient(uri, retryWrites=True, serverSelectionTimeoutMS=500000)
        db = client["League_of_Legends_Database"]

        collection = db["gamelogs"] 
        team_names = collection.distinct("teamname")
        player_names = collection.distinct("playername")

        return db, team_names, player_names
    
db, team_names, player_names = init_conn()

@st.cache_data(ttl = 60)
def init_team_data(team):  
        
        collection = db["gamelogs"] 
        cursor = collection.find({"teamname": team})
    
        raw_display = pd.DataFrame(list(cursor))
        raw_display = raw_display[['playername', 'teamname', 'playername_avg_kill_share_win', 'playername_avg_death_share_win', 'playername_avg_assist_share_win', 'playername_avg_cs_share_win', 'playername_avg_kill_share_loss', 'playername_avg_death_share_loss', 'playername_avg_assist_share_loss', 'playername_avg_cs_share_loss']]

        team_data = raw_display.drop_duplicates(subset = ['playername'])

        return team_data

st.table(init_team_data("T1"))