Notes_Writer / app.py
Manasa1's picture
Update app.py
1834974 verified
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")
@st.cache_resource
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.")