Update src/streamlit_app.py

src/streamlit_app.py

@@ -266,6 +266,39 @@ if st.button("Show Band Power Chart (TP9)"):
 
     except Exception as e:
         st.error(f"❌ Error: {str(e)}")
+# Button to show EEG animation
+if st.button("Show Animated EEG Trace (TP9)"):
+    try:
+        # Setup figure and axes
+        fig, ax = plt.subplots()
+        line, = ax.plot([], [], lw=2)
+        ax.set_xlim(0, 5)
+        ax.set_ylim(np.min(tp10_filtered_mne), np.max(tp10_filtered_mne))
+        ax.set_title("Animated EEG Trace - TP9")
+
+        # Init function
+        def init():
+            line.set_data([], [])
+            return line,
+
+        # Animation function
+        def animate(i):
+            start = i * int(sampling_rate * 0.1)
+            end = start + int(sampling_rate * 5)
+            x = np.arange(start, end) / sampling_rate
+            y = tp10_filtered_mne[start:end]
+            line.set_data(x, y)
+            return line,
+
+        # Create animation
+        ani = FuncAnimation(fig, animate, init_func=init, frames=100, interval=100, blit=True)
+
+        # Save to temporary file as GIF
+        with tempfile.NamedTemporaryFile(delete=False, suffix='.gif') as tmpfile:
+            ani.save(tmpfile.name, writer='pillow', fps=10)
+            st.image(tmpfile.name, caption="Animated EEG Trace", use_column_width=True)
+
+        plt.close(fig)  # Close figure to avoid duplicate rendering
 
-
-
+    except Exception as e:
+        st.error(f"❌ Error: {str(e)}")