wehub-resource-sync
93 lines
2.6 KiB
Python
93 lines
2.6 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
Demo script for Video RAG with Gemini
|
|
This script demonstrates how to use the Gemini API for video understanding
|
|
"""
|
|
|
|
import google.generativeai as genai
|
|
import os
|
|
from dotenv import load_dotenv
|
|
|
|
load_dotenv()
|
|
|
|
def demo_video_chat():
|
|
"""Simple demo of video chat functionality"""
|
|
|
|
# Configure API
|
|
api_key = os.getenv("GEMINI_API_KEY")
|
|
if not api_key:
|
|
print("❌ Please set GEMINI_API_KEY in your .env file")
|
|
return
|
|
|
|
genai.configure(api_key=api_key)
|
|
|
|
# Initialize model
|
|
model = genai.GenerativeModel('gemini-1.5-pro')
|
|
|
|
print("🎬 Video RAG Demo")
|
|
print("================")
|
|
print("This demo shows how to upload a video and chat with it using Gemini API")
|
|
print()
|
|
|
|
# Get video file path
|
|
video_path = input("Enter path to your video file: ").strip()
|
|
|
|
if not os.path.exists(video_path):
|
|
print(f"❌ File not found: {video_path}")
|
|
return
|
|
|
|
try:
|
|
print("📤 Uploading video...")
|
|
video_file = genai.upload_file(path=video_path, display_name="demo_video")
|
|
|
|
print("⏳ Processing video...")
|
|
while video_file.state.name == "PROCESSING":
|
|
print(" Still processing...")
|
|
import time
|
|
time.sleep(5)
|
|
video_file = genai.get_file(video_file.name)
|
|
|
|
if video_file.state.name == "FAILED":
|
|
print("❌ Video processing failed")
|
|
return
|
|
|
|
print("✅ Video processed successfully!")
|
|
print()
|
|
|
|
# Interactive chat loop
|
|
print("💬 You can now chat with your video. Type 'quit' to exit.")
|
|
print(" Example questions:")
|
|
print(" - What is happening in this video?")
|
|
print(" - Describe the main events")
|
|
print(" - Who are the people in this video?")
|
|
print()
|
|
|
|
while True:
|
|
user_input = input("You: ").strip()
|
|
|
|
if user_input.lower() in ['quit', 'exit', 'q']:
|
|
break
|
|
|
|
if not user_input:
|
|
continue
|
|
|
|
try:
|
|
print("🤖 Thinking...")
|
|
response = model.generate_content([video_file, user_input])
|
|
print(f"AI: {response.text}")
|
|
print()
|
|
except Exception as e:
|
|
print(f"❌ Error generating response: {e}")
|
|
|
|
# Cleanup
|
|
print("🧹 Cleaning up...")
|
|
genai.delete_file(video_file.name)
|
|
print("✅ Demo completed!")
|
|
|
|
except Exception as e:
|
|
print(f"❌ Error: {e}")
|
|
|
|
if __name__ == "__main__":
|
|
demo_video_chat()
|
|
|