add application file

app.py

@@ -0,0 +1,143 @@
+import gradio as gr
+import requests
+from yt_dlp import YoutubeDL
+import os
+import tempfile
+from typing import Optional
+import json
+import subprocess
+
+# Define the FastAPI URL
+API_URL = "http://localhost:9998"
+
+def download_youtube_audio(url: str, output_dir: Optional[str] = None) -> str:
+    if output_dir is None:
+        output_dir = tempfile.gettempdir()
+
+    yt_dlp_command = [
+        "/home/ubuntu/miniconda3/envs/gradio_whisper/bin/yt-dlp",
+        "-f", "bestaudio/best",
+        "-x",  # Extract audio
+        "--audio-format", "mp3",
+        "--audio-quality", "192K",
+        "-o", os.path.join(output_dir, "%(id)s.%(ext)s"),
+        "-v",
+        "--print", "after_move:filepath",  # Print the output filepath
+        "--username", "oauth2",
+        "--password", "", 
+        "--no-playlist",
+        "--print-json",
+        url
+    ]
+
+    try:
+        print(' '.join(yt_dlp_command))
+        result = subprocess.run(yt_dlp_command, capture_output=True, text=True, check=True)
+        
+        # Parse the JSON output to get video information
+        video_info = json.loads(result.stdout.splitlines()[-2])  # The last line is the filepath
+        
+        # Get the output filepath from the last line of stdout
+        audio_file = result.stdout.splitlines()[-1].strip()
+        
+        print(f"Successfully downloaded: {video_info['title']}")
+        return audio_file
+    except subprocess.CalledProcessError as e:
+        raise Exception(f"Error downloading YouTube audio: {e.stderr}")
+    except json.JSONDecodeError:
+        raise Exception("Error parsing video information")
+    except Exception as e:
+        raise Exception(f"Unexpected error: {str(e)}")
+
+
+def run_asr(audio_file, youtube_url):
+    temp_file = None
+    try:
+        if youtube_url:
+            # It's a YouTube URL
+            audio_file = download_youtube_audio(youtube_url)
+            temp_file = audio_file
+        elif not audio_file:
+            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'}
+        response = requests.post(f"{API_URL}/asr", data=data, files=files)
+
+        if response.status_code == 200:
+            return response.json().get("text", "")
+        else:
+            return f"Error: {response.status_code}"
+    except Exception as e:
+        return f"Error: {str(e)}"
+    finally:
+        # Clean up the temporary file if it was a YouTube download
+        if temp_file and os.path.exists(temp_file):
+            os.remove(temp_file)
+
+def embed_youtube(youtube_url):
+    if youtube_url:
+        try:
+            # video_id = YoutubeDL().extract_info(youtube_url, download=False)['id']\
+            video_id = youtube_url.split("v=")[1]
+            print(video_id)
+            embed_html = f'<iframe width="560" height="315" src="https://www.youtube.com/embed/{video_id}" frameborder="0" allow="autoplay; encrypted-media" allowfullscreen></iframe>'
+            return gr.update(value=embed_html, visible=True), "", None
+        except:
+            return gr.update(value="", visible=False), "Invalid YouTube URL", None
+    return gr.update(value="", visible=False), "", None
+
+def clear_on_audio_input(audio):
+    if audio is not None:
+        return "", gr.update(value="", visible=False), ""
+    return gr.update(), gr.update(), gr.update()
+
+# Create the Gradio interface with improved aesthetics
+with gr.Blocks(theme=gr.themes.Soft()) as demo:
+    gr.Markdown("# 🎙️ Audio Transcription Service")
+    gr.Markdown("Upload an audio file, record your voice, or paste a YouTube URL to get an English transcription.")
+    
+    with gr.Row():
+        with gr.Column(scale=2):
+            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)
+        with gr.Column(scale=3):
+            result = gr.Textbox(
+                label="Transcription Result",
+                placeholder="Your transcription will appear here...",
+                lines=10
+            )
+
+    run_button = gr.Button("🚀 Transcribe Audio", variant="primary")
+    run_button.click(run_asr, inputs=[audio_input, youtube_input], outputs=[result])
+
+    # Update video player and clear transcription and audio input when YouTube URL is entered
+    youtube_input.change(
+        fn=embed_youtube,
+        inputs=[youtube_input],
+        outputs=[video_player, result, audio_input]
+    )
+
+    # Clear transcription, YouTube input, and video player when audio is input
+    audio_input.change(
+        fn=clear_on_audio_input,
+        inputs=[audio_input],
+        outputs=[result, video_player, youtube_input]
+    )
+
+    gr.Markdown("### How to use:")
+    gr.Markdown("1. Upload an audio file or record your voice using the microphone, OR paste a YouTube URL.")
+    gr.Markdown("2. If you paste a YouTube URL, the video will be displayed for your reference, and any previous transcription or audio input will be cleared.")
+    gr.Markdown("3. If you upload or record audio, any previous transcription, YouTube URL, and video will be cleared.")
+    gr.Markdown("4. Click the 'Transcribe Audio' button to start the process.")
+    gr.Markdown("5. Wait for a few seconds, and your transcription will appear in the result box.")
+
+# Launch the Gradio interface
+demo.launch(
+    server_name='0.0.0.0', 
+    server_port=5008, 
+    ssl_certfile='/home/ubuntu/astarwiz_com/astarwiz_com.crt', 
+    ssl_keyfile='/home/ubuntu/astarwiz_com/astarwiz_com.key', 
+    ssl_verify=False
+)