Update app.py

app.py

@@ -18,46 +18,59 @@ class DACApi:
 
     @spaces.GPU
     def encode_audio(self, audio):
-        # Convert audio to WAV
-        audio = AudioSegment.from_file(audio.name)
-        wav_io = io.BytesIO()
-        audio.export(wav_io, format="wav")
-        wav_io.seek(0)
-
-        # Load audio signal
-        signal = AudioSignal(wav_io)
-        signal = signal.to(self.device)
-
-        # Compress audio
-        print("Compressing audio...")
-        compressed = self.model.compress(signal)
-
-        # Detach the compressed tensor before serialization
-        compressed_detached = compressed.detach()
-
-        # Save compressed audio to BytesIO
-        output = io.BytesIO()
-        compressed_detached.save(output)
-        output.seek(0)
-        
-        return output
+        try:
+            # Convert audio to WAV
+            audio = AudioSegment.from_file(audio.name)
+            wav_io = io.BytesIO()
+            audio.export(wav_io, format="wav")
+            wav_io.seek(0)
+
+            # Load audio signal
+            signal = AudioSignal(wav_io)
+            signal = signal.to(self.device)
+
+            # Compress audio within a no_grad context
+            with torch.no_grad():
+                print("Compressing audio...")  # You can keep this for console logging
+                compressed = self.model.compress(signal)
+
+            # Detach the compressed tensor (additional safety measure)
+            compressed_detached = compressed.detach()
+
+            # Save compressed audio to BytesIO
+            output = io.BytesIO()
+            compressed_detached.save(output)
+            output.seek(0)
+
+            return output
+
+        except Exception as e:
+            # Display error message in a popup
+            gr.Warning(f"An error occurred during encoding: {e}")
+            return None  # Return None to indicate failure
 
     @spaces.GPU
     def decode_audio(self, compressed_file):
-        # Load compressed audio
-        compressed = dac.DACFile.load(compressed_file)
-        compressed = compressed.to(self.device)
-
-        # Decompress audio
-        print("Decompressing audio...")
-        decompressed = self.model.decompress(compressed)
-        
-        # Save decompressed audio to BytesIO
-        output = io.BytesIO()
-        decompressed.write(output, format='wav')
-        output.seek(0)
-        
-        return output
+        try:
+            # Load compressed audio
+            compressed = dac.DACFile.load(compressed_file)
+            compressed = compressed.to(self.device)
+
+            # Decompress audio
+            print("Decompressing audio...")  # You can keep this for console logging
+            decompressed = self.model.decompress(compressed)
+
+            # Save decompressed audio to BytesIO
+            output = io.BytesIO()
+            decompressed.write(output, format='wav')
+            output.seek(0)
+
+            return output
+
+        except Exception as e:
+            # Display error message in a popup
+            gr.Warning(f"An error occurred during decoding: {e}")
+            return None  # Return None to indicate failure
 
 dac_api = DACApi()
 
@@ -72,19 +85,19 @@ def decode(compressed_file):
 # Gradio interface
 with gr.Blocks() as demo:
     gr.Markdown("# Audio Compression with DAC")
-    
+
     with gr.Tab("Encode"):
         audio_input = gr.Audio(type="filepath", label="Input Audio")
         encode_button = gr.Button("Encode")
         encoded_output = gr.File(label="Compressed Audio")
-        
+
         encode_button.click(encode, inputs=audio_input, outputs=encoded_output)
-    
+
     with gr.Tab("Decode"):
         compressed_input = gr.File(label="Compressed Audio")
         decode_button = gr.Button("Decode")
         decoded_output = gr.Audio(label="Decompressed Audio")
-        
+
         decode_button.click(decode, inputs=compressed_input, outputs=decoded_output)
 
-demo.launch()
+demo.queue().launch()