Update app.py

app.py

@@ -29,7 +29,7 @@ def infer(message, history, enable_history=False, max_new_tokens=512):
         )
         model.eval()
 
-    # Add history only if enabled
+    # History handling
     if enable_history and history:
         prev = "\n".join(
             [f"User: {h[0]}\nGamunu: {h[1]}" for h in history if h[1] is not None]
@@ -63,19 +63,66 @@ with gr.Blocks() as demo:
         enable_history = gr.Checkbox(label="Enable chat history", value=False)
         max_new_tokens = gr.Slider(64, 1024, value=512, step=32, label="🔢 Max New Tokens")
 
-    # Use only stable arguments (compatible everywhere)
+    # Example categories
+    gr.Markdown("### 🧩 Try an Example")
+
+    example_categories = {
+        "🧮 Maths & Logic": [
+            "රු. 250 කින් මිලදී ගත් දේ රු. 400 නම් එය ප්‍රතිශතයකින් කීයද වැඩි වී ඇත?",
+            "සාමාන්‍ය ප්‍රවණතාවය (mean) ගණනය කරන්නේ කෙසේද?",
+            "මීටර් 5 දිගක ලී දණ්ඩක් කොටස් 1.25 මීටර් දිග කොටස් 4කට බෙදන්න. කොපමණ දිගක් ඉතිරි වේද?",
+        ],
+        "🌿 Science & Environment": [
+            "පෘථිවියේ වායු මණ්ඩලය ස්තර 5ක් ලෙස වෙන් වේ. ඒවා නම් කරන්න.",
+            "ෆොටෝසින්තසිස් ක්‍රියාවලිය පැහැදිලි කරන්න.",
+            "පරිසර පද්ධතියක් යනු කුමක්ද? උදාහරණයක් දෙන්න.",
+        ],
+        "🎭 Roleplay / Creative": [
+            "ඔබ ගුරුවරයෙකු ලෙස ක්‍රියාකරන්න. ශිෂ්‍යයාට ඉතිහාසය උගන්වන්න.",
+            "ඔබ වෛද්‍යවරයෙකු ලෙස ක්‍රියාකරන්න. රෝගියෙකුට උණක් ඇති විට උපදෙස් දෙන්න.",
+            "ඔබ කවියෙකු ලෙස 'ගඟක්' ගැන කෙටි කවියක් ලියන්න.",
+        ],
+        "🧠 Reasoning / Dependalism": [
+            "මිනිස් සමාජය තනිවම නොව යෙදුණු බව පැහැදිලි කරන්න.",
+            "අපි අන් අය මත රඳා පවතින ආකාරය උදාහරණ සමග පැහැදිලි කරන්න.",
+            "පරිසරය හා ආර්ථිකය අතර සම්බන්ධය විස්තර කරන්න.",
+        ],
+        "🗣 Language / NLP": [
+            "මෙම වාක්‍යය සිංහලයට පරිවර්තනය කරන්න: 'The sun rises in the east.'",
+            "වචන 'ආදරය' සඳහා පරිවර්තන 3ක් දෙන්න.",
+            "වචනයක් එකක් අරගෙන එයට අදාළ විරුද්ධ පදයක් කියන්න.",
+        ],
+    }
+
+    category_dropdown = gr.Dropdown(
+        label="Choose a Category",
+        choices=list(example_categories.keys()),
+        value="🧮 Maths & Logic",
+    )
+
+    example_dropdown = gr.Dropdown(
+        label="Select Example",
+        choices=example_categories["🧮 Maths & Logic"],
+    )
+
+    def update_examples(category):
+        return gr.update(choices=example_categories[category], value=example_categories[category][0])
+
+    category_dropdown.change(update_examples, inputs=category_dropdown, outputs=example_dropdown)
+
     chat = gr.ChatInterface(
         fn=lambda message, history: infer(message, history, enable_history.value, max_new_tokens.value),
         title="Gamunu 4B Instruct Alpha",
-        examples=[
-            ["ෆොටෝසින්තසිස් ක්‍රියාවලිය පැහැදිලි කරන්න."],
-            ["මෙම වාක්‍යය සිංහලයට පරිවර්තනය කරන්න: 'The sun rises in the east.'"],
-            ["රු. 250 කින් මිලදී ගත් දේ රු. 400 නම් එය ප්‍රතිශතයකින් කීයද වැඩි වී ඇත?"],
-            ["ඔබ ගුරුවරයෙකු ලෙස ක්‍රියාකරන්න. ශිෂ්‍යයාට ඉතිහාසය උගන්වන්න."],
-            ["ඔබ කවියෙකු ලෙස 'ගඟක්' ගැන කෙටි කවියක් ලියන්න."],
-        ]
     )
 
+    # Button to load example into chat input
+    def load_example(example):
+        return gr.update(value=example)
+
+    load_btn = gr.Button("📥 Load Example into Chat")
+    chat_textbox = chat.textbox  # the message input in ChatInterface
+    load_btn.click(load_example, inputs=example_dropdown, outputs=chat_textbox)
+
     gr.Markdown("""
 ---
 🪶 **Model:** [`manthilaffs/Gamunu-4B-Instruct-Alpha`](https://huggingface.co/manthilaffs/Gamunu-4B-Instruct-Alpha)