support timestamp toggle

app.py

@@ -69,7 +69,7 @@ def download_youtube_audio(youtube_url: str, output_dir: Optional[str] = None) -
         print("Error:", response.status_code, response.text)
         return None  # Return None on failure
 
-def run_asr(audio_file, youtube_url):
+def run_asr(audio_file, youtube_url, with_timestamp):
     temp_file = None
     try:
         if youtube_url:
@@ -80,7 +80,7 @@ def run_asr(audio_file, youtube_url):
             return "Please provide either an audio file or a YouTube URL."
 
         files = {'file': open(audio_file, 'rb')}
-        data = {'language': 'en', 'model_name': 'whisper-large-v2-imda'}
+        data = {'language': 'en', 'model_name': 'whisper-large-v2-imda', 'with_timestamp': with_timestamp}
         response = requests.post(f"{API_URL}/asr", data=data, files=files)
 
         if response.status_code == 200:
@@ -119,6 +119,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             audio_input = gr.Audio(sources=['microphone', 'upload'], type="filepath", label="Audio Input")
             youtube_input = gr.Textbox(label="YouTube URL", placeholder="Or paste a YouTube URL here...")
             video_player = gr.HTML(visible=False)
+            timestamp_toggle = gr.Checkbox(label="Include Timestamps", value=False)
         with gr.Column(scale=3):
             result = gr.Textbox(
                 label="Transcription Result",
@@ -127,7 +128,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             )
 
     run_button = gr.Button("🚀 Transcribe Audio", variant="primary")
-    run_button.click(run_asr, inputs=[audio_input, youtube_input], outputs=[result])
+    run_button.click(run_asr, inputs=[audio_input, youtube_input, timestamp_toggle], outputs=[result])
 
     # Update video player and clear transcription and audio input when YouTube URL is entered
     youtube_input.change(
@@ -151,4 +152,4 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown("5. Wait for a few seconds, and your transcription will appear in the result box.")
 
 # Launch the Gradio interface
-demo.launch()
+demo.launch(server_name='0.0.0.0')