Update app.py

app.py

@@ -140,26 +140,28 @@ def solve_math_problem(question, additional_info, max_tokens):
     final_answer = parse_final_answer(result)
     return full_answer, final_answer
 
+
 def main():
     with gr.Blocks() as demo:
         gr.Markdown(title)
         question_field = gr.Code(language='python', label="🤔Enter your math problem")
         additional_info_field = gr.Text(value=additional_info_prompt, label="🪜Optional train-of-thought")
-        max_tokens = gr.Slider(minimum=150, maximum=1200, value=250, label="Max Tokens")
+        max_tokens = gr.Slider(minimum=150, maximum=1200, value=250, label="🪙Max Tokens")
         submit_button = gr.Button("📉Solve")
         full_answer_output = gr.Code(label="🔮TonicsMathAssistant📉", interactive=False)
         final_answer_output = gr.Textbox(label="Final Answer")
 
-        # Define a function to update the question and additional info fields
-        def set_example(example_index):
+        example_index = gr.Number(label="Example Index", visible=False)
+
+        def set_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
+            question = examples[int(index)]
+            additional_info = additional_info_prompt
+            return question, additional_info
 
         with gr.Accordion("🤖 UniMath Examples", open=False):
-            gr.Markdown("Click an example to load it into the solver.")
-            for i in range(len([unimath1, unimath2, unimath3, unimath4])):
+            gr.Markdown("Click on a 🤖 UniMath Example to load it into the solver.")
+            for i, _ in enumerate([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])