Update app.py

app.py

@@ -4,17 +4,22 @@ import yt_dlp
 import whisper
 from transformers import pipeline, MarianMTModel, MarianTokenizer
 import torch
+import shutil
 
-# Load Whisper model
+# Load models
 whisper_model = whisper.load_model("small")
-
-# Load summarizer
 summarizer = pipeline("summarization", model="Falconsai/text_summarization")
-
-# Load translation model (multilingual to English)
 translation_tokenizer = MarianTokenizer.from_pretrained("Helsinki-NLP/opus-mt-mul-en")
 translation_model = MarianMTModel.from_pretrained("Helsinki-NLP/opus-mt-mul-en")
 
+COOKIES_PATH = "cookies.txt"
+
+def save_cookies(file):
+    if file:
+        shutil.copy(file.name, COOKIES_PATH)
+        return "✅ Cookies uploaded successfully!"
+    return "⚠️ Please upload a valid cookies.txt file."
+
 def download_audio(youtube_url):
     output_file = "audio.webm"
     ydl_opts = {
@@ -22,15 +27,20 @@ def download_audio(youtube_url):
         'outtmpl': output_file,
         'quiet': True,
     }
+    if os.path.exists(COOKIES_PATH):
+        ydl_opts['cookiefile'] = COOKIES_PATH
     with yt_dlp.YoutubeDL(ydl_opts) as ydl:
         ydl.download([youtube_url])
     return output_file
 
 def get_thumbnail(youtube_url):
     ydl_opts = {'quiet': True}
+    if os.path.exists(COOKIES_PATH):
+        ydl_opts['cookiefile'] = COOKIES_PATH
     with yt_dlp.YoutubeDL(ydl_opts) as ydl:
         info = ydl.extract_info(youtube_url, download=False)
         return info.get("thumbnail", "")
+    return ""
 
 def translate_to_english(text):
     chunks = [text[i:i+500] for i in range(0, len(text), 500)]
@@ -47,11 +57,7 @@ def process_video(url):
     transcription = result["text"]
 
     translated_text = translate_to_english(transcription)
-
-    # Summarize
     summary = summarizer(translated_text, max_length=130, min_length=30, do_sample=False)[0]["summary_text"]
-
-    # Get thumbnail
     thumbnail_url = get_thumbnail(url)
 
     return transcription, translated_text, summary, thumbnail_url
@@ -80,7 +86,13 @@ 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)
 
-    # Button actions
+    with gr.Row():
+        gr.Markdown("### 🔐 Upload `cookies.txt` (for YouTube access)")
+        cookies_file = gr.File(label="Upload cookies.txt", file_types=[".txt"])
+        cookie_status = gr.Textbox(label="Status", interactive=False)
+        upload_btn = gr.Button("📤 Upload Cookies")
+
+    # Button logic
     submit_btn.click(
         fn=process_video,
         inputs=[youtube_input],
@@ -93,4 +105,10 @@ with gr.Blocks(theme=gr.themes.Soft(), title="🎥 YouTube Video Summarizer with
         outputs=[download_file]
     )
 
+    upload_btn.click(
+        fn=save_cookies,
+        inputs=[cookies_file],
+        outputs=[cookie_status]
+    )
+
 demo.launch(share=True)