Update src/streamlit_app.py

src/streamlit_app.py

@@ -8,6 +8,7 @@ from mne.channels import make_dig_montage
 from matplotlib.animation import FuncAnimation
 import requests
 import streamlit as st
+import tempfile
 
 # Set page config
 st.set_page_config(page_title="Muse EEG Topomap Viewer", layout="centered")
@@ -60,7 +61,7 @@ def plot_eeg_topomap_muse_from_csv(csv_url, save_path_animation=None, save_direc
     if save_path_animation:
         if save_directory:
             os.makedirs(save_directory, exist_ok=True)
-            save_path_animation = os.path.join(save_directory, save_path_animation)
+            save_path_animation = os.path.join(save_directory, os.path.basename(save_path_animation))
 
         fig, ax = plt.subplots()
 
@@ -86,15 +87,16 @@ generate_button = st.button("Generate Topomap Animation")
 
 if generate_button:
     try:
-        gif_path = plot_eeg_topomap_muse_from_csv(
-            csv_url,
-            save_directory = os.path.join(".", "animations")
-            os.makedirs(save_directory, exist_ok=True)
-            save_path_animation = os.path.join(save_directory, "eeg_topomap_animation.gif")
-            show_names=True
-        )
-        if gif_path:
-            st.image(gif_path, caption="EEG Topomap Animation")
+        with tempfile.TemporaryDirectory() as tmpdir:
+            save_path_animation = os.path.join(tmpdir, "eeg_topomap_animation.gif")
+
+            gif_path = plot_eeg_topomap_muse_from_csv(
+                csv_url,
+                save_path_animation=save_path_animation,
+                save_directory=tmpdir,
+                show_names=True
+            )
+            if gif_path:
+                st.image(gif_path, caption="EEG Topomap Animation")
     except Exception as e:
         st.error(f"❌ Error: {str(e)}")
-#update