Update app.py

app.py

@@ -3,7 +3,7 @@ import numpy as np
 import matplotlib.pyplot as plt
 import random
 
-# Classe RubiksCube
+# Classe RubiksCube (mantém o mesmo código)
 class RubiksCube:
     def __init__(self):
         self.cube = np.zeros((6, 3, 3), dtype=int)
@@ -160,6 +160,10 @@ def scramble_cube(n_moves):
     status = "Resolvido!" if cube.is_solved() else "Não resolvido"
     return fig, "\n".join(moves), status
 
+def show_initial_state():
+    fig = create_cube_visualization(cube.cube)
+    return fig, "", "Cubo inicial"
+
 # Criar instância do cubo
 cube = RubiksCube()
 
@@ -188,6 +192,7 @@ with gr.Blocks(title="Cubo Mágico") as demo:
             output_text = gr.Textbox(label="Movimentos realizados", lines=5)
             status_text = gr.Textbox(label="Status")
     
+    # Conectar os botões às funções
     scramble_btn.click(
         fn=scramble_cube,
         inputs=[n_moves],
@@ -200,9 +205,12 @@ with gr.Blocks(title="Cubo Mágico") as demo:
         outputs=[cube_plot, output_text, status_text]
     )
     
-    # Mostrar estado inicial
-    init_fig = create_cube_visualization(cube.cube)
-    cube_plot.update(value=init_fig)
+    # Carregar estado inicial
+    demo.load(
+        fn=show_initial_state,
+        inputs=None,
+        outputs=[cube_plot, output_text, status_text]
+    )
 
 # Iniciar a aplicação
 demo.launch()