go back to image

app.py

@@ -47,14 +47,15 @@ def plot_stft(audio_file):
     fig.update_xaxes(visible=True, showgrid=False)
     fig.update_yaxes(visible=True, showgrid=False)
 
-    # Convert the figure to an HTML string
-    html_code = pio.to_html(fig, full_html=False, config={'displaylogo': False, 'modeBarButtonsToRemove': ['toImage', 'zoomIn', 'zoomOut', 'resetScale']})
+    # Save the figure as an image
+    image_path = 'stft_plot.png'
+    fig.write_image(image_path)
 
-    return html_code
+    return image_path
 
 # Gradio interface
-demo = gr.Interface(fn=plot_stft, 
-                    inputs=gr.Audio(type="filepath"), 
-                    outputs="html")
+demo = gr.Interface(fn=plot_stft,
+                    inputs=gr.Audio(type="filepath"),
+                    outputs="image")
 
 demo.launch()

requirements.txt

@@ -3,4 +3,6 @@ numpy
 plotly
 scipy
 librosa
+kaleido
+