Update app.py

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 from transformers import pipeline
+from pydub import AudioSegment
 
 pipe = pipeline("automatic-speech-recognition", "facebook/wav2vec2-base-960h")
 
@@ -45,14 +46,40 @@ dictionary = {' ': '/',
 }
 
 
+
 def asr(speech):
+  dummy = AudioSegment.from_file("dummy.mp3", format="mp3")
+  dot = AudioSegment.from_file("dot.mp3", format="mp3")
+  dash = AudioSegment.from_file("dash.mp3", format="mp3")
+  space = AudioSegment.from_file("space.mp3", format="mp3")
+  i=0
   transcript = pipe(speech)['text']
   morseCode = ""
   for character in transcript:
     morseCode += dictionary[character] + " "
-  return morseCode
+  while i<len(morseCode):
+    if morseCode[i] == ".":
+      dummy = dummy + dot
+      i = i+1
+    elif morseCode[i] == "-":
+      dummy = dummy + dash
+      i = i+1
+    elif morseCode[i] == " ":
+      dummy = dummy + space
+      i = i+1
+    else:
+      i = i+1
+  morseAudio = dummy.export("output.mp3", format="mp3")
+  return transcript, morseCode, "output.mp3"
   
 gr.Interface(fn=asr,
-             inputs = gr.inputs.Audio(source="upload", type="filepath", label="Upload your audio file here"),
-             outputs = gr.outputs.Textbox(type="str",label="Output Text"),
+             inputs = gr.inputs.Audio(source="microphone", type="filepath", optional=False, label="Please record your voice"),
+#             inputs = gr.inputs.Audio(source="upload", type="filepath", label="Upload your audio file here"),
+						 outputs = [gr.outputs.Textbox(type="str", label="Text Translation"),
+						            gr.outputs.Textbox(type="str", label="Morse Code"),
+												gr.outputs.Audio(type="file", label="Audio Translation")],
+             title = "Speech to Morse Code",
+             description = "This app will translate your speech into text, morse code, and audio.",
+						 examples = [["hello world.mp3"], ["intro.mp3"]],
+						 article = "Author: <a href=\"https://huggingface.co/epdavid2\">Edson</a>"
              ).launch()