Update app.py

app.py

@@ -1,9 +1,13 @@
-# app.py
 import os
 from smolagents import CodeAgent, HfApiModel, tool
 import datetime
 import pytz
-import gradio as gr
+import yaml
+from Gradio_UI import GradioUI
+from tools.final_answer import FinalAnswerTool
+
+# Выводим токен для отладки (в логах он будет скрыт как секрет)
+print("HF_TOKEN from environment:", "Set" if os.getenv("HF_TOKEN") else "Not set")
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
@@ -26,16 +30,17 @@ def final_answer(answer: str) -> str:
     """
     return answer
 
-model = HfApiModel(
-    max_tokens=100,
-    temperature=0.5,
-    model_id='Qwen/Qwen2.5-Coder-7B-Instruct',
-)
+try:
+    model = HfApiModel(
+        max_tokens=100,
+        temperature=0.5,
+        model_id='Qwen/Qwen2.5-Coder-7B-Instruct',
+    )
+except Exception as e:
+    print(f"Error initializing model: {str(e)}")
 
-prompt_templates = {
-    "default": "Answer: {{question}}",
-    "system_prompt": "You are a helpful assistant. Use tools if necessary to answer the question."
-}
+with open("prompts.yaml", 'r') as stream:
+    prompt_templates = yaml.safe_load(stream)
 
 agent = CodeAgent(
     model=model,
@@ -45,11 +50,4 @@ agent = CodeAgent(
     prompt_templates=prompt_templates,
 )
 
-def process_input(user_input):
-    try:
-        response = agent.run(user_input)
-        return response
-    except Exception as e:
-        return f"Error: {str(e)}"
-
-gr.Interface(fn=process_input, inputs="text", outputs="text").launch()
+GradioUI(agent).launch()