Update app.py

app.py

@@ -1,80 +1,40 @@
-from smolagents import CodeAgent, HfApiModel, load_tool
+from smolagents import CodeAgent, load_tool
 import yaml
 from smolagents import tool
 from duckduckgo_search import DDGS
+import requests
 
 from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 
-@tool
-def DuckDuckGoSearchTool(query: str) -> str:
-    """
-    Инструмент для поиска информации в интернете с помощью DuckDuckGo.
-    Args:
-        query: Поисковый запрос.
-    """
-    with DDGS() as ddgs:
-        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
+#... (your existing code)...
 
-final_answer = FinalAnswerTool()
+class DeepSeekModel:
+    def __init__(self, api_key):
+        self.api_key = api_key
+        self.url = "https://api.deepseek.ai/v1/chat/completions"  # Replace with the correct API endpoint
 
-# Define both model IDs
-primary_model_id = 'Qwen/Qwen2.5-Coder-32B-Instruct'
-fallback_model_id = 'https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud'
+    def generate(self, messages):
+        headers = {
+            "Authorization": f"Bearer {self.api_key}",
+            "Content-Type": "application/json"
+        }
+        data = {
+            "messages": messages,
+            # Add any other required parameters for the DeepSeek API
+        }
+        response = requests.post(self.url, headers=headers, json=data)
+        response.raise_for_status()  # Raise an exception for bad status codes
+        return response.json()  # Assuming the response is in JSON format
 
-# Try to initialize the primary model
-try:
-    model = HfApiModel(
-        max_tokens=2096,
-        temperature=0.5,
-        model_id=primary_model_id,
-        custom_role_conversions=None,
-    )
-    print(f"Using primary model: {primary_model_id}")
-except Exception as e:
-    print(f"Error initializing primary model: {e}")
-    print(f"Falling back to secondary model: {fallback_model_id}")
-    # If the primary model fails, use the fallback model
-    model = HfApiModel(
-        max_tokens=2096,
-        temperature=0.5,
-        model_id=fallback_model_id,
-        custom_role_conversions=None,
-    )
+# Initialize the DeepSeek model
+model = DeepSeekModel(api_key="sk-eea44c5fc6be4e289cf0c1ae5bd91b58")
 
-image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
-
-# Загрузка prompt_templates из prompts.yaml
-with open("prompts.yaml", 'r') as stream:
-    try:
-        prompt_templates = yaml.safe_load(stream)
-    except yaml.YAMLError as exc:
-        print(exc)
-
-# Проверка, что prompt_templates имеет правильный формат
-if isinstance(prompt_templates, dict) and 'system_prompt' in prompt_templates:
-    # Если все ок, используем загруженный шаблон
-    system_prompt = prompt_templates['system_prompt']
-
-    prompt_templates = {'system_prompt': system_prompt}
+#... (rest of your code)...
 
 agent = CodeAgent(
-    model=model,
-    tools=[DuckDuckGoSearchTool, final_answer, image_generation_tool],
-    max_steps=6,
-    verbosity_level=1,
-    grammar=None,
-    planning_interval=None,
-    name=None,
-    description=None,
-    prompt_templates=prompt_templates
+    model=model,  # Use the DeepSeekModel instance
+    #... (rest of your agent parameters)...
 )
 
-GradioUI(agent).launch()
+#... (rest of your code)...