Update app.py

app.py

@@ -6,12 +6,13 @@ from scipy.fftpack import fft
 import streamlit as st
 import tempfile
 import os
+import subprocess
 
 # Streamlit App
 def main():
-    st.title("MP3 Fourier Transform Visualizer")
+    st.title("MP3 Fourier Transform Visualizer: 科学未来館エディション")
 
-    uploaded_file = st.file_uploader("Upload an MP3 file", type=["mp3"])
+    uploaded_file = st.file_uploader("音声ファイルをアップロード (MP3)", type=["mp3"])
 
     if uploaded_file is not None:
         # Convert MP3 to WAV for easier processing
@@ -45,17 +46,20 @@ def main():
 
         fft_frames = np.array(fft_frames)
 
-        # Create animation
-        fig, ax = plt.subplots()
-        line, = ax.plot(freqs, fft_frames[0])
+        # Create animation with colorful design
+        fig, ax = plt.subplots(facecolor="black")
+        line, = ax.plot(freqs, fft_frames[0], lw=2, color="cyan")
         ax.set_xlim(0, np.max(freqs))
         ax.set_ylim(0, np.max(fft_frames))
-        ax.set_xlabel("Frequency (Hz)")
-        ax.set_ylabel("Amplitude")
-        ax.set_title("Frequency Spectrum Over Time")
+        ax.set_xlabel("Frequency (Hz)", color="white")
+        ax.set_ylabel("Amplitude", color="white")
+        ax.set_title("音の周波数スペクトル", color="white")
+        ax.tick_params(colors="white")
+        fig.patch.set_facecolor("black")
 
         def update(frame):
             line.set_ydata(fft_frames[frame])
+            line.set_color(plt.cm.jet(frame / len(fft_frames)))  # Dynamic color
             return line,
 
         ani = FuncAnimation(fig, update, frames=len(fft_frames), blit=True)
@@ -65,11 +69,23 @@ def main():
             ani.save(temp_video.name, fps=30, extra_args=['-vcodec', 'libx264'])
             video_path = temp_video.name
 
-        st.video(video_path)
+        # Merge audio and video using ffmpeg
+        output_path = tempfile.NamedTemporaryFile(delete=False, suffix="_output.mp4").name
+        audio_path = tempfile.NamedTemporaryFile(delete=False, suffix=".wav").name
+        audio.export(audio_path, format="wav")
+
+        ffmpeg_command = [
+            "ffmpeg", "-y", "-i", video_path, "-i", audio_path, "-c:v", "copy", "-c:a", "aac", output_path
+        ]
+        subprocess.run(ffmpeg_command)
+
+        st.video(output_path)
 
         # Cleanup temporary files
         os.remove(temp_mp3.name)
         os.remove(video_path)
+        os.remove(audio_path)
+        os.remove(output_path)
 
 if __name__ == "__main__":
     main()