AgentNaser's picture
Update app.py
c926290 verified
raw
history blame
1.37 kB
import requests
import pytz
import yaml
import os
from tools.final_answer import FinalAnswerTool
from Gradio_UI import GradioUI
@tool
def summarize_text(text: str, max_length: int = 130) -> str:
"""تلخيص النصوص باستخدام نموذج BART من Hugging Face
Args:
text: النص الأصلي المراد تلخيصه
max_length: الحد الأقصى لطول الملخص (افتراضي: 130 كلمة)
"""
try:
API_URL = "https://api-inference.huggingface.co/models/facebook/bart-large-cnn"
headers = {"Authorization": f"Bearer {os.environ['HF_TOKEN']}"}
data = {
"inputs": text,
"parameters": {"max_length": max_length}
@tool
def get_current_time_in_timezone(timezone: str) -> str:
"""أداة لجلب الوقت الحالي في منطقة زمنية محددة.
Args:
timezone: سلسلة نصية تمثل منطقة زمنية صالحة (مثال: 'America/New_York').
"""
try:
tz = pytz.timezone(timezone)
local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
final_answer,
DuckDuckGoSearchTool(),
image_generation_tool,
summarize_text,
get_current_time_in_timezone # تأكد من إضافتها هنا
],
max_steps=6,
verbosity_level=1,