Update main.py

main.py

@@ -664,37 +664,42 @@ with demo.route("Object Localization", "/object-localization") as obj_loc:
             )
 
 # ── VQA PAGE ──────────────────────────────────────────────────────────
-with demo.route("Visual Question Answering", "/vqa"):
+# with demo.route("Visual Question Answering", "/vqa"):
     # gr.Markdown("# 🤔 Visual Question Answering")
     # gr.Markdown("### 🔒 Coming Soon")
     # gr.Markdown(
     #     "This challenge is currently under development. "
     # )
 
-    # ── VQA PAGE ──────────────────────────────────────────────────────────
+# ── VQA PAGE ──────────────────────────────────────────────────────────
 with demo.route("Visual Question Answering", "/vqa") as vqa_page:
+
     with gr.Row():
         with gr.Column(scale=5):
             gr.Markdown("# 🤔 Visual Question Answering Challenge")
-            gr.Markdown("Answer open-ended questions about images taken by blind users.")
+            gr.Markdown(
+                "Answer open-ended questions about images taken by blind users. "
+                "Models are evaluated on answer accuracy and relevance."
+            )
         with gr.Column(scale=1, min_width=160):
             gr.LoginButton(size="lg")
-            vqa_greeting = gr.Markdown("👋 Log in to view or submit.")
+            vqa_greeting = gr.Markdown("👋 Log in to submit.")
 
     gr.Markdown("---")
 
     with gr.Tabs():
-        # Task 2: Leaderboard Tab
+
+        # 1. Leaderboard Tab
         with gr.TabItem("🏆 Leaderboard"):
             gr.Markdown("### VQA Challenge Rankings")
-            lb_msg = gr.Markdown("")
+            gr.Markdown("Ranked by **Accuracy** (descending).")
             
-            # This ensures an empty DF is shown if there are no submissions
+            vqa_lb_msg = gr.Markdown("")
             vqa_lb_table = gr.Dataframe(interactive=False, wrap=True)
             refresh_vqa_lb_btn = gr.Button("🔄 Refresh Leaderboard", variant="secondary", size="sm")
 
             def refresh_vqa_leaderboard(profile: gr.OAuthProfile | None):
-                # Define structure for the empty state
+                # Task 2: Default empty DF structure
                 cols = ["Rank", "Team", "Model", "Accuracy", "Scored At"]
                 empty_df = pd.DataFrame(columns=cols)
                 
@@ -702,33 +707,101 @@ with demo.route("Visual Question Answering", "/vqa") as vqa_page:
                     df = _load_leaderboard_df()
                     # Filter for VQA challenge specifically
                     if not df.empty and "challenge_type" in df.columns:
-                        df = df[df["challenge_type"].str.lower() == "vqa"]
+                        df = df[df["challenge_type"].str.lower() == "visual question answering"]
                 except Exception as e:
-                    return empty_df, f"❌ Error: {e}", get_user_greeting(profile)
+                    return empty_df, f"❌ Could not load leaderboard: {e}", get_user_greeting(profile)
 
                 if df.empty:
-                    # Task 2: Return empty DF if no submissions
                     return empty_df, "ℹ️ No VQA submissions yet. Be the first!", get_user_greeting(profile)
 
-                # Process the data if it exists
                 df_display = df.copy()
                 df_display.insert(0, "Rank", range(1, len(df_display) + 1))
-                # ... (rest of your processing logic)
+                
+                if "timestamp" in df_display.columns:
+                    df_display["Scored At"] = pd.to_datetime(
+                        df_display["timestamp"], unit="s", errors="coerce"
+                    ).dt.strftime("%d %b %Y, %I:%M %p")
+                
+                # Filter to VQA specific metrics
+                display_cols = ["Rank", "team", "model", "accuracy", "Scored At"]
+                df_display = df_display[[c for c in display_cols if c in df_display.columns]]
+                df_display.rename(columns={"team": "Team", "model": "Model", "accuracy": "Accuracy"}, inplace=True)
+                
                 return df_display, "", get_user_greeting(profile)
 
