Spaces:
Running
Running
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 | |
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") | |
if API_KEY: | |
genai.configure(api_key=API_KEY) | |
# Page configuration | |
st.set_page_config( | |
page_title="Multimodal AI Agent Video Notes Writer", | |
page_icon="π₯", | |
layout="wide" | |
) | |
st.title("Phidata Video AI Notes Writer π₯π€π¬") | |
st.header("Powered by Gemini 2.0 Flash Exp") | |
def initialize_agent(): | |
return Agent( | |
name="Video AI Notes-Writer", | |
model=Gemini(id="gemini-2.0-flash-exp"), | |
tools=[DuckDuckGo()], | |
markdown=True, | |
) | |
# Initialize the agent | |
multimodal_Agent = initialize_agent() | |
# File uploader | |
video_file = st.file_uploader( | |
"Upload a video file", type=['mp4'], help="Upload a video for AI analysis" | |
) | |
if video_file: | |
with tempfile.NamedTemporaryFile(delete=False, suffix='.mp4') as temp_video: | |
temp_video.write(video_file.read()) | |
video_path = temp_video.name | |
st.video(video_path, format="video/mp4", start_time=0) | |
user_query = st.text_area( | |
"What insights are you seeking from the video?", | |
value="Generate detailed student-style notes from this video in paragraph format, with thorough explanations of concepts and ideas.", | |
help="Provide specific questions or insights you want from the video." | |
) | |
if st.button("π Analyze Video", key="analyze_video_button"): | |
if not user_query: | |
st.warning("Please enter a question or insight to analyze the video.") | |
else: | |
try: | |
with st.spinner("Processing video and creating detailed notes..."): | |
# Upload and process video file | |
processed_video = upload_file(video_path) | |
while processed_video.state.name == "PROCESSING": | |
time.sleep(1) | |
processed_video = get_file(processed_video.name) | |
# Enhanced prompt for detailed student-style notes with improved structure | |
analysis_prompt = ( | |
f""" | |
Analyze the uploaded video thoroughly and generate EXTREMELY DETAILED STUDENT NOTES in full paragraph format. | |
Your notes should: | |
- Begin with a comprehensive introduction that outlines the topic and objectives. | |
- Divide the content into clearly labelled sections with headings and subheadings (e.g., "Introduction", "Main Concepts", "Detailed Explanations", and "Conclusion"). | |
- Provide detailed explanations of key concepts, ensuring each section includes a narrative summary of its key takeaways. | |
- Include smooth transitions between sections and paragraphs for a cohesive flow. | |
- Reference and explain any visual aids, diagrams, or slides mentioned in the lecture. | |
- Incorporate reflective questions or critical insights to prompt further analysis of the material. | |
- Maintain a student-friendly yet rigorous tone, similar to well-structured textbook content. | |
- Conclude with a summary paragraph that captures the main ideas. | |
- Remember to include every point discussed in the video. Never ever miss a point discussed in a video. | |
- The Notes should be in detail, it should help the user to know whatever content is present in the video. | |
- If there is a mathematical expression you need to write it down | |
-If there is a code then you have to write it down. | |
IMPORTANT: DO NOT use bullet points or lists. Present all information in well-developed and explained paragraphs. | |
Additional context/question from user: {user_query} | |
The final output should resemble detailed, structured textbook content that thoroughly explains all concepts from the video. | |
""" | |
) | |
# AI agent processing | |
response = multimodal_Agent.run(analysis_prompt, videos=[processed_video]) | |
# Display the result | |
st.subheader("π Detailed Notes") | |
st.markdown(response.content) | |
# Simple download option | |
st.download_button( | |
label="Download Notes", | |
data=response.content, | |
file_name="video_notes.md", | |
mime="text/markdown", | |
) | |
except Exception as error: | |
st.error(f"An error occurred during analysis: {error}") | |
finally: | |
# Clean up temporary video file | |
Path(video_path).unlink(missing_ok=True) | |
else: | |
st.info("Upload a video file to begin analysis.") | |