Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 import tensorflow as tf
 import librosa
+import librosa.display
 import numpy as np
 import matplotlib.pyplot as plt
 import sounddevice as sd
@@ -54,14 +55,15 @@ def process_audio(audio_file, breath_in_time, breath_out_time):
         plt.title("Audio Waveform with Inhale/Exhale Segments")
         plt.xlabel("Time (s)")
         plt.ylabel("Amplitude")
-        plt.savefig("waveform_highlighted.png")
+        waveform_file = "waveform_highlighted.png"
+        plt.savefig(waveform_file)
         plt.close()
 
         result_table = "Segment\tType\tDuration (s)\tAmplitude\n" + "\n".join(
             f"{row['Segment']}\t{row['Type']}\t{row['Duration (s)']}\t{row['Amplitude']}" for row in results
         )
 
-        return result_table, "waveform_highlighted.png"
+        return result_table, waveform_file
 
     except Exception as e:
         return f"Error: {str(e)}", None
@@ -125,8 +127,7 @@ with gr.Blocks() as demo:
         }
     </script>
 """):
-    pass  # Ensure this is indented properly
-
+        pass  # Ensure this is indented properly
 
     def handle_record_and_visualize(breath_in, breath_out):
         total_duration = breath_in + breath_out