You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
58 lines
1.5 KiB
58 lines
1.5 KiB
import os
|
|
from dotenv import load_dotenv
|
|
from swarms.models import Gemini
|
|
from swarms.structs import Agent
|
|
|
|
# Load environment variables
|
|
load_dotenv()
|
|
|
|
# Get the API key from the environment
|
|
api_key = os.environ.get("GOOGLE_API_KEY")
|
|
if not api_key:
|
|
raise ValueError("Gemini API key not found. Please set it in your .env file")
|
|
|
|
# Initialize the first agent (Gemini Pro Vision) for image analysis
|
|
agent_analysis = Agent(
|
|
llm=Gemini(
|
|
temperature=0.8,
|
|
model_name="gemini-pro-vision",
|
|
gemini_api_key=api_key
|
|
),
|
|
max_loops=1,
|
|
autosave=True,
|
|
dashboard=True,
|
|
)
|
|
|
|
# Initialize the second agent (Gemini Pro) for text-based tasks
|
|
agent_instruction = Agent(
|
|
llm=Gemini(
|
|
temperature=0.8,
|
|
model_name="gemini-pro",
|
|
gemini_api_key=api_key
|
|
),
|
|
max_loops=1,
|
|
autosave=True,
|
|
dashboard=True,
|
|
)
|
|
|
|
# Task for the first agent: Image Analysis
|
|
task_analysis = """
|
|
Analyze this logo and summarize its content, then give a structured list of suggested improvements.
|
|
"""
|
|
img = 'swarmslogobanner.png' # Replace with your image file name
|
|
|
|
# Run the first agent on the task
|
|
analysis_results = agent_analysis.run(task=task_analysis, img=img)
|
|
|
|
# Task for the second agent: Instructions for a better logo
|
|
task_instruction = f"""
|
|
Based on the analysis and suggestions for the logo:
|
|
{analysis_results}
|
|
|
|
Generate detailed instructions for creating an improved version of this logo.
|
|
"""
|
|
|
|
# Run the second agent on the task
|
|
instruction_results = agent_instruction.run(task=task_instruction)
|
|
|
|
... |