Upload app.py

app.py

@@ -15,13 +15,13 @@ games_api = cfbd.GamesApi(api_config)
 betting_api = cfbd.BettingApi(api_config)
 
 
-def predict(game_details):
+def predict(team, year):
+    matchups = get_game_details(year, team)
     learn = load_learner("cfb-prediction.pkl")
 
-    pdf = pd.DataFrame([game_details])
+    pdf = pd.DataFrame(matchups)
     dl = learn.dls.test_dl(pdf)
     pdf["predicted"] = learn.get_preds(dl=dl)[0].numpy()
-    pdf['margin'] = abs(game_details['home_points'] - game_details['away_points'])
     
     return pdf[['home_team','away_team','home_points','away_points','spread','margin','predicted']]
 
@@ -34,43 +34,43 @@ def predict(game_details):
 
 def home_team_change(team, year):
     games = games_api.get_games(team=team, year=year)
-    names = []
-    for game in games:
-        if game.home_team == team:
-            names.append(game.away_team)
-        else:
-            names.append(game.home_team)
 
-    return gr.update(choices=names, value=None)
+    return games
 
 
-def get_game_details(year, target, opponent):
+def get_game_details(year, target):
     games = games_api.get_games(team=target, year=year)
-    the_game = null
+    lines = betting_api.get_lines(team=target, year=year)
+    matchups = []
 
     for game in games:
-        if game.home_team == opponent or game.away_team == opponent:
-            the_game = dict(
-                id=game.id,
-                year=game.season,
-                week=game.week,
-                neutral_site=game.neutral_site,
-                home_team=game.home_team,
-                home_conference=game.home_conference,
-                home_points=game.home_points,
-                home_elo=game.home_pregame_elo,
-                away_team=game.away_team,
-                away_conference=game.away_conference,
-                away_points=game.away_points,
-                away_elo=game.away_pregame_elo,
-            )
-            break
-            
-    spread = betting_api.get_lines(game_id=the_game['id'],year=year)
+        the_game = dict(
+            id=game.id,
+            year=game.season,
+            week=game.week,
+            neutral_site=game.neutral_site,
+            home_team=game.home_team,
+            home_conference=game.home_conference,
+            home_points=game.home_points,
+            home_elo=game.home_pregame_elo,
+            away_team=game.away_team,
+            away_conference=game.away_conference,
+            away_points=game.away_points,
+            away_elo=game.away_pregame_elo,
+        )
+        game_lines = [l for l in lines if l.id == the_game['id']]
+
+        if len(game_lines) > 0:
+            if len(game_lines[0].lines) > 0:
+                game_line = game_lines[0].lines[0]
+                the_game['spread'] = float(game_line.spread)
 
-    the_game['spread'] = spread[0].lines[0].spread
+        the_game['margin'] = the_game['away_points'] - the_game['home_points']                
+     
+        matchups.append(the_game)
 
-    return the_game
+
+    return matchups
 
 def lookup_teams(teams_api):
     team_names = []
@@ -96,14 +96,6 @@ with gr.Blocks() as app:
     with gr.Row():
         with gr.Column(scale=1):
             ddTeams1 = gr.Dropdown(choices=teams, label="Target Team")
-        with gr.Column(scale=1):
-            oponents = gr.Dropdown(choices=[], label="Opponents", interactive=True)
-    #     with gr.Column(scale=1):
-    #         ddTeams2 = gr.Dropdown(choices=teams, label="Away Team")
-    #         label2 = (gr.Label(num_top_classes=4, label="Away Team Details"))
-    #         ddTeams2.select(home_team_change, inputs=ddTeams2, outputs=[label2])
-    with gr.Row():
-            game_details = gr.JSON({})
     with gr.Row():
         submit = gr.Button("Predict")
     with gr.Row():
@@ -114,14 +106,8 @@ with gr.Blocks() as app:
             col_count=(7, "dynamic")
         )
 
-    ddTeams1.select(home_team_change, inputs=[ddTeams1, yearDd], outputs=[oponents])
-    oponents.select(
-        get_game_details,
-        inputs=[yearDd, ddTeams1, oponents],
-        outputs=[game_details]
-    )
     submit.click(
-        predict, inputs=[game_details], outputs=[data_table]
+        predict, inputs=[ddTeams1, yearDd], outputs=[data_table]
     )
 app.launch()