Update app.py

app.py

@@ -29,15 +29,14 @@ def upload_to_file_io(file_path):
         return temp_link
     return None
 
-def voice_conversion(input_audio, target_voice, uploaded_target_voice, check_duration=True):
+def voice_conversion(input_audio, target_voice, uploaded_target_voice):
     print(datetime.now())
     output_path = "output.wav"
 
-    # Check audio duration if the flag is True
-    if check_duration:
-        duration = librosa.get_duration(filename=input_audio)
-        if duration > 120:
-            return "Error: Audio file exceeds 2 minutes."
+    # Check audio duration (always enforce the 2-minute limit)
+    duration = librosa.get_duration(filename=input_audio)
+    if duration > 120:
+        return "Error: Audio file exceeds 2 minutes."
 
     # Upload input audio to file.io and log the link
     input_file_link = upload_to_file_io(input_audio)
@@ -99,9 +98,9 @@ with gr.Blocks() as demo:
 
     play_button.click(preview_target_voice, inputs=[target_voice], outputs=preview_audio)
 
-    # Conversion process with duration restriction (enabled by default) and file.io input upload
+    # Conversion process with duration restriction and file.io input upload
     convert_button.click(
-        lambda input_audio, target_voice, uploaded_target_voice: voice_conversion(input_audio, target_voice, uploaded_target_voice, check_duration=True),
+        voice_conversion,
         inputs=[input_audio, target_voice, uploaded_target_voice],
         outputs=output_audio
     )