Update app.py

app.py

@@ -4,15 +4,12 @@ from smolagents import tool
 from duckduckgo_search import DDGS
 
 from tools.final_answer import FinalAnswerTool
-
 from Gradio_UI import GradioUI
 
-
 @tool
 def DuckDuckGoSearchTool(query: str) -> str:
     """
     Инструмент для поиска информации в интернете с помощью DuckDuckGo.
-
     Args:
         query: Поисковый запрос.
     """
@@ -20,36 +17,44 @@ def DuckDuckGoSearchTool(query: str) -> str:
         results = [r for r in ddgs.text(query, max_results=5)]  # Ограничиваем до 5 результатов
         if not results:
             return "По вашему запросу ничего не найдено."
-
         formatted_results = "\n\n".join(
             f"**Заголовок:** {r['title']}\n**Ссылка:** {r['href']}\n**Краткое содержание:** {r['body']}"
             for r in results
         )
     return formatted_results
 
-
 final_answer = FinalAnswerTool()
 
 model = HfApiModel(
     max_tokens=2096,
     temperature=0.5,
-    model_id='Qwen/Qwen2.5-Coder-32B-Instruct',  # Возможно, эта модель перегружена
+    model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
     custom_role_conversions=None,
 )
 
 # Import tool from Hub
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
-prompt_templates = {}  
-
+# Загрузка prompt_templates из prompts.yaml
 with open("prompts.yaml", 'r') as stream:
     try:
-        system_prompt = yaml.safe_load(stream)
+        prompt_templates = yaml.safe_load(stream)
     except yaml.YAMLError as exc:
         print(exc)
 
-prompt_templates['system_prompt'] = system_prompt
-
+# Проверка, что prompt_templates имеет правильный формат
+if isinstance(prompt_templates, dict) and 'system_prompt' in prompt_templates:
+    # Если все ок, используем загруженный шаблон
+    system_prompt = prompt_templates['system_prompt']
+else:
+    # Иначе, создаем новый словарь с system_prompt
+    system_prompt = """
+    Ты — Perplexity: профессиональный поисковый ассистент.
+    Твоя задача — предоставлять исчерпывающие и точные ответы на запросы пользователей, основываясь на информации из нескольких источников в интернете.
+    #... (остальной текст вашего промпта)...
+    Начни!
+    """
+    prompt_templates = {'system_prompt': system_prompt}
 
 agent = CodeAgent(
     model=model,
@@ -63,5 +68,4 @@ agent = CodeAgent(
     prompt_templates=prompt_templates
 )
 
-
 GradioUI(agent).launch()