Update app.py

app.py

@@ -8,6 +8,28 @@ import tempfile
 import os
 import subprocess
 
+# Define frequency ranges for musical notes (approximation)
+note_frequencies = {
+    "C": 261.63,
+    "D": 293.66,
+    "E": 329.63,
+    "F": 349.23,
+    "G": 392.00,
+    "A": 440.00,
+    "B": 493.88,
+}
+
+# Function to map frequency to a color based on octave and note
+def frequency_to_color(freq):
+    if freq < note_frequencies["C"] / 2 or freq > note_frequencies["B"] * 2:
+        return "gray"  # Out of range
+
+    octave = int(np.log2(freq / note_frequencies["C"]))
+    for note, base_freq in note_frequencies.items():
+        if base_freq * (2 ** octave) <= freq < base_freq * (2 ** (octave + 1)):
+            return plt.cm.hsv((list(note_frequencies.keys()).index(note) + octave * len(note_frequencies)) / 12)
+    return "gray"
+
 # Streamlit App
 def main():
     st.title("MP3 Fourier Transform Visualizer: 科学未来館エディション")
@@ -46,7 +68,7 @@ def main():
 
         fft_frames = np.array(fft_frames)
 
-        # Create animation with colorful design
+        # Create animation with note-based colors
         fig, ax = plt.subplots(facecolor="black")
         line, = ax.plot(freqs, fft_frames[0], lw=2, color="cyan")
         ax.set_xlim(0, np.max(freqs))
@@ -59,7 +81,8 @@ def main():
 
         def update(frame):
             line.set_ydata(fft_frames[frame])
-            line.set_color(plt.cm.jet(frame / len(fft_frames)))  # Dynamic color
+            colors = [frequency_to_color(freq) for freq in freqs]
+            line.set_color(colors[int(frame % len(colors))])  # Dynamic color per frequency range
             return line,
 
         ani = FuncAnimation(fig, update, frames=len(fft_frames), blit=True)