Added Timestamps option

app.py

@@ -5,21 +5,34 @@ import gradio as gr
 model = whisper.load_model('base')
 
 
-def transcribe(inputs, task):
-    if inputs is None:
-        raise gr.Error("No audio file submitted! Please upload or record an audio file before submitting your request.")
 
+def transcribe(inputs , timestamp):
+    if inputs is None:
+          raise gr.Error("No audio file submitted! Please upload or record an audio file before submitting your request.")
+    output = ""
     result = model.transcribe(inputs)
-    return  result["text"]
+    if timestamp == "Yes":
+      for indx, segment in enumerate(result['segments']):
+        output += str(datetime.timedelta (seconds=segment['start'])) +" "+ str(datetime.timedelta (seconds=segment['end'])) + "\n"
+        output += segment['text'].strip() + '\n'
+    else:
+      output = result["text"]
+
+
+    print(result)
+    return  output
+
+
 
 
 interface = gr.Interface(
     fn=transcribe,
-    inputs=gr.Audio(sources=["upload"],type="filepath"),
+    inputs=[gr.Audio(sources=["upload"],type="filepath"),
+            gr.Radio(["Yes", "No"], label="Timestamp", info="Displays with timestamp if needed."),],
     outputs="text",
     title="Whisper Large V3: Transcribe Audio",
     description=(
-        "Transcribe long-form microphone or audio inputs with the click of a button! Demo uses the OpenAI Whisper"
+        "Transcribe long-form microphone or audio inputs with the click of a button! Demo uses the OpenAI Whisper API"
     )
 )