mirror of
https://github.com/MervinPraison/PraisonAI.git
synced 2026-11-07 12:47:55 -10:00
102 lines
2.8 KiB
Python
102 lines
2.8 KiB
Python
"""
|
|
Basic Camera Integration Example
|
|
|
|
This example demonstrates how to capture a single frame from a camera
|
|
and analyze it using a PraisonAI vision agent.
|
|
|
|
Requirements:
|
|
pip install praisonaiagents opencv-python
|
|
|
|
Usage:
|
|
python camera-basic.py
|
|
|
|
Environment Variables:
|
|
OPENAI_API_KEY=your_openai_api_key
|
|
"""
|
|
|
|
import cv2
|
|
import os
|
|
from praisonaiagents import Agent, Task, Agents
|
|
|
|
|
|
def capture_and_analyze():
|
|
"""Capture a single frame from camera and analyze it"""
|
|
|
|
# Create vision agent
|
|
vision_agent = Agent(
|
|
name="CameraAnalyst",
|
|
role="Camera Feed Analyzer",
|
|
goal="Analyze camera captures in real-time",
|
|
backstory="Expert in real-time visual analysis with computer vision skills",
|
|
llm="gpt-5-nano"
|
|
)
|
|
|
|
# Initialize camera (0 for default camera)
|
|
cap = cv2.VideoCapture(0)
|
|
|
|
if not cap.isOpened():
|
|
print("Error: Could not open camera")
|
|
return None
|
|
|
|
try:
|
|
# Capture frame
|
|
ret, frame = cap.read()
|
|
|
|
if ret:
|
|
# Save frame temporarily
|
|
image_path = "temp_camera_capture.jpg"
|
|
cv2.imwrite(image_path, frame)
|
|
print(f"Frame captured and saved as {image_path}")
|
|
|
|
# Create analysis task
|
|
task = Task(
|
|
name="analyze_camera_feed",
|
|
description="Analyze what you see in this camera capture. Identify objects, people, and activities.",
|
|
expected_output="Detailed analysis of the camera content including objects, people, activities, and overall scene description",
|
|
agent=vision_agent,
|
|
images=[image_path]
|
|
)
|
|
|
|
# Run analysis
|
|
agents = Agents(
|
|
agents=[vision_agent],
|
|
tasks=[task],
|
|
process="sequential",
|
|
verbose=1
|
|
)
|
|
|
|
result = agents.start()
|
|
|
|
# Clean up temporary file
|
|
if os.path.exists(image_path):
|
|
os.remove(image_path)
|
|
|
|
return result
|
|
|
|
else:
|
|
print("Error: Could not capture frame")
|
|
return None
|
|
|
|
finally:
|
|
cap.release()
|
|
cv2.destroyAllWindows()
|
|
|
|
|
|
def main():
|
|
"""Main function"""
|
|
print("📸 Starting basic camera analysis...")
|
|
print("Make sure your camera is connected and accessible")
|
|
|
|
result = capture_and_analyze()
|
|
|
|
if result:
|
|
print("\n🔍 Analysis Results:")
|
|
for task_id, task_result in result["task_results"].items():
|
|
print(f"\nTask {task_id}:")
|
|
print(task_result.raw)
|
|
else:
|
|
print("❌ Failed to capture and analyze camera feed")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main() |