feat: centralize example inputs

app.py

@@ -15,6 +15,8 @@ import psutil
 import torch
 from transformers import AutoModel
 
+from example_inputs import EXAMPLE_INPUTS
+
 
 def _ensure_nltk_punkt_resources() -> None:
     """Download punkt resources on first run so inference never fails."""
@@ -29,13 +31,6 @@ def _ensure_nltk_punkt_resources() -> None:
 DEFAULT_MODEL = "hotchpotch/open-provence-reranker-japanese-xsmall-v1"
 DEFAULT_THRESHOLD = 0.1
 DEFAULT_BATCH_SIZE = 8
-DEFAULT_EXAMPLE_QUESTION = "日本の首都はどこですか？"
-DEFAULT_EXAMPLE_CONTEXT = """
-今日は学校に行き、さまざまなことを学んだり、友達と学食でたらふく食べた。
-日本の首都は東京で、東京は日本の政治、経済、文化の中心地らしい。約1,400万人の人口を抱える世界有数の大都市とのことだ。
-夜は飲み会に誘われたが、参加せずに帰宅した、今月そんなにお金が残ってないからなぁ、残念だ。
-""".strip()
-
 MODEL_SCRIPT_ENV = "OPEN_PROVENCE_MODELING_SCRIPT_PATH"
 
 
@@ -309,30 +304,10 @@ def build_interface() -> gr.Blocks:
             )
 
         gr.Examples(
-            examples=[
-                [
-                    DEFAULT_EXAMPLE_QUESTION,
-                    "",
-                    DEFAULT_EXAMPLE_CONTEXT,
-                ],
-                [
-                    "What's your favorite Japanese food?",
-                    "",
-                    (
-                        "Work deadlines piled up today, and I kept rambling about budget spreadsheets to my roommate.\n"
-                        "Next spring I'm planning a trip to Japan so I can wander Kyoto's markets and taste every regional dish I find.\n"
-                        "Sushi is honestly my favourite—I want to grab a counter seat and let the chef serve endless nigiri until I'm smiling through soy sauce.\n"
-                        "Later I remembered to water the plants and pay the electricity bill before finally getting some sleep."
-                    ),
-                ],
-            ],
-            inputs=[
-                question_input,
-                title_input,
-                text_input,
-            ],
+            examples=[[item.question, item.title, item.text] for item in EXAMPLE_INPUTS],
+            inputs=[question_input, title_input, text_input],
             label="Sample input",
-            examples_per_page=2,
+            examples_per_page=len(EXAMPLE_INPUTS),
         )
 
         run_button = gr.Button("Run inference", variant="primary")

example_inputs.py

@@ -0,0 +1,29 @@
+from __future__ import annotations
+
+from dataclasses import dataclass
+
+
+@dataclass(frozen=True)
+class ExampleInput:
+    question: str
+    title: str
+    text: str
+
+
+EXAMPLE_INPUTS: tuple[ExampleInput, ...] = (
+    ExampleInput(
+        question="日本の首都はどこですか？",
+        title="",
+        text="""今日は学校に行き、さまざまなことを学んだり、友達と学食でたらふく食べた。\n日本の首都は東京で、東京は日本の政治、経済、文化の中心地らしい。約1,400万人の人口を抱える世界有数の大都市とのことだ。\n夜は飲み会に誘われたが、参加せずに帰宅した、今月そんなにお金が残ってないからなぁ、残念だ。""",
+    ),
+    ExampleInput(
+        question="What's your favorite Japanese food?",
+        title="",
+        text="""Work deadlines piled up today, and I kept rambling about budget spreadsheets to my roommate.\nNext spring I'm planning a trip to Japan so I can wander Kyoto's markets and taste every regional dish I find.\nSushi is honestly my favourite—I want to grab a counter seat and let the chef serve endless nigiri until I'm smiling through soy sauce.\nLater I remembered to water the plants and pay the electricity bill before finally getting some sleep.""",
+    ),
+    ExampleInput(
+        question="この文章の要旨を一言で教えてください。",
+        title="",
+        text="""新しい研究によると、都市部の公共交通の改善と緑地の拡充は、住民の幸福度を大幅に押し上げる可能性があるという。\n東京や大阪などの大都市では、近年自転車専用レーンや歩行者天国の整備が進み、\n交通事故の発生件数が減少しているだけでなく、観光産業や地域コミュニティの活性化にも寄与している。""",
+    ),
+)