Update app.py

app.py

@@ -1,11 +1,20 @@
-from smolagents import CodeAgent, DuckDuckGoSearchTool
+from smolagents import CodeAgent, DuckDuckGoSearchTool, tool
 import datetime
 import requests
 import pytz
 import yaml
-from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 
+# Создаем свой собственный инструмент для финального ответа
+@tool
+def final_answer(answer: str) -> str:
+    """Provide the final answer to the user. Use this when you have the answer to the user's question.
+    
+    Args:
+        answer: The final answer to present to the user
+    """
+    return answer
+
 # Инструменты (tools), используемые агентом:
 # - web_search: поиск в интернете через DuckDuckGo
 # - get_weather: получение текущей погоды для города
@@ -117,20 +126,36 @@ def suggest_weather_sources(city: str) -> str:
     
     return f"Надежные источники погоды для {city}:\n" + "\n".join(sources)
 
-final_answer = FinalAnswerTool()
-
 # Используем локальную модель вместо HfApiModel
-from transformers import AutoModelForCausalLM, AutoTokenizer
-from smolagents import TransformersModel
-
-model = TransformersModel(
-    model_id='Qwen/Qwen2.5-Coder-1.5B',  # Используем меньшую модель для совместимости
-    max_tokens=2096,
-    temperature=0.5
-)
+try:
+    from transformers import AutoModelForCausalLM, AutoTokenizer
+    from smolagents import TransformersModel
+    
+    model = TransformersModel(
+        model_id='Qwen/Qwen2.5-Coder-1.5B',  # Используем меньшую модель для совместимости
+        max_tokens=2096,
+        temperature=0.5
+    )
+except ImportError:
+    # Если transformers не установлен, используем простую модель
+    from smolagents import LiteLLMModel
+    model = LiteLLMModel(
+        model_id='gpt-3.5-turbo',
+        max_tokens=2096,
+        temperature=0.5
+    )
 
-with open("prompts.yaml", 'r') as stream:
-    prompt_templates = yaml.safe_load(stream)
+# Загружаем промпты, если файл существует, иначе создаем базовый шаблон
+try:
+    with open("prompts.yaml", 'r') as stream:
+        prompt_templates = yaml.safe_load(stream)
+except FileNotFoundError:
+    prompt_templates = {
+        "final_answer": {
+            "pre_messages": "Based on my research: ",
+            "post_messages": ""
+        }
+    }
 
 if "final_answer" not in prompt_templates:
     prompt_templates["final_answer"] = {