Update app.py

app.py

@@ -1,21 +1,20 @@
-import streamlit as st 
+import streamlit as st
 from phi.agent import Agent
 from phi.model.google import Gemini
 from phi.tools.duckduckgo import DuckDuckGo
-from google.generativeai import upload_file,get_file
+from phi.tools.youtube_tools import YouTubeTools  # Import YouTubeTools for YouTube summarization
+from google.generativeai import upload_file, get_file
 import google.generativeai as genai
 
 import time
 from pathlib import Path
-
 import tempfile
-
 from dotenv import load_dotenv
 load_dotenv()
 
 import os
 
-API_KEY=os.getenv("GOOGLE_API_KEY")
+API_KEY = os.getenv("GOOGLE_API_KEY")
 if API_KEY:
     genai.configure(api_key=API_KEY)
 
@@ -29,24 +28,44 @@ st.set_page_config(
 st.title("Phidata Video AI Summarizer Agent 🎥🎤🖬")
 st.header("Powered by Gemini 2.0 Flash Exp")
 
-
+# Initialize the agent
 @st.cache_resource
 def initialize_agent():
     return Agent(
         name="Video AI Summarizer",
         model=Gemini(id="gemini-2.0-flash-exp"),
-        tools=[DuckDuckGo()],
+        tools=[DuckDuckGo(), YouTubeTools()],  # Add YouTubeTools here for YouTube video summarization
         markdown=True,
     )
 
-## Initialize the agent
-multimodal_Agent=initialize_agent()
+multimodal_Agent = initialize_agent()
+
+# Input field for user query
+user_query = st.text_area(
+    "What insights are you seeking from the video?",
+    placeholder="Ask anything about the video content. The AI agent will analyze and gather additional context if needed.",
+    help="Provide specific questions or insights you want from the video."
+)
 
-# File uploader
+# Check for either video file upload or YouTube URL
 video_file = st.file_uploader(
     "Upload a video file", type=['mp4', 'mov', 'avi'], help="Upload a video for AI analysis"
 )
 
+youtube_url = st.text_input("Or, paste a YouTube URL to summarize:")
+
+# Unified analysis prompt
+def generate_analysis_prompt(user_query, video_source):
+    return (
+        f"""
+        Analyze the {video_source} for content and context.
+        Respond to the following query using video insights and supplementary web research:
+        {user_query}
+        Provide a detailed, user-friendly, and actionable response.
+        """
+    )
+
+# Process uploaded video
 if video_file:
     with tempfile.NamedTemporaryFile(delete=False, suffix='.mp4') as temp_video:
         temp_video.write(video_file.read())
@@ -54,13 +73,7 @@ if video_file:
 
     st.video(video_path, format="video/mp4", start_time=0)
 
-    user_query = st.text_area(
-        "What insights are you seeking from the video?",
-        placeholder="Ask anything about the video content. The AI agent will analyze and gather additional context if needed.",
-        help="Provide specific questions or insights you want from the video."
-    )
-
-    if st.button("🔍 Analyze Video", key="analyze_video_button"):
+    if st.button("🔍 Analyze Uploaded Video", key="analyze_uploaded_video_button"):
         if not user_query:
             st.warning("Please enter a question or insight to analyze the video.")
         else:
@@ -72,16 +85,8 @@ if video_file:
                         time.sleep(1)
                         processed_video = get_file(processed_video.name)
 
-                    # Prompt generation for analysis
-                    analysis_prompt = (
-                        f"""
-                        Analyze the uploaded video for content and context.
-                        Respond to the following query using video insights and supplementary web research:
-                        {user_query}
-
-                        Provide a detailed, user-friendly, and actionable response.
-                        """
-                    )
+                    # Use the unified prompt for analysis (Uploaded video)
+                    analysis_prompt = generate_analysis_prompt(user_query, "uploaded video")
 
                     # AI agent processing
                     response = multimodal_Agent.run(analysis_prompt, videos=[processed_video])
@@ -95,8 +100,30 @@ if video_file:
             finally:
                 # Clean up temporary video file
                 Path(video_path).unlink(missing_ok=True)
+
+# Process YouTube URL
+elif youtube_url:
+    if st.button("🔍 Summarize YouTube Video", key="summarize_youtube_button"):
+        if not user_query:
+            st.warning("Please enter a question or insight to analyze the YouTube video.")
+        else:
+            try:
+                with st.spinner("Fetching video captions and summarizing..."):
+                    # Use the unified prompt for summarization (YouTube URL)
+                    analysis_prompt = generate_analysis_prompt(user_query, "YouTube URL")
+
+                    # Summarize the YouTube video using the agent
+                    response = multimodal_Agent.get_response(f"{analysis_prompt} {youtube_url}")
+                    
+                    # Display the result
+                    st.subheader("Video Summary")
+                    st.markdown(response["content"], unsafe_allow_html=True)
+
+            except Exception as error:
+                st.error(f"An error occurred: {error}")
+
 else:
-    st.info("Upload a video file to begin analysis.")
+    st.info("Upload a video file or paste a YouTube URL to begin analysis.")
 
 # Customize text area height
 st.markdown(
@@ -109,5 +136,3 @@ st.markdown(
     """,
     unsafe_allow_html=True
 )
-
-