Update app.py

app.py

@@ -114,7 +114,7 @@ def verify_team(team_name: str):
     )
 
 
-def answer_prompt(prompt: str):
+def answer_prompt(prompt: str, team_name: str):
     prompt = prompt.strip()
 
     if not prompt:
@@ -127,6 +127,10 @@ def answer_prompt(prompt: str):
 
     reply = generate_response(prompt)
 
+    teams = load_registered_teams()
+    TEAM = teams.get(team_name.strip().casefold())
+    print(TEAM)
+    
     result = supabase.table("submissions").insert({
         "team_id": TEAM["id"],
         "prompt": prompt,
@@ -212,7 +216,7 @@ def main():
     
         submit_button.click(
             answer_prompt,
-            inputs=[prompt_box],
+            inputs=[prompt_box, team_name],
             outputs=[
                 response_box,
                 submit_button,