Update app.py

app.py

@@ -13,13 +13,15 @@ url: str = os.environ.get("SUPABASE_URL")
 key: str = os.environ.get("SUPABASE_KEY")
 supabase: Client = create_client(url, key)
 
+TEAM = None
+
 def load_registered_teams():
     result = supabase.table("teams").select("*").execute()
     
     if not result or not result.data:
         return None
         
-    return {team["name"].casefold(): team["name"] for team in result.data}
+    return {team["name"].casefold(): team for team in result.data}
 
 
 INTRO_DEFAULT = "### The Safety Game · Prompt Submission\n"
@@ -45,6 +47,7 @@ PROMPT_UNLOCKED = (
 TEAM_PANEL_VISIBLE = gr.update()
 TEAM_PANEL_HIDDEN = gr.update(visible=False)
 
+model_name = "openai/gpt-oss-20b"
 
 def generate_response(prompt: str):
     messages = [
@@ -54,7 +57,7 @@ def generate_response(prompt: str):
 
     try:
         client = InferenceClient(
-            model="openai/gpt-oss-20b",
+            model=model_name,
             token=os.getenv("HF_TOKEN"),
         )
         out = client.chat_completion(
@@ -91,8 +94,8 @@ def verify_team(team_name: str):
             TEAM_PANEL_VISIBLE,
             gr.update(value=INTRO_DEFAULT),
         )
-
-    official_name = teams.get(team_name.casefold())
+    TEAM = teams.get(team_name.casefold())
+    official_name = TEAM["name"]
 
     if official_name:
         return (
@@ -122,6 +125,16 @@ def answer_prompt(prompt: str):
         )
 
     reply = generate_response(prompt)
+
+    result = supabase.from("submissions").insert({
+        team_id = TEAM["id"],
+        prompt=prompt,
+        response=response,
+        model=model_name,
+        score=0.67,
+    })
+    print(result)
+    
     return (
         reply,
         gr.update(interactive=False),