Update app.py

app.py

@@ -69,9 +69,11 @@ def generate_response(user_message, max_tokens, temperature, top_k, top_p, repet
 
     yield new_history, new_history
 
+# Fixed: Match the keys with your button labels
 example_messages = {
-    "Verilog example": "Design a 4-bit adder.",
+    "Math reasoning": "Calculate the sum of the first 10 prime numbers and explain your reasoning step by step.",
     "Logic puzzle": "Four people (Alex, Blake, Casey, and Dana) each have a different favorite color (red, blue, green, yellow) and a different favorite fruit (apple, banana, cherry, date). Given the following clues: 1) The person who likes red doesn't like dates. 2) Alex likes yellow. 3) The person who likes blue likes cherries. 4) Blake doesn't like apples or bananas. 5) Casey doesn't like yellow or green. Who likes what color and what fruit?",
+    "Verilog example": "Design a 4-bit adder circuit in Verilog with proper test benches."
 }
 
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
@@ -140,6 +142,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             with gr.Row():
                 example1_button = gr.Button("Math reasoning")
                 example2_button = gr.Button("Logic puzzle")
+                example3_button = gr.Button("Verilog example")
 
     submit_button.click(
         fn=generate_response,
@@ -157,6 +160,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         outputs=[chatbot, history_state]
     )
 
+    # Fixed: Now these will work without KeyError
     example1_button.click(
         fn=lambda: gr.update(value=example_messages["Math reasoning"]),
         inputs=None,
@@ -167,5 +171,10 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         inputs=None,
         outputs=user_input
     )
+    example3_button.click(
+        fn=lambda: gr.update(value=example_messages["Verilog example"]),
+        inputs=None,
+        outputs=user_input
+    )
 
 demo.launch(ssr_mode=False, share=True)