Update app.py

app.py

@@ -40,9 +40,17 @@ pretrains = {
     }
 }
 
+def get_audio_duration(audio_file):
+    with wave.open(audio_file, 'rb') as audio:
+        frames = audio.getnframes()
+        rate = audio.getframerate()
+        duration = frames / float(rate)
+    return duration
+
 def get_training_info(audio_file):
     if audio_file is None:
         return 'Please provide an audio file!'
+    
     duration = get_audio_duration(audio_file)
     sample_rate = wave.open(audio_file, 'rb').getframerate()
 
@@ -60,17 +68,14 @@ def get_training_info(audio_file):
         if min_duration <= duration < max_duration:
             break
     else:
-        return 'Duration is not within the specified range dude'
-
-    return f'u should use the **{pretrain}** pretrain with **{epochs}** epochs at **{sample_rate/1000}khz** sample rate\n good luck at training ;).'
-
+        return 'Duration is not within the specified range.'
 
+    return f'You should use the **{pretrain}** pretrain with **{epochs}** epochs at **{sample_rate/1000}kHz** sample rate. Good luck with your training!'
 
 with gr.Blocks() as demo:
-    audio_p = gr.Audio(label="Your Audio here")
-    audio_q = Textbox(label="Your Output here")
+    audio_p = gr.Audio(source="upload", type="filepath", label="Your Audio here")
+    audio_q = gr.Textbox(label="Your Output here")
     wtar = gr.Button("Start!")
     wtar.click(get_training_info, inputs=[audio_p], outputs=[audio_q])
 
-
-    demo.launch(debug=True)
+    demo.launch(debug=True)