Update app.py

app.py

@@ -3,10 +3,9 @@ import gradio as gr
 import yt_dlp
 import whisper
 from transformers import pipeline, MarianMTModel, MarianTokenizer
-from pytube import YouTube
 import torch
 
-# Load Whisper (tiny for CPU)
+# Load Whisper model
 whisper_model = whisper.load_model("small")
 
 # Load summarizer
@@ -27,10 +26,20 @@ def download_audio(youtube_url):
         ydl.download([youtube_url])
     return output_file
 
+def get_thumbnail(youtube_url):
+    ydl_opts = {'quiet': True}
+    with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+        info = ydl.extract_info(youtube_url, download=False)
+        return info.get("thumbnail", "")
+
 def translate_to_english(text):
-    inputs = translation_tokenizer(text, return_tensors="pt", truncation=True, max_length=512)
-    translated = translation_model.generate(**inputs, max_length=512)
-    return translation_tokenizer.decode(translated[0], skip_special_tokens=True)
+    chunks = [text[i:i+500] for i in range(0, len(text), 500)]
+    translated = []
+    for chunk in chunks:
+        inputs = translation_tokenizer(chunk, return_tensors="pt", truncation=True, max_length=512)
+        output = translation_model.generate(**inputs, max_length=512)
+        translated.append(translation_tokenizer.decode(output[0], skip_special_tokens=True))
+    return " ".join(translated)
 
 def process_video(url):
     audio_path = download_audio(url)
@@ -43,10 +52,9 @@ def process_video(url):
     summary = summarizer(translated_text, max_length=130, min_length=30, do_sample=False)[0]["summary_text"]
 
     # Get thumbnail
-    yt = YouTube(url)
-    thumbnail_url = yt.thumbnail_url
+    thumbnail_url = get_thumbnail(url)
 
-    return transcription, translated_text, summary, thumbnail_url, summary
+    return transcription, translated_text, summary, thumbnail_url
 
 def download_summary(text):
     filename = "summary.txt"
@@ -54,6 +62,7 @@ def download_summary(text):
         f.write(text)
     return filename
 
+# UI
 with gr.Blocks(theme=gr.themes.Soft(), title="🎥 YouTube Video Summarizer with LLaMA") as demo:
     gr.Markdown("## 🧠 Multilingual YouTube Summarizer")
     gr.Markdown("Upload a video link and get the transcript, English translation, and summary.")
@@ -71,8 +80,17 @@ with gr.Blocks(theme=gr.themes.Soft(), title="🎥 YouTube Video Summarizer with
             download_file = gr.File(label="Download Link")
         video_thumb = gr.Image(label="🎞️ Video Thumbnail", width=256)
 
-    submit_btn.click(fn=process_video, inputs=[youtube_input],
-                     outputs=[transcript_output, translation_output, summary_output, video_thumb, download_file])
-    download_btn.click(fn=download_summary, inputs=[summary_output], outputs=[download_file])
+    # Button actions
+    submit_btn.click(
+        fn=process_video,
+        inputs=[youtube_input],
+        outputs=[transcript_output, translation_output, summary_output, video_thumb]
+    )
+
+    download_btn.click(
+        fn=download_summary,
+        inputs=[summary_output],
+        outputs=[download_file]
+    )
 
-demo.launch()
+demo.launch(share=True)