Update app.py

app.py

@@ -7,60 +7,35 @@ from tools.final_answer import FinalAnswerTool
 
 from Gradio_UI import GradioUI
 
+from transformers import pipeline
+from typing import Optional
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
+
+
 @tool
-def weather_check(city: str) -> str:
-    """获取指定城市的实时天气并生成艺术化展示
+def text_translator(
+    text: str, 
+    target_lang: Optional[str] = "auto"
+) -> str:
+    """智能中英互译工具
     Args:
-        city: 城市名称（支持中英文，如'北京'或'New York'）
+        text: 待翻译文本（自动检测源语言）
+        target_lang: 目标语言代码（zh/en/ja等，默认自动判断）
     """
-    # 天气图标映射库
-    weather_art = {
-        "晴": r'''
-         \   /     ☀️
-       .-"""-. 
-     _/_______\_
-    (___________)
-        ''',
-        "雨": r'''
-       ´¯`•.¸¸.•´¯`
-      (´•̥̥̥o•̥̥̥`).•´¯`•.☔️
-       `·.¸(  •̥̥̥o•̥̥̥  )
-        ''',
-        "云": r'''
-         .-~~~-.
-        /  ☁️   \
-       ;         ;
-       |         |
-        \       /
-         `~~~~~`
-        '''
-    }
-    
     try:
-        response = requests.get(
-            f"https://api.open-meteo.com/v1/forecast?city={city}&current_weather=true"
-        )
-        data = response.json()
-        condition = data["current"]["weathercode"]
+        is_chinese = any('\u4e00' <= char <= '\u9fff' for char in text)
+        src_lang = "zh" if is_chinese else "en"
+        tgt_lang = "en" if is_chinese else "zh" if target_lang == "auto" else target_lang
         
-        # 转换天气代码为文字描述
-        weather_map = {
-            0: "晴", 1: "晴", 2: "云", 3: "云",
-            45: "雾", 48: "雾", 51: "雨", 61: "雨"
-        }
-        status = weather_map.get(condition, "未知")
-        
-        # 生成艺术报告
-        art = weather_art.get(status, r'''
-        ╱╲╱╲
-        ◖_◗◖_◗
-        ‾‾‾‾‾‾
-        ''')
-        return f"{city}当前天气：{status}\n{art}"
+        model_name = "Helsinki-NLP/opus-mt-zh-en" if src_lang == "zh" else "Helsinki-NLP/opus-mt-en-zh"
+        translator = pipeline("translation", model=model_name)
+
+        result = translator(text, max_length=512)[0]['translation_text']
         
+        return f"翻译结果 ({src_lang}→{tgt_lang}):\n{result}"
+    
     except Exception as e:
-        return f"天气查询失败：{str(e)}"
+        return f"翻译失败：{str(e)}"
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
@@ -99,7 +74,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, weather_check], ## add your tools here (don't remove final answer)
+    tools=[final_answer, text_translator], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,