Adapt to Gradio 6.x API (remove show_download_button, move theme to launch)

app.py

@@ -12,20 +12,16 @@ from pathlib import Path
 import gradio as gr
 
 ROOT = Path(__file__).parent
-REAL_DIR = ROOT / "data" / "real"
-AI_DIR = ROOT / "data" / "ai"
-REAL_IMGS = sorted(REAL_DIR.glob("*"))
-AI_IMGS = sorted(AI_DIR.glob("*"))
+REAL_IMGS = sorted((ROOT / "data" / "real").glob("*"))
+AI_IMGS = sorted((ROOT / "data" / "ai").glob("*"))
 
 # Baselines from the GPT4o-Receipt paper (Zhang*, Ren*, et al., 2026)
 HUMAN_F1 = 0.852
-HUMAN_VISUAL_DELTA = 1.87  # Largest visual discrimination of any evaluator
 CLAUDE_F1 = 0.975
 GEMINI_F1 = 0.890
 
 
 def new_round(state):
-    """Pick one real + one AI, randomize left/right."""
     real = random.choice(REAL_IMGS)
     ai = random.choice(AI_IMGS)
     ai_on_left = random.random() < 0.5
@@ -38,9 +34,9 @@ def new_round(state):
         str(right),
         state,
         gr.update(value="### Which receipt is AI-generated?", visible=True),
-        gr.update(visible=True),  # left button
-        gr.update(visible=True),  # right button
-        gr.update(visible=False), # next button
+        gr.update(visible=True),
+        gr.update(visible=True),
+        gr.update(visible=False),
     )
 
 
@@ -85,33 +81,21 @@ def guess(side, state):
     return (
         state,
         gr.update(value=msg, visible=True),
-        gr.update(visible=False),  # hide L button
-        gr.update(visible=False),  # hide R button
-        gr.update(visible=True),   # show Next button
+        gr.update(visible=False),
+        gr.update(visible=False),
+        gr.update(visible=True),
     )
 
 
-def reset(state):
-    state = {"round": 0, "correct": 0, "total": 0}
-    return new_round(state)
-
-
-with gr.Blocks(
-    title="Spot the AI Receipt — Scam.AI",
-    theme=gr.themes.Soft(primary_hue="blue"),
-    css="""
-    #title { text-align: center; }
-    .receipt-img img { object-fit: contain !important; max-height: 500px; }
-    """,
-) as demo:
+# Gradio 6.x: theme/css moved to launch(); no show_download_button kwarg
+with gr.Blocks(title="Spot the AI Receipt — Scam.AI") as demo:
     gr.Markdown(
         "# 🧾 Spot the AI Receipt\n"
         "*One of these receipts is authentic. The other was fully synthesized by "
         "GPT-4o + GPT-Image-1. Can you tell the difference?*\n\n"
         "*Built by [Scam.AI](https://www.scam.ai) · Data: "
         "[gpt4o-receipt](https://huggingface.co/datasets/Scam-AI/gpt4o-receipt) "
-        "+ [CORD-v2](https://huggingface.co/datasets/naver-clova-ix/cord-v2)*",
-        elem_id="title",
+        "+ [CORD-v2](https://huggingface.co/datasets/naver-clova-ix/cord-v2)*"
     )
 
     state = gr.State({"round": 0, "correct": 0, "total": 0})
@@ -120,18 +104,15 @@ with gr.Blocks(
 
     with gr.Row():
         with gr.Column():
-            img_l = gr.Image(label="Receipt A", elem_classes="receipt-img",
-                             interactive=False, show_download_button=False)
+            img_l = gr.Image(label="Receipt A", interactive=False)
             btn_l = gr.Button("👉 A is the AI", variant="primary", size="lg")
         with gr.Column():
-            img_r = gr.Image(label="Receipt B", elem_classes="receipt-img",
-                             interactive=False, show_download_button=False)
+            img_r = gr.Image(label="Receipt B", interactive=False)
             btn_r = gr.Button("👉 B is the AI", variant="primary", size="lg")
 
     result = gr.Markdown(visible=False)
     btn_next = gr.Button("→ Next round", variant="secondary", visible=False)
 
-    # Wiring
     btn_l.click(
         lambda s: guess("left", s),
         inputs=state,
@@ -147,10 +128,9 @@ with gr.Blocks(
         outputs=[img_l, img_r, state, prompt, btn_l, btn_r, btn_next],
     )
 
-    # Auto-start on load
     demo.load(new_round, inputs=state,
               outputs=[img_l, img_r, state, prompt, btn_l, btn_r, btn_next])
 
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(theme=gr.themes.Soft(primary_hue="blue"))