Fix figure

app.py

@@ -92,7 +92,7 @@ def scale_vector(vector, new_min, new_max):
 
 
 
-def create_spectrogram_with_att(wav_file, att_contour, att):
+def create_spectrogram_with_att(wav_file, att_contour, att ):
     # Read the WAV file
     sampling_rate, data = wavfile.read(wav_file)
 
@@ -101,19 +101,21 @@ def create_spectrogram_with_att(wav_file, att_contour, att):
     fig, ax = plt.subplots(figsize=(10, 5))
 
     # Plot the spectrogram
-    ax.pcolormesh(t, f, 10 * np.log10(Sxx), shading='gouraud')
+    ax.pcolormesh(t, f, 10 * np.log10(Sxx), shading='gouraud')  # Use grayscale colormap
     ax.set_ylabel('Frequency (Hz)')
     ax.set_xlabel('Time (s)')
     ax.set_title(f'Spectrogram with {att} Contour')
     ax.set_ylim(0, 8000)  # Adjust the frequency range if necessary
 
+    ax_att = ax.twinx()
     # Plot the att contour
-    time_pitch = np.arange(0, len(att_contour) * 0.02, 0.02)  # Assuming pitch_contour is sampled every 20 ms
-    ax.plot(time_pitch, att_contour, color='blue', label=f'{att} Contour')
-    ax.legend()
+    time_att = np.arange(0, len(att_contour) * 0.02, 0.02)  # Assuming pitch_contour is sampled every 20 ms
+    ax_att.plot(time_att, att_contour, color='blue', label=f'{att} Contour')
+    ax_att.legend()
 
     return fig
 
+
 def plot_contour(audio_file, att):
     indx = engine.processor.tokenizer.convert_tokens_to_ids([f'p_{att}'])
     att_contour = engine.logits.squeeze()[:,indx]