Update app.py

app.py

@@ -31,6 +31,9 @@ CHUNK_LENGTH_S = 30  # Split long audio into 30-second chunks
 SAMPLE_RATE = 16000  # Whisper expects 16kHz audio
 YT_LENGTH_LIMIT_S = 3600  # Limit YouTube videos to 1 hour
 
+# Detect if running on Hugging Face Spaces (YouTube won't work there due to network restrictions)
+IS_HF_SPACE = os.environ.get("SPACE_ID") is not None
+
 # Lazy load state for the Whisper model
 _WHISPER_STATE = {"initialized": False, "pipe": None, "device": "cpu"}
 
@@ -451,50 +454,51 @@ with gr.Blocks(title="Whisper-ASR") as demo:
                 api_name="transcribe",
             )
 
-        # Tab 2: YouTube
-        with gr.TabItem("YouTube"):
-            with gr.Row():
-                with gr.Column():
-                    yt_url_input = gr.Textbox(
-                        label="YouTube URL",
-                        placeholder="Paste a YouTube video URL here...",
-                        lines=1,
-                    )
-
-                    with gr.Row():
-                        yt_task_radio = gr.Radio(
-                            choices=["transcribe", "translate"],
-                            value="transcribe",
-                            label="Task",
-                            info="Translate converts any language to English",
-                        )
-                        yt_language_dropdown = gr.Dropdown(
-                            choices=LANGUAGES,
-                            value="auto",
-                            label="Language",
+        # Tab 2: YouTube (only shown when running locally)
+        if not IS_HF_SPACE:
+            with gr.TabItem("YouTube"):
+                with gr.Row():
+                    with gr.Column():
+                        yt_url_input = gr.Textbox(
+                            label="YouTube URL",
+                            placeholder="Paste a YouTube video URL here...",
+                            lines=1,
                         )
 
-                    yt_timestamps_checkbox = gr.Checkbox(
-                        label="Return Timestamps",
-                        value=False,
-                    )
+                        with gr.Row():
+                            yt_task_radio = gr.Radio(
+                                choices=["transcribe", "translate"],
+                                value="transcribe",
+                                label="Task",
+                                info="Translate converts any language to English",
+                            )
+                            yt_language_dropdown = gr.Dropdown(
+                                choices=LANGUAGES,
+                                value="auto",
+                                label="Language",
+                            )
+
+                        yt_timestamps_checkbox = gr.Checkbox(
+                            label="Return Timestamps",
+                            value=False,
+                        )
 
-                    yt_transcribe_btn = gr.Button("Transcribe YouTube", variant="primary")
+                        yt_transcribe_btn = gr.Button("Transcribe YouTube", variant="primary")
 
-                with gr.Column():
-                    yt_embed = gr.HTML(label="Video")
-                    yt_output = gr.Textbox(
-                        label="Transcription",
-                        placeholder="Transcribed text will appear here...",
-                        lines=10,
-                    )
+                    with gr.Column():
+                        yt_embed = gr.HTML(label="Video")
+                        yt_output = gr.Textbox(
+                            label="Transcription",
+                            placeholder="Transcribed text will appear here...",
+                            lines=10,
+                        )
 
-            yt_transcribe_btn.click(
-                fn=transcribe_youtube_streaming,
-                inputs=[yt_url_input, yt_task_radio, yt_language_dropdown, yt_timestamps_checkbox],
-                outputs=[yt_embed, yt_output],
-                api_name="transcribe_youtube",
-            )
+                yt_transcribe_btn.click(
+                    fn=transcribe_youtube_streaming,
+                    inputs=[yt_url_input, yt_task_radio, yt_language_dropdown, yt_timestamps_checkbox],
+                    outputs=[yt_embed, yt_output],
+                    api_name="transcribe_youtube",
+                )
 
 
 if __name__ == "__main__":