Update app.py

app.py

@@ -37,7 +37,6 @@ def gerar_sequencia_completa(angulo_camera: int, progress=gr.Progress(track_tqdm
     NUM_CICLOS = 10
     clipes_de_video = []
     
-    # Estado inicial da simulação
     estado = {
         "ponto_b": np.array([-40., 20., 10.]),
         "vetor_inercia": np.array([30., 10., -20.])
@@ -46,17 +45,99 @@ def gerar_sequencia_completa(angulo_camera: int, progress=gr.Progress(track_tqdm
     for ciclo_atual in range(NUM_CICLOS):
         progress(ciclo_atual / NUM_CICLOS, desc=f"Processando Ciclo {ciclo_atual + 1}/{NUM_CICLOS}")
         
-        # 1. Atualiza o estado para o novo ciclo
         ponto_a = estado["ponto_b"]
         vetor_inercia_anterior = estado["vetor_inercia"]
         ponto_b = np.random.rand(3) * 120 - 60
 
-        # 2. Calcula a trajetória
         ponto_controle = ponto_a + vetor_inercia_anterior
         pontos_curva = [ponto_a, ponto_controle, ponto_b]
-        x_traj, y_traj, z_traj = bezier_curve_3d(pontos_curva) # 60 vetores
+        x_traj, y_traj, z_traj = bezier_curve_3d(pontos_curva)
         total_frames = len(x_traj)
 
-        # 3. Aprende o eco para o próximo ciclo
-        tamanho_eco = 20 # Eco menor para trajetórias mais curtas
-        pontos_eco = list(zip(x_traj[-tamanho
+        tamanho_eco = 20
+        # --- CORREÇÃO DO SYNTAXERROR AQUI ---
+        pontos_eco = list(zip(x_traj[-tamanho_eco:], y_traj[-tamanho_eco:], z_traj[-tamanho_eco:]))
+        # ------------------------------------
+        vetor_inercia_novo = aprender_com_o_eco_3d(pontos_eco)["velocity_vector"]
+        
+        estado = {"ponto_b": ponto_b, "vetor_inercia": vetor_inercia_novo}
+
+        fig = plt.figure(figsize=(8, 8)); ax = fig.add_subplot(111, projection='3d')
+        
+        cor_fundo = '#0a0a0a'; fig.patch.set_facecolor(cor_fundo); ax.set_facecolor(cor_fundo)
+        ax.xaxis.pane.fill = False; ax.yaxis.pane.fill = False; ax.zaxis.pane.fill = False
+        ax.grid(color='#222222', linestyle='--')
+
+        ax.plot(x_traj, y_traj, z_traj, color='cyan', linewidth=1.5, alpha=0.2) # Trajetória completa mais sutil
+        ax.scatter(*ponto_a, s=150, c='lime', alpha=0.9, edgecolors='w', linewidth=0.5)
+        ax.scatter(*ponto_b, s=150, c='yellow', marker='X', alpha=0.9, edgecolors='w', linewidth=0.5)
+        ax.view_init(elev=30., azim=angulo_camera)
+        ax.set_xlim(-70, 70); ax.set_ylim(-70, 70); ax.set_zlim(-70, 70)
+        ax.set_xticklabels([]); ax.set_yticklabels([]); ax.set_zticklabels([])
+        
+        bola, = ax.plot([], [], [], 'o', color='#ff4500', markersize=8, markeredgecolor='white', linewidth=2)
+        
+        # --- AJUSTE ESTÉTICO: RASTRO DA BOLA ---
+        rastro, = ax.plot([], [], [], '-', color='#ff4500', linewidth=4, alpha=0.6)
+        tamanho_rastro = 25 # O rastro terá 25 vetores de comprimento
+        # ------------------------------------
+
+        def update(frame):
+            bola.set_data_3d([x_traj[frame]], [y_traj[frame]], [z_traj[frame]])
+            
+            # Atualiza o rastro para mostrar os últimos 'tamanho_rastro' pontos
+            start_index = max(0, frame - tamanho_rastro)
+            rastro.set_data_3d(x_traj[start_index:frame+1], y_traj[start_index:frame+1], z_traj[start_index:frame+1])
+            
+            return bola, rastro,
+
+        ani = FuncAnimation(fig, update, frames=total_frames, interval=33, blit=True)
+        clip_filename = f"clip_{ciclo_atual}.mp4"
+        ani.save(clip_filename, writer='ffmpeg', fps=30, dpi=120)
+        plt.close(fig)
+        clipes_de_video.append(clip_filename)
+
+    progress(0.95, desc="Concatenando 10 clipes em um vídeo final...")
+    
+    with open("lista_de_clipes.txt", "w") as f:
+        for clip in clipes_de_video:
+            f.write(f"file '{os.path.abspath(clip)}'\n")
+            
+    video_final = "sequencia_convergencia_final.mp4"
+    comando = f"ffmpeg -y -f concat -safe 0 -i lista_de_clipes.txt -c copy {video_final}"
+    subprocess.run(comando, shell=True, check=True, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
+
+    for clip in clipes_de_video: os.remove(clip)
+    os.remove("lista_de_clipes.txt")
+    
+    return video_final
+
+# --- Interface Gradio (Inalterada) ---
+
+with gr.Blocks(theme=gr.themes.Base(primary_hue="blue")) as demo:
+    gr.Markdown(
+        """
+        # 🎬 Estúdio de Animação de Convergência 3D
+        Clique no botão para gerar uma sequência contínua de **10 ciclos**. Cada ciclo tem **60 vetores**.
+        O sistema irá renderizar cada ciclo e concatenar todos em um único vídeo de demonstração.
+        """
+    )
+    
+    with gr.Row():
+        with gr.Column(scale=1):
+            gr.Markdown("### 🎮 Controles")
+            angulo_camera_slider = gr.Slider(-180, 180, value=45, label="Ângulo da Câmera (Fixo)")
+            generate_btn = gr.Button("🚀 Gerar Sequência Completa (10 Ciclos)", variant="primary")
+            
+        with gr.Column(scale=2):
+            gr.Markdown("### 📽️ Vídeo Final")
+            video_output = gr.Video(label="Animação da Sequência Completa", autoplay=True)
+
+    generate_btn.click(
+        fn=gerar_sequencia_completa,
+        inputs=[angulo_camera_slider],
+        outputs=[video_output]
+    )
+
+if __name__ == "__main__":
+    demo.launch()