Update image.py

image.py

@@ -372,7 +372,7 @@ def image_tab():
 
         preview_image_btn.click(
             fn=preview_image_equation,
-            inputs=[extracted_eq_state, real_only],
+            inputs=[extracted_eq_state, real_image_checkbox],
             outputs=[image_equation_display, confirm_image_btn, edit_image_btn, 
                      image_steps_md, image_plot_output]
         )
@@ -388,8 +388,8 @@ def image_tab():
 
         confirm_image_btn.click(
             fn=confirm_image_solution,
-            inputs=[extracted_eq_state, real_only],
-            outputs=[image_steps_md, image_plot_output, image_solution_txt]  # ✅ includes solution
+            inputs=[extracted_eq_state, real_image_checkbox],
+            outputs=[image_steps_md, image_plot_output, image_solution_txt]
         )
 
         def enable_manual_edit(eq_data):