Update app.py

app.py

@@ -26,10 +26,14 @@ cancel_encode = False
 cancel_decode = False
 cancel_stream = False
 
-@spaces.GPU(duration=30)
+@spaces.GPU(duration=30)  # Changed from 250 to 30
 def encode_audio(audio_file_path):
     global cancel_encode
 
+    if audio_file_path is None:
+        print("No audio file provided")
+        return None
+
     try:
         # Load the audio file
         waveform, sample_rate = torchaudio.load(audio_file_path)
@@ -79,10 +83,10 @@ def encode_audio(audio_file_path):
 # Add this function to handle the output
 def handle_encode_output(file_path):
     if file_path is None:
-        return None, gr.Markdown("Encoding failed. Please check the input file and try again.", visible=True)
+        return None, gr.Markdown("Encoding failed. Please ensure you've uploaded an audio file and try again.", visible=True)
     return file_path, gr.Markdown(visible=False)
 
-@spaces.GPU(duration=30)
+@spaces.GPU(duration=30)  # Changed from 250 to 30
 def decode_audio(encoded_file_path):
     global cancel_decode
 
@@ -123,7 +127,7 @@ def decode_audio(encoded_file_path):
     finally:
         cancel_decode = False  # Reset cancel flag after decoding
 
-@spaces.GPU(duration=30)
+@spaces.GPU(duration=30)  # Changed from 250 to 30
 async def stream_decode_audio(encoded_file_path) -> Generator[tuple, None, None]:
     global cancel_stream
 
@@ -168,8 +172,13 @@ with gr.Blocks() as demo:
         encoded_output = gr.File(label="Encoded File (.owie)", type="filepath")  # Using "filepath" mode
         encode_error_message = gr.Markdown(visible=False)
 
+        def encode_wrapper(audio):
+            if audio is None:
+                return None, gr.Markdown("Please upload an audio file before encoding.", visible=True)
+            return handle_encode_output(encode_audio(audio))
+
         encode_button.click(
-            lambda x: handle_encode_output(encode_audio(x)),
+            encode_wrapper,
             inputs=input_audio,
             outputs=[encoded_output, encode_error_message]
         )