Update app.py

app.py

@@ -24,15 +24,23 @@ def encode_audio(audio_path):
         tokens = tokens.cpu().numpy()
     
     # Save to a temporary file
-    with tempfile.NamedTemporaryFile(suffix='.oterin', delete=False) as tmp_file:
-        np.save(tmp_file.name, tokens)
-        token_path = tmp_file.name
+    temp_file = tempfile.NamedTemporaryFile(suffix='.oterin', delete=False)
+    temp_file.close()  # Close the file before writing to it
+    np.save(temp_file.name, tokens)
     
-    return token_path, f"Encoded to {len(tokens)} tokens"
+    # Ensure the file exists and has content
+    if os.path.exists(temp_file.name) and os.path.getsize(temp_file.name) > 0:
+        return temp_file.name, f"Encoded to {len(tokens)} tokens"
+    else:
+        raise Exception("Failed to create token file")
 
 @spaces.GPU(duration=60)
 def decode_tokens(token_path):
     """Decode tokens to audio"""
+    # Ensure the file exists and has content
+    if not os.path.exists(token_path) or os.path.getsize(token_path) == 0:
+        return None, "Error: Empty or missing token file"
+    
     tokens = np.load(token_path)
     # Convert to torch tensor if needed by the model
     if hasattr(semanticodec, 'decode_requires_tensor') and semanticodec.decode_requires_tensor:
@@ -47,6 +55,10 @@ def decode_tokens(token_path):
     sf.write(output_buffer, waveform[0, 0], 32000, format='WAV')
     output_buffer.seek(0)
     
+    # Verify the buffer has content
+    if output_buffer.getbuffer().nbytes == 0:
+        return None, "Error: Failed to generate audio"
+    
     return output_buffer, f"Decoded {len(tokens)} tokens to audio"
 
 @spaces.GPU(duration=80)
@@ -69,6 +81,10 @@ def process_both(audio_path):
     sf.write(output_buffer, waveform[0, 0], 32000, format='WAV')
     output_buffer.seek(0)
     
+    # Verify the buffer has content
+    if output_buffer.getbuffer().nbytes == 0:
+        return None, "Error: Failed to generate audio"
+    
     return output_buffer, f"Encoded to {len(tokens)} tokens\nDecoded {len(tokens)} tokens to audio"
 
 # Create Gradio interface