from Source.Predict import predict
from flask import Flask, render_template, jsonify, request
import requests
import pandas as pd
import numpy as np

# update past picks
try:
    predict.update_past_predictions()
except KeyError as e:
    print("Couldn't update past predictions.")

# get week, season
week, season = predict.get_week()

app = Flask(__name__)
app.secret_key = 'green-flounder'

games = predict.get_games()[['Date','Away Team','Home Team']]

@app.route('/')
def index():
    return render_template('index.html')

@app.route('/get_games')
def get_games():
    return jsonify(games.to_dict(orient='records'))

@app.route('/submit_games', methods=['POST'])
def submit_games():
    data = request.json
    data = pd.DataFrame(data).replace('', np.nan).dropna()
    home_teams = data['HomeTeam'].values
    away_teams = data['AwayTeam'].values
    ou_lines = data['OverUnderLine'].values

    moneylines = []
    over_unders = []
    for home,away,total in zip(home_teams,away_teams,ou_lines):
        moneyline, over_under = predict.predict(home,away,season,week,total)
        moneylines.append(moneyline)
        over_unders.append(over_under)
    
    return jsonify({'moneylines': moneylines,
                    'over_unders': over_unders})
    
if __name__ == '__main__':
    app.run(host='0.0.0.0', port='7860')