Update app.py

app.py

@@ -4,30 +4,38 @@ import os
 import shutil
 from faster_whisper import WhisperModel
 
-# --- 1. Model Setup (Turbo Settings) ---
+# --- 1. Model Setup ---
 model = None
 
 def load_model():
     global model
     if model is None:
-        print("📥 Loading Whisper Model (Base + Turbo Settings)...")
-        # 'base' model with int8 quantization for speed/accuracy balance
+        print("📥 Loading Whisper Model...")
         model = WhisperModel("base", device="cpu", compute_type="int8")
         print("✅ Model Loaded!")
     return model
 
-# --- 2. Logic: Download Audio from URL ---
+# --- 2. Helper: Find FFmpeg ---
+def get_ffmpeg_dir():
+    # System me ffmpeg kahan hai, ye pata lagao
+    path = shutil.which("ffmpeg")
+    if path:
+        return os.path.dirname(path) # Folder ka rasta return karo
+    return "/usr/bin" # Default fallback
+
+# --- 3. Logic: Download Audio from URL ---
 def download_audio_from_url(url):
     output_path = "downloaded_audio"
-    # Cleanup old files
     if os.path.exists(f"{output_path}.mp3"): os.remove(f"{output_path}.mp3")
 
-    ffmpeg_dir = "/usr/bin" # System Path for Hugging Face
+    # Dynamic FFmpeg Path (Ye error fix karega)
+    ffmpeg_dir = get_ffmpeg_dir()
+    print(f"🔧 FFmpeg found at: {ffmpeg_dir}")
 
     ydl_opts = {
         'format': 'bestaudio/best',
         'outtmpl': output_path,
-        'ffmpeg_location': ffmpeg_dir,
+        'ffmpeg_location': ffmpeg_dir, # <--- FIXED
         'postprocessors': [{
             'key': 'FFmpegExtractAudio',
             'preferredcodec': 'mp3',
@@ -47,12 +55,11 @@ def download_audio_from_url(url):
             ydl.download([url])
         return f"{output_path}.mp3"
     except Exception as e:
-        raise Exception(f"URL Download Error: {str(e)}")
+        raise Exception(f"Download Fail: {str(e)}")
 
-# --- 3. Main Transcribe Function (Handles Both) ---
+# --- 4. Main Transcribe Function (Handles Both) ---
 def transcribe_media(url_input, file_input):
     
-    # Decide source: Priority given to File if both exist, else URL
     audio_file_path = None
     
     try:
@@ -67,15 +74,15 @@ def transcribe_media(url_input, file_input):
             audio_file_path = download_audio_from_url(url_input)
             
         else:
-            return "⚠️ Error: Please provide either a Link or Upload a File."
+            return "⚠️ Error: Link daalein ya File upload karein."
 
-        # --- Transcribe ---
         if not os.path.exists(audio_file_path):
-             return "❌ Error: File not found."
+             return "❌ Error: File nahi mili."
 
+        # --- Transcribe ---
         current_model = load_model()
         
-        # Turbo Settings: beam_size=1 (Fast), vad_filter=True (Skip Silence)
+        # Turbo Settings: beam_size=1 (Fast)
         segments, _ = current_model.transcribe(
             audio_file_path, 
             beam_size=1, 
@@ -88,21 +95,19 @@ def transcribe_media(url_input, file_input):
     except Exception as e:
         return f"❌ Error: {str(e)}"
 
-# --- 4. Turbo UI with Tabs ---
+# --- 5. UI ---
 css = """
 .container {max-width: 900px; margin: auto;}
 .gr-button-primary {background: linear-gradient(90deg, #1CB5E0 0%, #000851 100%); border: none; color: white;}
-.tab-nav {font-weight: bold; font-size: 1.1em;}
 """
 
 with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
     
     with gr.Column(elem_classes="container"):
         gr.Markdown("# 🚀 Turbo Transcriber (Link & Upload)")
-        gr.Markdown("Paste a TikTok link **OR** upload an Audio/Video file to get the text.")
+        gr.Markdown("Paste a TikTok link **OR** upload an Audio/Video file.")
 
         with gr.Tabs():
-            
             # TAB 1: Link
             with gr.TabItem("🔗 Paste Link"):
                 url_in = gr.Textbox(label="TikTok / YouTube URL", placeholder="https://...")
@@ -110,24 +115,13 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
 
             # TAB 2: File Upload
             with gr.TabItem("📂 Upload File"):
-                file_in = gr.Audio(label="Upload Audio or Video", type="filepath", sources=["upload", "microphone"])
+                file_in = gr.Audio(label="Upload File", type="filepath", sources=["upload", "microphone"])
                 btn_file = gr.Button("📂 Transcribe File", variant="primary")
 
-        # Output Area (Common for both)
         transcript_out = gr.Code(label="Transcript Result", language="markdown", interactive=False, lines=15)
 
-    # --- Button Actions ---
-    # Logic: Pass 'None' to the unused input
-    btn_url.click(
-        fn=transcribe_media, 
-        inputs=[url_in, gr.State(None)], # Link diya, File None
-        outputs=transcript_out
-    )
-    
-    btn_file.click(
-        fn=transcribe_media, 
-        inputs=[gr.State(None), file_in], # Link None, File diya
-        outputs=transcript_out
-    )
+    # Actions
+    btn_url.click(fn=transcribe_media, inputs=[url_in, gr.State(None)], outputs=transcript_out)
+    btn_file.click(fn=transcribe_media, inputs=[gr.State(None), file_in], outputs=transcript_out)
 
 demo.launch()