Update app.py

app.py

@@ -58,7 +58,6 @@ def download_audio_youtube(url, output_path="audio.wav", cookies_path=None):
         with yt_dlp.YoutubeDL(ydl_opts) as ydl:
             ydl.download([url])
     except Exception as e:
-        # On failure, run yt-dlp in subprocess to list formats
         try:
             list_cmd = ["yt-dlp", "-F", url]
             if cookies_path:
@@ -66,7 +65,7 @@ def download_audio_youtube(url, output_path="audio.wav", cookies_path=None):
             result = subprocess.run(list_cmd, capture_output=True, text=True, timeout=15)
             formats = result.stdout or "No formats found."
         except Exception as format_err:
-            formats = f"⚠️ Could not list formats due to: {format_err}"
+            formats = f"\u26a0\ufe0f Could not list formats due to: {format_err}"
 
         raise RuntimeError(
             f"yt-dlp failed: {e}\n\n"
@@ -174,14 +173,19 @@ def generate_questions_with_pipeline(text, num_questions=5):
 
 # === MAIN FUNCTION ===
 
-def process_input_gradio(url_input, file_input, cookies_file):
+def process_input_gradio(url_input, file_input, cookies_file, text_input):
     try:
         cookies_path = None
         if cookies_file is not None:
             cookies_path = "cookies.txt"
             shutil.copyfile(cookies_file.name, cookies_path)
 
-        if file_input is not None:
+        transcript = ""
+
+        if text_input:
+            transcript = text_input.strip()
+
+        elif file_input is not None:
             audio_path = extract_audio_from_video(file_input.name)
             chunks, sr = split_audio(audio_path, chunk_length_sec=15)
             transcript = transcribe_chunks_dataset(chunks, sr)
@@ -196,14 +200,14 @@ def process_input_gradio(url_input, file_input, cookies_file):
                         chunks, sr = split_audio(audio_path, chunk_length_sec=15)
                         transcript = transcribe_chunks_dataset(chunks, sr)
                     except Exception as e:
-                        return f"⚠️ Could not download this YouTube video due to restrictions. Please upload the video manually.\nDetails: {e}", ""
+                        return f"\u26a0\ufe0f Could not download this YouTube video due to restrictions. Please upload the video manually.\nDetails: {e}", ""
             else:
                 video_file = download_video_direct(url_input)
                 audio_path = extract_audio_from_video(video_file)
                 chunks, sr = split_audio(audio_path, chunk_length_sec=15)
                 transcript = transcribe_chunks_dataset(chunks, sr)
         else:
-            return "Please provide a URL or upload a video file.", ""
+            return "Please provide a URL, upload a video file, or paste text.", ""
 
         summary = summarize_with_bart(transcript)
         questions = generate_questions_with_pipeline(summary)
@@ -219,13 +223,14 @@ iface = gr.Interface(
         gr.Textbox(label="YouTube or Direct Video URL", placeholder="https://..."),
         gr.File(label="Or Upload a Video File", file_types=[".mp4", ".mkv", ".webm"]),
         gr.File(label="Optional cookies.txt for YouTube", file_types=[".txt"]),
+        gr.Textbox(label="Or Paste Transcript/Text Directly", lines=10, placeholder="Paste transcript or text here...")
     ],
     outputs=[
         gr.Textbox(label="Summary", lines=10),
         gr.Textbox(label="Generated Questions", lines=10),
     ],
     title="Lecture Summary & Question Generator",
-    description="Provide a YouTube/Direct video URL or upload a video file. If the video is restricted, upload cookies.txt or the video file directly."
+    description="Provide a YouTube/Direct video URL, upload a video file, or paste text. If the video is restricted, upload cookies.txt or the video file directly."
 )
 
-iface.launch()
+iface.launch()