Update app.py

app.py

@@ -1,33 +1,40 @@
 import gradio as gr
-from youtube_transcript_api import YouTubeTranscriptApi
 from deep_translator import GoogleTranslator
 import re
 
-# ✅ Extract video ID from YouTube URL
+# Extract video ID from URL
 def extract_video_id(url):
-    # Supports standard, short, and embed URLs
     regex = r"(?:v=|\/)([0-9A-Za-z_-]{11})"
     match = re.search(regex, url)
     return match.group(1) if match else url.strip()
 
-# ✅ Agent 1: Summarizer
+# Simulated transcript agent
 def summarize_youtube(video_url):
     try:
         video_id = extract_video_id(video_url)
-        transcript = YouTubeTranscriptApi.get_transcript(video_id)
-        text = " ".join([entry["text"] for entry in transcript[:15]])
-        return text[:500]
+
+        # Predefined summaries for demonstration
+        saved_transcripts = {
+            "dQw4w9WgXcQ": "We're no strangers to love. You know the rules and so do I...",
+            "F9cTlfD7ZGM": "This is a video about AI agents and their collaboration.",
+            "HMcFwjWVprs": "In this talk, we cover the future of GenAI protocols and their applications."
+        }
+
+        if video_id not in saved_transcripts:
+            return "❌ Transcript not available for this video. Please use a supported demo video."
+
+        return saved_transcripts[video_id]
     except Exception as e:
-        return f"❌ Error in Summarizer: {str(e)}"
+        return f"❌ Error: {str(e)}"
 
-# ✅ Agent 2: Translator
+# Translator agent
 def translate_to_spanish(english_text):
     try:
         return GoogleTranslator(source='auto', target='es').translate(english_text)
     except Exception as e:
-        return f"❌ Error in Translator: {str(e)}"
+        return f"❌ Error: {str(e)}"
 
-# ✅ Workflow Function (Master Agent)
+# Workflow (master agent)
 def agent_workflow(video_url):
     summary = summarize_youtube(video_url)
     if summary.startswith("❌"):
@@ -35,7 +42,7 @@ def agent_workflow(video_url):
     translated = translate_to_spanish(summary)
     return summary, translated
 
-# ✅ Gradio UI
+# Gradio UI
 demo = gr.Interface(
     fn=agent_workflow,
     inputs=gr.Textbox(label="🎥 Enter YouTube Video Link"),
@@ -43,8 +50,5 @@ demo = gr.Interface(
         gr.Textbox(label="🧠 English Summary"),
         gr.Textbox(label="🌍 Spanish Translation")
     ],
-    title="Lead with AI Agents Hackathon Project",
-    description="🧠 Agent 1: Summarizes YouTube Video · 🌍 Agent 2: Translates Summary to Spanish"
-)
-
-demo.launch()
+    title="🧠 AI Agents: YouTube Summary + Spanish Translation",
+    description="Agent 1: Summarizer (simulated)