Update app.py

app.py

@@ -2,13 +2,12 @@ import gradio as gr
 from deep_translator import GoogleTranslator
 import re
 
-# ✅ Extract video ID from YouTube URL
+# ✅ Extract video ID from any YouTube URL format
 def extract_video_id(url):
-    regex = r"(?:v=|\/)([0-9A-Za-z_-]{11})"
-    match = re.search(regex, url)
+    match = re.search(r"(?:v=|\/)([0-9A-Za-z_-]{11})", url)
     return match.group(1) if match else url.strip()
 
-# ✅ Agent 1: Summarizer (Simulated)
+# ✅ Agent 1: Simulated YouTube Summary
 def summarize_youtube(video_url):
     try:
         video_id = extract_video_id(video_url)
@@ -20,18 +19,32 @@ def summarize_youtube(video_url):
         }
 
         if video_id not in saved_transcripts:
-            return "❌ Transcript not available for this video.", "", ""
+            return "❌ Transcript not available for this video.", "", "", ""
 
         summary = saved_transcripts[video_id]
         translation = GoogleTranslator(source='auto', target='es').translate(summary)
         video_embed_link = f"https://www.youtube.com/embed/{video_id}"
-        return summary, translation, video_embed_link
+        steps = simulate_video_steps(summary)
+
+        return summary, translation, video_embed_link, steps
     except Exception as e:
-        return f"❌ Error: {str(e)}", "", ""
+        return f"❌ Error: {str(e)}", "", "", ""
+
+# ✅ Agent 2: Simulated Key Moments / Scene Steps
+def simulate_video_steps(summary):
+    if not summary or summary.startswith("❌"):
+        return "No steps available."
 
-# ✅ Master Agent Workflow with Embedded YouTube Preview
+    return "\n".join([
+        "📌 Step 1: Introduction of topic",
+        "🔍 Step 2: Main discussion about agents",
+        "✅ Step 3: Final conclusion & translation"
+    ])
+
+# ✅ Master Agent Workflow
 def run_agents(url):
-    summary, translation, embed = summarize_youtube(url)
+    summary, translation, embed, steps_text = summarize_youtube(url)
+    
     if embed:
         video_html = f'''
         <div style="position:relative;padding-bottom:56.25%;height:0;overflow:hidden;">
@@ -41,19 +54,22 @@ def run_agents(url):
         '''
     else:
         video_html = ""
-    return summary, translation, video_html
 
-# ✅ Gradio UI
+    return summary, translation, video_html, steps_text
+
+# ✅ Gradio App
 with gr.Blocks() as demo:
-    gr.Markdown("## 🎥 AI Agents: YouTube Summary + Spanish Translator")
-    gr.Markdown("🔗 Enter a YouTube video URL to simulate AI agent collaboration.")
+    gr.Markdown("## 🤖 Lead With AI Agents: YouTube Video Analyzer")
+    gr.Markdown("Paste any **YouTube video link** to get AI-powered insights.")
 
-    input_url = gr.Textbox(label="Paste YouTube Link")
+    input_url = gr.Textbox(label="🔗 Paste YouTube Link")
     summary_output = gr.Textbox(label="🧠 English Summary")
     translation_output = gr.Textbox(label="🌍 Spanish Translation")
     video_output = gr.HTML()
+    steps_output = gr.Textbox(label="🎬 AI Scene Steps / Key Moments")
 
-    run_btn = gr.Button("🔍 Run Agents")
-    run_btn.click(fn=run_agents, inputs=input_url, outputs=[summary_output, translation_output, video_output])
+    run_btn = gr.Button("▶️ Run Agents")
+    run_btn.click(fn=run_agents, inputs=input_url,
+                  outputs=[summary_output, translation_output, video_output, steps_output])
 
-demo.launch()
+demo.launch()