Update app.py

app.py

@@ -19,12 +19,13 @@ def reset_tracker():
     exercise_log_str = "\n".join(f"{name}: {count}" for name, count in exercise_log.items())
     return exercise_log_str, total_exercises_completed, days_completed
 
-def update_exercise(button_id):
+def update_exercise(exercise_data):
     global exercise_log, total_exercises_completed, last_reset
     if date.today() > last_reset:
         reset_tracker()
-    if exercise_log[button_id] < 6:
-        exercise_log[button_id] += 1
+    exercise_name = exercise_data['data']
+    if exercise_log[exercise_name] < 6:
+        exercise_log[exercise_name] += 1
         total_exercises_completed += 1  # Increment on each exercise completion.
     exercise_log_str = "\n".join(f"{name}: {count}" for name, count in exercise_log.items())
     return exercise_log_str, total_exercises_completed, days_completed
@@ -32,23 +33,20 @@ def update_exercise(button_id):
 with gr.Blocks() as app:
     gr.Markdown("## Physical Therapy Exercise Tracker")
     with gr.Row():
-        ex_buttons = [gr.Button(f"Exercise {i+1}") for i in range(8)]
+        for i in range(8):
+            gr.Button(f"Exercise {i+1}", data=f"Exercise {i+1}").click(
+                update_exercise, 
+                inputs=[], 
+                outputs=[exercise_log_text, total_exercises_completed_num, days_completed_num]
+            )
     reset_button = gr.Button("Reset Daily Tracker")
     exercise_log_text = gr.Textbox(label="Exercise Log", lines=10, value="Please press an exercise button to start tracking")
     total_exercises_completed_num = gr.Number(label="Total Exercises Completed", value=total_exercises_completed)
     days_completed_num = gr.Number(label="Days Completed", value=days_completed)
 
-    for i, button in enumerate(ex_buttons, start=1):
-        button_id = f"Exercise {i}"
-        button.click(
-            fn=update_exercise,
-            inputs=button_id,  # Pass the button_id directly to the function
-            outputs=[exercise_log_text, total_exercises_completed_num, days_completed_num]
-        )
-
     reset_button.click(
         fn=reset_tracker,
-        inputs=[],  # No input is needed for reset
+        inputs=[], 
         outputs=[exercise_log_text, total_exercises_completed_num, days_completed_num]
     )