Update app.py

app.py

@@ -1,38 +1,10 @@
-import requests
+from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
+import datetime
 import pytz
 import yaml
-import os
 from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
-from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
-
-# ----- تصحيح: إضافة الاستيراد الناقص لـ datetime -----
-import datetime
-
-# ----- تصحيح: إغلاق الأقواس الناقصة في أداة التلخيص -----
-@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}
-        }  # <--- إضافة القوس المفقود هنا
-        response = requests.post(API_URL, headers=headers, json=data)
-        response.raise_for_status()
-        summary = response.json()[0]['summary_text']
-        return f"الملخص: {summary}"
-    except Exception as e:
-        return f"خطأ في التلخيص: {str(e)}"
 
-# ----- تصحيح: إضافة النقطة في نهاية الجملة الأولى من الـ Docstring -----
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """أداة لجلب الوقت الحالي في منطقة زمنية محددة.
@@ -47,7 +19,8 @@ def get_current_time_in_timezone(timezone: str) -> str:
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
-# ----- تصحيح: إضافة تعريف النموذج (model) -----
+final_answer = FinalAnswerTool()
+
 model = HfApiModel(
     max_tokens=2096,
     temperature=0.5,
@@ -55,17 +28,17 @@ model = HfApiModel(
     custom_role_conversions=None,
 )
 
-# ----- تصحيح: إضافة الأداة المُستوردة من الـ 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=[
-        FinalAnswerTool(),  # <--- تصحيح: إضافة الأقواس ()
+        final_answer,
         DuckDuckGoSearchTool(),
         image_generation_tool,
-        summarize_text,
         get_current_time_in_timezone
     ],
     max_steps=6,
@@ -74,7 +47,7 @@ agent = CodeAgent(
     planning_interval=None,
     name=None,
     description=None,
-    prompt_templates=yaml.safe_load(open("prompts.yaml"))
+    prompt_templates=prompt_templates
 )
 
 GradioUI(agent).launch()