File size: 2,905 Bytes
d852706 9b5b26a e00c28e 6aae614 8fe992b 9b5b26a d852706 ac15a8a d852706 9b5b26a d852706 ac15a8a 9b5b26a d852706 ac15a8a 9b5b26a d852706 ac15a8a d852706 ac15a8a d852706 ac15a8a d852706 ac15a8a d852706 9b5b26a 8c01ffb 6aae614 ae7a494 e121372 d852706 13d500a 8c01ffb 9b5b26a 8c01ffb 861422e d852706 8c01ffb 8fe992b b2b28f9 8c01ffb 861422e 8fe992b ac15a8a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 |
from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
import datetime
import requests
import pytz
import yaml
from tools.final_answer import FinalAnswerTool
from Gradio_UI import GradioUI
# A tool that fetches data from an API endpoint
import qrcode
import os
from PIL import Image
import io
import base64
@tool
def generate_qr_code(data: str) -> str:
"""Generates a QR code from the given data and returns it as a base64 string.
Args:
data: The text or URL to encode in the QR code.
Returns:
Base64 encoded QR code image.
"""
img = qrcode.make(data)
buffered = io.BytesIO()
img.save(buffered, format="PNG")
img_str = base64.b64encode(buffered.getvalue()).decode("utf-8")
return img_str
@tool
def generate_image_and_qr(prompt: str) -> dict:
"""Generates an image from a text prompt and returns both the image and QR code as base64 strings.
Args:
prompt: The text prompt for image generation.
Returns:
A dictionary containing base64 encoded image and QR code.
"""
try:
image = image_generation_tool(prompt=prompt)
buffered = io.BytesIO()
image.save(buffered, format="PNG")
image_str = base64.b64encode(buffered.getvalue()).decode("utf-8")
qr_code_str = generate_qr_code(image_str)
return {"image": image_str, "qr_code": qr_code_str}
except Exception as e:
return {"error": str(e)}
@tool
def get_current_time_in_timezone(timezone: str) -> str:
"""A tool that fetches the current local time in a specified timezone.
Args:
timezone: A string representing a valid timezone (e.g., 'America/New_York').
"""
try:
# Create timezone object
tz = pytz.timezone(timezone)
# Get current time in that timezone
local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
return f"The current local time in {timezone} is: {local_time}"
except Exception as e:
return f"Error fetching time for timezone '{timezone}': {str(e)}"
final_answer = FinalAnswerTool()
model = HfApiModel(
max_tokens=2096,
temperature=0.5,
model_id='Qwen/Qwen2.5-Coder-32B-Instruct', # it is possible that this model may be overloaded
custom_role_conversions=None,
)
# Import tool from Hub
image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
with open("prompts.yaml", 'r') as stream:
prompt_templates = yaml.safe_load(stream)
agent = CodeAgent(
model=model,
tools=[final_answer, generate_image_and_qr, generate_qr_code, image_generation_tool], # add your tools here (don't remove final answer)
max_steps=6,
verbosity_level=1,
grammar=None,
planning_interval=None,
name=None,
description=None,
prompt_templates=prompt_templates
)
GradioUI(agent).launch()
|