Update app.py

app.py

@@ -29,7 +29,7 @@ def infer(message, history, enable_history=False, max_new_tokens=512):
         )
         model.eval()
 
-    # History handling
+    # Add history only if enabled
     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,66 +63,19 @@ 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")
 
-    # 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)
-
+    # Use only stable arguments (compatible everywhere)
     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)