-            # --- CRITICAL: These must be indented INSIDE 'with demo.route' ---
-            refresh_vqa_lb_btn.click(
-                refresh_vqa_leaderboard, 
-                outputs=[vqa_lb_table, lb_msg, vqa_greeting]
+            refresh_vqa_lb_btn.click(refresh_vqa_leaderboard, outputs=[vqa_lb_table, vqa_lb_msg, vqa_greeting])
+            vqa_page.load(refresh_vqa_leaderboard, outputs=[vqa_lb_table, vqa_lb_msg, vqa_greeting])
+
+        # 2. Submit Tab
+        with gr.TabItem("🚀 Submit Predictions"):
+            with gr.Row():
+                vqa_submit_greeting = gr.Markdown("👋 Log in with HuggingFace to submit.")
+                vqa_cap_info = gr.Markdown("")
+            gr.Markdown("---")
+            with gr.Row():
+                with gr.Column(scale=3):
+                    gr.Markdown("#### Upload VQA Results")
+                    vqa_file_input = gr.File(label="Choose a JSON file", file_types=[".json"])
+                with gr.Column(scale=2):
+                    gr.Markdown("#### Submission Info")
+                    vqa_team_input = gr.Textbox(label="Team / Display Name")
+                    vqa_model_input = gr.Textbox(label="Model Name")
+                    vqa_phase_input = gr.Dropdown(
+                        label="Phase",
+                        choices=[p["label"] for p in PHASES],
+                        value=PHASES[0]["label"],
+                    )
+            vqa_submit_btn = gr.Button("Submit (Queue for Evaluation)", variant="primary", size="lg")
+            vqa_submit_status = gr.Markdown("")
+            vqa_sid_box = gr.Code(label="Submission ID", visible=False)
+
+            def do_vqa_submit(file, team, model, phase, profile: gr.OAuthProfile | None):
+                # We pass "Visual Question Answering" as the challenge type
+                msg, sid = handle_submit(file, team, model, phase, "Visual Question Answering", profile)
+                return msg, gr.update(value=sid, visible=bool(sid))
+
+            vqa_submit_btn.click(
+                do_vqa_submit,
+                inputs=[vqa_file_input, vqa_team_input, vqa_model_input, vqa_phase_input],
+                outputs=[vqa_submit_status, vqa_sid_box],
             )
-            
-            # This was the line causing your error! 
-            # It's now properly indented under 'vqa_page'
-            vqa_page.load(
-                refresh_vqa_leaderboard, 
-                outputs=[vqa_lb_table, lb_msg, vqa_greeting]
+
+            def update_vqa_submit_ui(profile: gr.OAuthProfile | None):
+                return get_user_greeting(profile), get_daily_cap_info(profile)
+
+            vqa_page.load(update_vqa_submit_ui, outputs=[vqa_submit_greeting, vqa_cap_info])
+
+        # 3. My Submissions Tab
+        with gr.TabItem("📋 My Submissions"):
+            vqa_my_sub_greeting = gr.Markdown("👋 Log in to view your submissions.")
+            vqa_my_sub_stats = gr.Markdown("")
+            with gr.Row():
+                vqa_phase_filter = gr.Dropdown(
+                    label="Filter by Phase",
+                    choices=["All"] + [p["codename"] for p in PHASES],
+                    value="All",
+                    scale=2
+                )
+                vqa_refresh_my_btn = gr.Button("🔄 Refresh", variant="secondary", scale=1)
+            vqa_my_sub_table = gr.Dataframe(interactive=False, wrap=True)
+
+            def refresh_vqa_my_subs(phase_filter, profile: gr.OAuthProfile | None):
+                df, msg, stats = load_my_submissions(phase_filter, profile)
+                # Further filter results to only show VQA
+                if not df.empty and "Challenge Type" in df.columns:
+                    df = df[df["Challenge Type"] == "Visual Question Answering"]
+                return df, msg, stats, get_user_greeting(profile)
+
+            vqa_refresh_my_btn.click(
+                refresh_vqa_my_subs,
+                inputs=[vqa_phase_filter],
+                outputs=[vqa_my_sub_table, gr.Markdown(), vqa_my_sub_stats, vqa_my_sub_greeting],
             )
-            
+            vqa_page.load(
+                refresh_vqa_my_subs,
+                inputs=[vqa_phase_filter],
+                outputs=[vqa_my_sub_table, gr.Markdown(), vqa_my_sub_stats, vqa_my_sub_greeting],
+            )            
 # ── ANSWER GROUNDING PAGE ─────────────────────────────────────────────
 with demo.route("Answer Grounding", "/answer-grounding"):
     gr.Markdown("# 📍 Answer Grounding")