Update app.py

app.py

@@ -3,20 +3,21 @@ import datetime
 import requests
 import pytz
 import yaml
-import os  # <--- أضف هذا السطر
+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']}"}  # <--- تصحيح هنا
+        headers = {"Authorization": f"Bearer {os.environ['HF_TOKEN']}"}
         data = {
             "inputs": text,
             "parameters": {"max_length": max_length}
@@ -30,7 +31,11 @@ def summarize_text(text: str, max_length: int = 130) -> str:
 
 @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")
@@ -57,7 +62,8 @@ agent = CodeAgent(
         final_answer,
         DuckDuckGoSearchTool(),
         image_generation_tool,
-        summarize_text  # <--- الأداة الجديدة هنا
+        summarize_text,
+        get_current_time_in_timezone  # تأكد من إضافتها هنا
     ],
     max_steps=6,
     verbosity_level=1,