Update app.py

app.py

@@ -4,7 +4,8 @@ from tempfile import TemporaryDirectory
 import numpy as np
 import pandas as pd
 import torch
-from rlgym_tools.rocket_league.misc.serialize import serialize_game_state, serialize_scoreboard, RF_SCOREBOARD, RF_EPISODE_SECONDS_REMAINING, SB_GAME_TIMER_SECONDS, SB_BLUE_SCORE, SB_ORANGE_SCORE
+from rlgym_tools.rocket_league.misc.serialize import serialize_game_state, serialize_scoreboard, \
+    RF_SCOREBOARD, RF_EPISODE_SECONDS_REMAINING, SB_GAME_TIMER_SECONDS, SB_BLUE_SCORE, SB_ORANGE_SCORE
 from rlgym_tools.rocket_league.replays.convert import replay_to_rlgym
 from rlgym_tools.rocket_league.replays.parsed_replay import ParsedReplay
 from tqdm import trange, tqdm
@@ -247,12 +248,15 @@ def gradio_app():
     with TemporaryDirectory() as temp_dir:
         with gr.Blocks() as demo:
             gr.Markdown("# Next Goal Predictor")
-            gr.Markdown("Upload a replay file to get a plot of the next goal prediction.")
+            gr.Markdown("Upload a replay file to get a plot of the next goal prediction.<br>"
+                        "The model is trained on about 50 000 hours of SSL and RLCS replays in 1s, 2s, and 3s.")
 
             # Use gr.Column to stack components vertically
             with gr.Column():
                 file_input = gr.File(label="Upload Replay File", type="filepath", file_types=[".replay"])
-                checkboxes = gr.CheckboxGroup(label="Options", choices=["Nullify goal difference", "Ignore ties"], type="index")
+                checkboxes = gr.Radio(label="Options", choices=["Nullify goal difference", "Ignore ties"], type="index",
+                                     info="You can choose to make the model think the goal difference is always 0 (so it doesn't prefer one team), "
+                                          "or make it pretend every situation is an overtime (e.g. ties are impossible).")
                 submit_button = gr.Button("Generate Predictions")
                 plot_output = gr.Plot(label="Predictions")
                 download_button = gr.DownloadButton("Download Predictions", visible=False)
@@ -262,8 +266,14 @@ def gradio_app():
                 nullify_goal_difference = 0 in checkbox_options
                 ignore_ties = 1 in checkbox_options
                 print(f"Processing file: {replay_file}")
+                
                 replay_stem = os.path.splitext(os.path.basename(replay_file))[0]
-                preds_file = os.path.join(temp_dir, f"predictions_{replay_stem}_{nullify_goal_difference=}_{ignore_ties=}.csv")
+                postfix = ""
+                if nullify_goal_difference:
+                    postfix += "_nullify_goal_difference"
+                elif ignore_ties:
+                    postfix += "_ignore_ties"
+                preds_file = os.path.join(temp_dir, f"predictions_{replay_stem}_{postfix}.csv")
                 if os.path.exists(preds_file):
                     print(f"Predictions file already exists: {preds_file}")
                     preds = pd.read_csv(preds_file)