add show indexes btn

app.py

@@ -40,10 +40,12 @@ with gr.Blocks(theme=theme, css=css) as demo:
             with gr.Tab(label="Генератор"):
                 
                 defocus_amt = gr.Number(label="Величина дефокуса", value=2, interactive=True)
-                num_zernikes = gr.Number(label="Количество коэффициентов Цернике", value=6, interactive=True)
+                with gr.Row():
+                    num_zernikes = gr.Number(label="Количество коэффициентов Цернике", value=6, interactive=True)
+                    show_zernikes_btn = gr.Button("Показать")
                 sliders_container = gr.Column(elem_classes=["container"])
             
-                @gr.render(inputs=[num_zernikes], triggers=[num_zernikes.change])
+                @gr.render(inputs=[num_zernikes], triggers=[show_zernikes_btn.click])
                 def render_count(count):
                     global zerns
                     zerns = {}

fraunhofer_test.py

@@ -37,7 +37,7 @@ class Fraunhofer:
         zernikes = list(zernikes)
         if len(zernikes) < 5:
             zernikes += [0] * (5-len(zernikes))
-        fig, axs = plt.subplots(1, 3, figsize = (20, 15))
+        
 
  
         wf = self.zernike.generate_zern_screen([1], self.radius)
@@ -50,17 +50,7 @@ class Fraunhofer:
         focal_image_minus = denoise(self.get_focal_point(wf*np.exp(f_minus * 1j)))
 
         return focal_image_plus, focal_image_minus, focal_image_plus-focal_image_minus
-        axs[0].imshow(focal_image_plus, cmap='grey')
-        axs[1].imshow(focal_image_minus, cmap='grey')
-        diff = (focal_image_plus-focal_image_minus)
-        #diff /= np.max(diff)
-    
-        axs[2].imshow(diff - np.min(diff), cmap='grey')
-        axs[0].axis('off')
-        axs[1].axis('off')
-        axs[2].axis('off')
-        fig.show()
-        return fig
+        
 
 
     

zern_generator.py

@@ -38,7 +38,7 @@ class Zernike:
             # "V-PENTAFOIL"
             ]
         
-    def set_image_params(self, span = (-(10 ** -2), 10 ** -2), npix = 2**9):
+    def set_image_params(self, span = (-(10 ** -2), 10 ** -2), npix = 2**8):
         self.span = span
         self.npix = npix
         ddx = np.linspace(*self.span, self.npix)