Update app.py

app.py

@@ -1,26 +1,34 @@
 import gradio as gr
 from pydub import AudioSegment
-import numpy as np
+import base64
 import tempfile
-import soundfile as sf
+import numpy as np
+import io
 
-def normalize_audio(file):
-    if file is None:
+def process_audio_gr(audio):
+    if audio is None:
         return None
-    audio = AudioSegment.from_file(file)
-    normalized_audio = audio.normalize()
-    temp_path = tempfile.NamedTemporaryFile(suffix=".wav", delete=False).name
-    normalized_audio.export(temp_path, format="wav")
-    samples, sample_rate = sf.read(temp_path)
-    return samples, sample_rate
+    # Load audio with pydub
+    audio_segment = AudioSegment.from_file(io.BytesIO(audio))
+    # Simple effect: normalize
+    normalized = audio_segment.normalize()
+    # Export to bytes
+    buf = io.BytesIO()
+    normalized.export(buf, format="wav")
+    buf.seek(0)
+    audio_bytes = buf.read()
+    # Convert to base64
+    audio_b64 = base64.b64encode(audio_bytes).decode()
+    data_uri = f"data:audio/wav;base64,{audio_b64}"
+    return data_uri
 
-app = gr.Interface(
-    fn=normalize_audio,
-    inputs=gr.Audio(type="filepath"),
-    outputs=gr.Audio(type="numpy"),
-    title="Normalize Audio Test",
-    description="Upload an audio file to normalize it and get the processed audio back."
+iface = gr.Interface(
+    fn=process_audio_gr,
+    inputs=gr.Audio(source="upload", type="file"),
+    outputs=gr.Audio(type="auto"),
+    title="Audio Normalize Demo",
+    description="Upload audio, get normalized audio back."
 )
 
 if __name__ == "__main__":
-    app.launch()
+    iface.launch()