Update app.py

app.py

@@ -1,22 +1,48 @@
 import gradio as gr
+import yt_dlp
+import os
+import re
 
-def chat_ocr(message, history):
-    # Placeholder response
-    return "Hi! This is your OCR training assistant chat. How can I help you today?"
+def is_youtube_url(url):
+    youtube_regex = (
+        r'(https?://)?(www\.)?'
+        r'(youtube|youtu|youtube-nocookie)\.(com|be)/'
+        r'(watch\?v=|embed/|v/|.+\?v=)?([^&=%\?]{11})'
+    )
+    return re.match(youtube_regex, url) is not None
+
+def chat_handler(message, history):
+    if is_youtube_url(message.strip()):
+        os.makedirs('downloads', exist_ok=True)
+        ydl_opts = {
+            'format': 'best[height<=720]',
+            'outtmpl': 'downloads/%(title)s.%(ext)s',
+        }
+        try:
+            with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+                info = ydl.extract_info(message, download=True)
+                filename = ydl.prepare_filename(info)
+                ext = os.path.splitext(filename)[1].lstrip('.')
+                video_src = f"/file/{os.path.basename(filename)}"
+                response = f"Here's the downloaded video:\n<video width=\"640\" controls><source src=\"{video_src}\" type=\"video/{ext}\"></video>"
+                return response
+        except Exception as e:
+            return f"Error downloading the video: {str(e)}"
+    else:
+        return "Hi! Send me a YouTube video link, and I'll download it for you."
 
-with gr.Blocks(title="OCR Chat Interface") as demo:
-    gr.Markdown("# 💬 OCR Training Chat")
-    gr.Markdown("Chat with your assistant about OCR model training and setup.")
+with gr.Blocks(title="YouTube Downloader Chat") as demo:
+    gr.Markdown("# 📹 YouTube Video Downloader Chat")
+    gr.Markdown("Paste a YouTube link in the chat, and I'll fetch the video for you.")
 
     chatbot = gr.ChatInterface(
-        fn=chat_ocr,
-        title="Multilingual OCR Assistant",
+        fn=chat_handler,
+        title="YouTube Downloader Assistant",
         examples=[
-            "How do I start training?",
-            "Which languages are supported?",
-            "What model should I use for Persian text?"
+            "https://www.youtube.com/watch?v=dQw4w9WgXcQ",
+            "https://youtu.be/dQw4w9WgXcQ"
         ]
     )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()