Update app.py

app.py

@@ -76,7 +76,36 @@ def submit_form(email, team, affiliation, student, oauth_token: gr.OAuthToken =
 
     return "Registration successful!"
 
+  return "Registration successful!"
 
+# Function to check registration status
+def check_registration(gr.OAuthToken = None):
+    if oauth_token is None or oauth_token.token is None:
+        raise gr.Error("You must be logged in to register!")
+    
+    user_info = api.whoami(token=oauth_token.token)
+    hf_user_id = user_info['name']
+
+    local_dir = snapshot_download(
+        repo_id=DATASET_REPO,
+        repo_type="dataset",
+        use_auth_token=hf_token,
+        local_dir=tempfile.mkdtemp(),
+        local_dir_use_symlinks=False
+    )
+    dataset_path = os.path.join(local_dir, DATASET_FILE)
+
+    if os.path.exists(dataset_path):
+        with open(dataset_path, "r") as f:
+            data = json.load(f)
+    else:
+        data = []
+
+    for entry in data:
+        if entry["hf_id"] == hf_user_id:
+            return f"✅ You are registered in team: **{entry['team']}**"
+
+    return "❌ You are not registered yet."
 
 with gr.Blocks() as demo:
     gr.Markdown("### Registration Form (HuggingFace login required)")
@@ -116,6 +145,11 @@ with gr.Blocks() as demo:
         inputs=[email, team, affiliation, student],
         outputs=output
     )
+    check_btn = gr.Button("Check Registration Status")
+    check_btn.click(
+        fn=check_registration,
+        outputs=output
+    )
 
 
 demo.launch()