Update app.py

app.py

@@ -149,16 +149,20 @@ def main():
         submit_button = gr.Button("📉Solve")
         full_answer_output = gr.Code(label="🔮TonicsMathAssistant📉", interactive=False)
         final_answer_output = gr.Textbox(label="Final Answer")
-        
-        def set_example(example):
-            question_field.update(example)
+
+        # Define a function to update the question and additional info fields
+        def set_example(example_index):
+            examples = [unimath1, unimath2, unimath3, unimath4]
+            question_field.update(examples[example_index])
             additional_info_field.update(additional_info_prompt)
-        
+            return "", ""  # Return empty strings as placeholders
+
         with gr.Accordion("🤖 UniMath Examples", open=False):
             gr.Markdown("Click an example to load it into the solver.")
-            for i, example in enumerate([unimath1, unimath2, unimath3, unimath4], start=1):
-                gr.Button(f"Example {i}", elem_id=f"example{i}").click(fn=set_example, inputs=example, outputs=[question_field, additional_info_field])
-        
+            for i in range(len([unimath1, unimath2, unimath3, unimath4])):
+                example_button = gr.Button(f"Example {i+1}")
+                example_button.click(fn=set_example, inputs=[i], outputs=[question_field, additional_info_field])
+
         submit_button.click(fn=solve_math_problem, inputs=[question_field, additional_info_field, max_tokens], outputs=[full_answer_output, final_answer_output])
 
     demo.launch()