fix : files cleaned
This commit is contained in:
parent
a25e103189
commit
f41c3582b9
4 changed files with 5 additions and 45 deletions
Binary file not shown.
Binary file not shown.
|
|
@ -2,8 +2,6 @@ from agno.agent import Agent
|
||||||
from agno.team.team import Team
|
from agno.team.team import Team
|
||||||
from agno.models.google import Gemini
|
from agno.models.google import Gemini
|
||||||
from agno.tools.duckduckgo import DuckDuckGoTools
|
from agno.tools.duckduckgo import DuckDuckGoTools
|
||||||
from PIL import Image
|
|
||||||
import easyocr
|
|
||||||
import os
|
import os
|
||||||
|
|
||||||
# --- Therapist Agent ---
|
# --- Therapist Agent ---
|
||||||
|
|
@ -22,7 +20,7 @@ def create_therapist_agent():
|
||||||
api_key=os.environ.get("GEMINI_API_KEY")
|
api_key=os.environ.get("GEMINI_API_KEY")
|
||||||
),
|
),
|
||||||
tools=[DuckDuckGoTools()],
|
tools=[DuckDuckGoTools()],
|
||||||
add_datetime_to_instructions=True, # Add timestamp context
|
add_datetime_to_instructions=True,
|
||||||
show_tool_calls=True,
|
show_tool_calls=True,
|
||||||
markdown=True
|
markdown=True
|
||||||
)
|
)
|
||||||
|
|
@ -43,7 +41,7 @@ def create_closure_agent():
|
||||||
api_key=os.environ.get("GEMINI_API_KEY")
|
api_key=os.environ.get("GEMINI_API_KEY")
|
||||||
),
|
),
|
||||||
tools=[DuckDuckGoTools()],
|
tools=[DuckDuckGoTools()],
|
||||||
add_datetime_to_instructions=True, # Add timestamp context
|
add_datetime_to_instructions=True,
|
||||||
show_tool_calls=True,
|
show_tool_calls=True,
|
||||||
markdown=True
|
markdown=True
|
||||||
)
|
)
|
||||||
|
|
@ -85,7 +83,6 @@ def create_honesty_agent():
|
||||||
"Include reasons why the user should move on, based on the situation.",
|
"Include reasons why the user should move on, based on the situation.",
|
||||||
"Format the output clearly with a header: **Brutal Honesty**",
|
"Format the output clearly with a header: **Brutal Honesty**",
|
||||||
],
|
],
|
||||||
# model=Gemini(id="gemini-2.0-flash"),
|
|
||||||
model=Gemini(
|
model=Gemini(
|
||||||
id="gemini-2.0-flash",
|
id="gemini-2.0-flash",
|
||||||
api_key=os.environ.get("GEMINI_API_KEY")
|
api_key=os.environ.get("GEMINI_API_KEY")
|
||||||
|
|
@ -96,23 +93,11 @@ def create_honesty_agent():
|
||||||
markdown=True
|
markdown=True
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
# # --- Texts Analyzer Agent (Updated with easyocr) ---
|
|
||||||
# def extract_text_with_easyocr(image):
|
|
||||||
# """Extract text from image using EasyOCR."""
|
|
||||||
# reader = easyocr.Reader(['en']) # Specify language
|
|
||||||
# results = reader.readtext(image)
|
|
||||||
|
|
||||||
# # Extract and combine text
|
|
||||||
# extracted_text = " ".join([result[1] for result in results])
|
|
||||||
# return extracted_text
|
|
||||||
|
|
||||||
# --- Team Leader (Breakup Recovery Team) ---
|
# --- Team Leader (Breakup Recovery Team) ---
|
||||||
def create_breakup_team():
|
def create_breakup_team():
|
||||||
return Team(
|
return Team(
|
||||||
name="Breakup Recovery Team",
|
name="Breakup Recovery Team",
|
||||||
mode="coordinate", # Team execution mode: coordinate or parallel
|
mode="coordinate", # Team execution mode: coordinate or parallel
|
||||||
# model=Gemini(id="gemini-2.0-flash"),
|
|
||||||
model=Gemini(
|
model=Gemini(
|
||||||
id="gemini-2.0-flash",
|
id="gemini-2.0-flash",
|
||||||
api_key=os.environ.get("GEMINI_API_KEY")
|
api_key=os.environ.get("GEMINI_API_KEY")
|
||||||
|
|
@ -120,9 +105,8 @@ def create_breakup_team():
|
||||||
members=[
|
members=[
|
||||||
create_therapist_agent(),
|
create_therapist_agent(),
|
||||||
create_closure_agent(),
|
create_closure_agent(),
|
||||||
create_routine_agent(), # Added Routine Planner Agent
|
create_routine_agent(),
|
||||||
create_honesty_agent(), # Added Brutal Honesty Agent
|
create_honesty_agent(),
|
||||||
# extract_text_with_easyocr()
|
|
||||||
],
|
],
|
||||||
description="You are a team helping someone recover from a breakup.",
|
description="You are a team helping someone recover from a breakup.",
|
||||||
instructions=[
|
instructions=[
|
||||||
|
|
|
||||||
|
|
@ -3,7 +3,6 @@ from agno.media import Image
|
||||||
from agno.models.google import Gemini
|
from agno.models.google import Gemini
|
||||||
from agno.tools.duckduckgo import DuckDuckGoTools
|
from agno.tools.duckduckgo import DuckDuckGoTools
|
||||||
from PIL import Image as PILImage
|
from PIL import Image as PILImage
|
||||||
import base64
|
|
||||||
from io import BytesIO
|
from io import BytesIO
|
||||||
import os
|
import os
|
||||||
|
|
||||||
|
|
@ -19,13 +18,6 @@ def load_image(image_data):
|
||||||
|
|
||||||
return img_bytes
|
return img_bytes
|
||||||
|
|
||||||
# # Check for Google API key
|
|
||||||
# if "GOOGLE_API_KEY" not in os.environ:
|
|
||||||
# raise EnvironmentError(
|
|
||||||
# "Please set your GOOGLE_API_KEY environment variable. "
|
|
||||||
# "You can get one from https://makersuite.google.com/app/apikey"
|
|
||||||
# )
|
|
||||||
|
|
||||||
def create_chat_analysis_agent():
|
def create_chat_analysis_agent():
|
||||||
"""Creates a specialized agent for analyzing chat patterns"""
|
"""Creates a specialized agent for analyzing chat patterns"""
|
||||||
return Agent(
|
return Agent(
|
||||||
|
|
@ -68,19 +60,3 @@ def analyze_chat_screenshot(image_data):
|
||||||
|
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
return f"Error analyzing chat screenshot: {str(e)}"
|
return f"Error analyzing chat screenshot: {str(e)}"
|
||||||
|
|
||||||
# Example local image path - replace with your image path
|
|
||||||
image_path = "breakup.jpg"
|
|
||||||
|
|
||||||
try:
|
|
||||||
# Load and process the image
|
|
||||||
image_data = load_image(image_path)
|
|
||||||
|
|
||||||
# Create the response using the agent
|
|
||||||
agent.print_response(
|
|
||||||
"Tell me about this image and give me the latest news about it.",
|
|
||||||
images=[Image(content=image_data)],
|
|
||||||
stream=True,
|
|
||||||
)
|
|
||||||
except Exception as e:
|
|
||||||
print(f"Error processing image: {e}")
|
|
||||||
Loading…
Reference in a new issue