Update app.py

app.py

@@ -4,59 +4,47 @@ import os
 import shutil
 import subprocess
 from faster_whisper import WhisperModel
-
-# 🔤 Hindi Script Fix
 from indic_transliteration import sanscript
 from indic_transliteration.sanscript import transliterate
 
 # ===============================
-# 1. Whisper Model (Lazy Load)
+# Whisper Model
 # ===============================
 model = None
 
 def load_model():
     global model
     if model is None:
-        print("📥 Loading Whisper Model...")
         model = WhisperModel("base", device="cpu", compute_type="int8")
-        print("✅ Model Loaded")
     return model
 
 # ===============================
-# 2. FFmpeg Path
+# FFmpeg
 # ===============================
 def get_ffmpeg():
     return shutil.which("ffmpeg") or "/usr/bin/ffmpeg"
 
 # ===============================
-# 3. Video → Audio
+# Video → Audio
 # ===============================
 def extract_audio(video_path):
-    audio_path = "uploaded_audio.wav"
-    if os.path.exists(audio_path):
-        os.remove(audio_path)
-
-    cmd = [
-        get_ffmpeg(),
-        "-i", video_path,
-        "-vn",
-        "-ac", "1",
-        "-ar", "16000",
-        audio_path,
-        "-y"
-    ]
-    subprocess.run(cmd, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
-    return audio_path
+    audio = "uploaded_audio.wav"
+    if os.path.exists(audio):
+        os.remove(audio)
+    subprocess.run(
+        [get_ffmpeg(), "-i", video_path, "-vn", "-ac", "1", "-ar", "16000", audio, "-y"],
+        stdout=subprocess.DEVNULL,
+        stderr=subprocess.DEVNULL
+    )
+    return audio
 
 # ===============================
-# 4. Download Audio from URL
+# URL → Audio
 # ===============================
-def download_audio_from_url(url):
-    output = "url_audio"
-
+def download_audio(url):
     ydl_opts = {
         "format": "bestaudio/best",
-        "outtmpl": output,
+        "outtmpl": "url_audio",
         "postprocessors": [{
             "key": "FFmpegExtractAudio",
             "preferredcodec": "wav",
@@ -64,14 +52,12 @@ def download_audio_from_url(url):
         "quiet": True,
         "nocheckcertificate": True,
     }
-
     with yt_dlp.YoutubeDL(ydl_opts) as ydl:
         ydl.download([url])
-
     return "url_audio.wav"
 
 # ===============================
-# 5. Hindi Script Normalizer
+# Script Fix
 # ===============================
 def normalize_script(text, lang):
     if lang == "hi":
@@ -82,91 +68,95 @@ def normalize_script(text, lang):
     return text
 
 # ===============================
-# 6. Main Transcribe Logic
+# Transcribe
 # ===============================
-def transcribe_media(url_input, file_input, language_choice):
-
+def transcribe(url, file, lang_choice):
     try:
-        audio_path = None
-
-        # ---------- FILE ----------
-        if file_input:
-            ext = os.path.splitext(file_input)[1].lower()
-            if ext in [".mp3", ".wav", ".m4a"]:
-                audio_path = file_input
-            else:
-                audio_path = extract_audio(file_input)
-
-        # ---------- URL ----------
-        elif url_input and url_input.strip():
-            audio_path = download_audio_from_url(url_input)
-
+        if file:
+            ext = os.path.splitext(file)[1].lower()
+            audio = file if ext in [".mp3", ".wav", ".m4a"] else extract_audio(file)
+        elif url:
+            audio = download_audio(url)
         else:
-            return "⚠️ Please paste a link or upload a file."
-
-        if not os.path.exists(audio_path):
-            return "❌ Audio processing failed."
+            return "⚠️ Please provide a URL or upload a file."
 
         model = load_model()
-
-        # Language handling
-        language = None if language_choice == "Auto Detect" else language_choice
+        language = None if lang_choice == "Auto Detect" else lang_choice
 
         segments, info = model.transcribe(
-            audio_path,
+            audio,
             beam_size=1,
             vad_filter=True,
             language=language
         )
 
-        detected_lang = info.language
-        raw_text = " ".join(seg.text for seg in segments)
-        final_text = normalize_script(raw_text, detected_lang)
+        text = " ".join(s.text for s in segments)
+        text = normalize_script(text, info.language)
 
-        return f"🌍 Detected Language: {detected_lang}\n\n{final_text.strip()}"
+        return f"🌍 Language: {info.language}\n\n{text.strip()}"
 
     except Exception as e:
         return f"❌ Error: {str(e)}"
 
 # ===============================
-# 7. UI
+# MODERN UI
 # ===============================
 css = """
-.container {max-width: 900px; margin: auto;}
+body {
+    background: radial-gradient(circle at top, #0f2027, #203a43, #2c5364);
+}
+.glass {
+    background: rgba(255,255,255,0.08);
+    backdrop-filter: blur(18px);
+    border-radius: 18px;
+    padding: 25px;
+    box-shadow: 0 20px 40px rgba(0,0,0,0.4);
+}
 .gr-button-primary {
-    background: linear-gradient(90deg,#667eea,#764ba2);
+    background: linear-gradient(135deg,#00c6ff,#0072ff);
     border: none;
     color: white;
+    font-weight: 600;
+}
+.gr-input, .gr-textarea {
+    background: rgba(255,255,255,0.12) !important;
+    color: white !important;
+}
+h1, h2, label, .markdown-text {
+    color: #ffffff !important;
 }
 """
 
-with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
-    with gr.Column(elem_classes="container"):
+with gr.Blocks(css=css, theme=gr.themes.Base()) as demo:
+    with gr.Column(elem_classes="glass"):
         gr.Markdown("## 🚀 Universal Transcript Tool")
         gr.Markdown(
-            "Supports **YouTube, TikTok, Instagram, Facebook, Twitter/X**\n\n"
-            "Hindi output is always **Devanagari** 🇮🇳"
+            "Modern UI • YouTube • TikTok • Instagram (Upload) • Fast Whisper\n\n"
+            "**Note:** Instagram URLs may be blocked on Hugging Face."
         )
 
         with gr.Tabs():
             with gr.TabItem("🔗 Paste Link"):
-                url_in = gr.Textbox(label="Video URL")
+                url = gr.Textbox(
+                    label="Video URL",
+                    placeholder="YouTube / TikTok link"
+                )
                 btn_url = gr.Button("🎧 Transcribe Link", variant="primary")
 
             with gr.TabItem("📂 Upload File"):
-                file_in = gr.File(
+                file = gr.File(
                     label="Upload Video / Audio",
                     file_types=[".mp4", ".mkv", ".mov", ".webm", ".avi", ".mp3", ".wav"]
                 )
                 btn_file = gr.Button("📂 Transcribe File", variant="primary")
 
-        # 🌍 Language Selector
-        language_selector = gr.Dropdown(
+        lang = gr.Dropdown(
+            label="🌍 Transcript Language",
             choices=[
                 "Auto Detect",
-                "hi",  # Hindi (Devanagari)
-                "ur",  # Urdu
-                "en",  # English
+                "hi",
+                "ur",
+                "en",
                 "ar",
                 "fr",
                 "de",
@@ -175,13 +165,15 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
                 "ja",
                 "zh"
             ],
-            value="Auto Detect",
-            label="🌍 Select Transcript Language"
+            value="Auto Detect"
         )
 
-        output = gr.Code(label="Transcript Output", lines=15)
+        output = gr.Code(
+            label="Transcript Output",
+            lines=14
+        )
 
-    btn_url.click(transcribe_media, [url_in, gr.State(None), language_selector], output)
-    btn_file.click(transcribe_media, [gr.State(None), file_in, language_selector], output)
+    btn_url.click(transcribe, [url, gr.State(None), lang], output)
+    btn_file.click(transcribe, [gr.State(None), file, lang], output)
 
 demo.launch()