Update app.py

app.py

@@ -2,18 +2,17 @@ import gradio as gr
 from deep_translator import GoogleTranslator
 import re
 
-# ✅ Extract video ID from YouTube URL
+# ✅ Extract video ID from URL
 def extract_video_id(url):
     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)
+# ✅ Agent 1: Simulated Summarizer
 def summarize_youtube(video_url):
     try:
         video_id = extract_video_id(video_url)
 
-        # Simulated transcript summaries for demo
         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 how they collaborate.",
@@ -21,37 +20,34 @@ def summarize_youtube(video_url):
         }
 
         if video_id not in saved_transcripts:
-            return "❌ Transcript not available for this video. Try one of the demo videos listed."
+            return "❌ Transcript not available for this video.", "", ""
 
-        return saved_transcripts[video_id]
+        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
     except Exception as e:
-        return f"❌ Error in Summarizer: {str(e)}"
-
-# ✅ Agent 2: Translator
-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)}"
-
-# ✅ Master Agent workflow
-def agent_workflow(video_url):
-    summary = summarize_youtube(video_url)
-    if summary.startswith("❌"):
-        return summary, ""
-    translation = translate_to_spanish(summary)
-    return summary, translation
-
-# ✅ Gradio UI
-demo = gr.Interface(
-    fn=agent_workflow,
-    inputs=gr.Textbox(label="🎥 Enter YouTube Video Link"),
-    outputs=[
-        gr.Textbox(label="🧠 English Summary"),
-        gr.Textbox(label="🌍 Spanish Translation")
-    ],
-    title="🧠 AI Agents Hackathon – Hugging Face App",
-    description="Agent 1: Summarizer (simulated) • Agent 2: Translator (real)"
-)
+        return f"❌ Error: {str(e)}", "", ""
+
+# ✅ Gradio App
+with gr.Blocks() as demo:
+    gr.Markdown("## 🎥 AI Agents: YouTube Summary + Spanish Translator")
+    gr.Markdown("Enter a YouTube video URL to simulate agent behavior.")
+
+    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()
+
+    def run_agents(url):
+        summary, translation, embed = summarize_youtube(url)
+        if embed:
+            video_html = f'<iframe width="100%" height="315" src="{embed}" frameborder="0" allowfullscreen></iframe>'
+        else:
+            video_html = ""
+        return summary, translation, video_html
+
+    run_btn = gr.Button("🔍 Run Agents")
+    run_btn.click(fn=run_agents, inputs=input_url, outputs=[summary_output, translation_output, video_output])
 
 demo.launch()