Update app.py

app.py

@@ -1267,17 +1267,18 @@ def synthesis_from_spectrum(df):
 
     
 
-def change_textbox(choice):
+def change_textbox(choice, length, height, width, diameter):
+    fig = make_voxels(choice, length, height, width, diameter)
     if choice == "cylinder":
-        return [gradio.Slider.update(visible=True), gradio.Slider.update(visible=False), gradio.Slider.update(visible=True), gradio.Slider.update(visible=False)]
+        return [gradio.Slider.update(visible=True), gradio.Slider.update(visible=False), gradio.Slider.update(visible=True), gradio.Slider.update(visible=False), gradio.Plot.update(values=fig)]
     elif choice == "sphere":
-        return [gradio.Slider.update(visible=False), gradio.Slider.update(visible=False), gradio.Slider.update(visible=True), gradio.Slider.update(visible=False)]
+        return [gradio.Slider.update(visible=False), gradio.Slider.update(visible=False), gradio.Slider.update(visible=True), gradio.Slider.update(visible=False), gradio.Plot.update(values=fig)]
     elif choice == "box":
-        return [gradio.Slider.update(visible=True), gradio.Slider.update(visible=True), gradio.Slider.update(visible=False), gradio.Slider.update(visible=True)]
+        return [gradio.Slider.update(visible=True), gradio.Slider.update(visible=True), gradio.Slider.update(visible=False), gradio.Slider.update(visible=True), gradio.Plot.update(values=fig)]
     elif choice == "wedge":
-        return [gradio.Slider.update(visible=True), gradio.Slider.update(visible=True), gradio.Slider.update(visible=False), gradio.Slider.update(visible=True)]
+        return [gradio.Slider.update(visible=True), gradio.Slider.update(visible=True), gradio.Slider.update(visible=False), gradio.Slider.update(visible=True), gradio.Plot.update(values=fig)]
     elif choice == "cone":
-        return [gradio.Slider.update(visible=True), gradio.Slider.update(visible=False), gradio.Slider.update(visible=True), gradio.Slider.update(visible=False)]
+        return [gradio.Slider.update(visible=True), gradio.Slider.update(visible=False), gradio.Slider.update(visible=True), gradio.Slider.update(visible=False), gradio.Plot.update(values=fig)]
 
    
 import random   
@@ -1313,12 +1314,13 @@ with gradio.Blocks() as demo:
                 diameter = gradio.Slider(label="Diameter", interactive=True, minimum=3.0, maximum=10.0, value=6.5, visible=False)
                 length = gradio.Slider(label="Length", interactive=True, minimum=3.0, maximum=10.0, value=6.5)
             
-                radio.change(fn=change_textbox, inputs=radio, outputs=[height, width, diameter, length])    
+                
                 num = gradio.Number(42, label="Type the index of the shape you would like to use or randomly select it.", visible=False)
                 
                 btn1 = gradio.Button("Randomize")                                   
             with gradio.Column():
                 geo = gradio.Plot(label="Geometry")
+                
     
         with gradio.Row():
             btn2 = gradio.Button("Estimate Spectrum")
@@ -1329,9 +1331,11 @@ with gradio.Blocks() as demo:
     
           with gradio.Column():
               true = gradio.Timeseries(x="Frequency", y=['Surge', 'Heave', 'Pitch'], label="True")
-    
+        
+        radio.change(fn=change_textbox, inputs=[radio, length, height, width, diameter], outputs=[height, width, diameter, length, geo])    
         whence_commeth_geometry.change(fn=geometry_change, inputs=[whence_commeth_geometry], outputs=[radio, height, width, diameter, length, num, true])
         num.change(fn=geometry, inputs=[num], outputs=[geo])
+        
         btn1.click(fn=lambda: random.randint(1, 4999), inputs=[], outputs=[num])
         btn2.click(fn=simple_analysis, inputs=[num], outputs=[pred, true])