Update app.py

app.py

@@ -65,6 +65,16 @@ def visualize_memory_grid(memory_grid):
     ax.set_ylabel("Memory Cells (Rows)")
     st.pyplot(fig)  # Render plot in Streamlit using st.pyplot
 
+# Visualize the Spike Potential Output
+def visualize_spike_output(output):
+    # Plot the spike potential (filtered output)
+    fig, ax = plt.subplots(figsize=(6, 4))
+    ax.plot(output.detach().numpy(), marker='o', linestyle='-', color='b')
+    ax.set_title("Spike Potential Output")
+    ax.set_xlabel("Spike Index")
+    ax.set_ylabel("Spike Potential")
+    st.pyplot(fig)
+
 # Generate spikes for the given pressure value
 def generate_spikes_for_pressure(pressure):
     # Simulate the pressure as the input (scale to appropriate format for LSTM)
@@ -84,6 +94,9 @@ def generate_spikes_for_pressure(pressure):
         # Visualize the memory grid (4x4 grid)
         visualize_memory_grid(memory_grid)
 
+        # Visualize the spike potential output
+        visualize_spike_output(output)
+
         # Optionally, print the output
         print(f"Filtered Output: {output}")
         print(f"Memory Grid after update:\n{memory_grid}")