Update app.py

app.py

@@ -2,14 +2,42 @@ import os
 import requests
 import pandas as pd
 import gradio as gr
-from smolagents import CodeAgent, DuckDuckGoSearchTool, LiteLLMModel
+from huggingface_hub import InferenceClient
+from smolagents import CodeAgent, DuckDuckGoSearchTool
 
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
-MODEL_ID = "huggingface/Qwen/Qwen2.5-Coder-32B-Instruct"
+MODEL_ID = "Qwen/Qwen2.5-Coder-32B-Instruct"
+
+class CustomHFModel:
+    def __init__(self, model_id: str, token: str):
+        self.model_id = model_id
+        self.client = InferenceClient(model=model_id, token=token)
+
+    def __call__(self, messages, **kwargs):
+        try:
+            formatted_messages = []
+            for m in messages:
+                if isinstance(m, dict):
+                    formatted_messages.append(m)
+                else:
+                    role = getattr(m, "role", "user")
+                    content = getattr(m, "content", str(m))
+                    formatted_messages.append({"role": role, "content": str(content)})
+
+            response = self.client.chat_completion(
+                messages=formatted_messages,
+                max_tokens=200,
+                temperature=0.0
+            )
+            return response.choices[0].message
+        except Exception as e:
+            print(f"Model error: {e}")
+            class Choice:
+                content = "0"
+            return Choice()
 
 def create_agent(token: str):
-    # LiteLLMModel принимает токен HF и работает со всеми моделями Hub
-    model = LiteLLMModel(model_id=MODEL_ID, api_key=token)
+    model = CustomHFModel(model_id=MODEL_ID, token=token)
     search_tool = DuckDuckGoSearchTool()
     
     agent = CodeAgent(
@@ -20,7 +48,6 @@ def create_agent(token: str):
     return agent
 
 def solve_question(agent, question: str) -> str:
-    # Автоматический переворот для вопросов-инверсий
     if question.strip().startswith(".") or "ecnetnes" in question:
         question = question[::-1]
 
@@ -39,7 +66,7 @@ def solve_question(agent, question: str) -> str:
             ans = ans.split("\n")[-1].strip()
         return ans.strip(" .\"'") if ans else "0"
     except Exception as e:
-        print(f"Agent error on question '{question[:20]}...': {e}")
+        print(f"Agent error: {e}")
         return "0"
 
 def run_and_submit_all(profile: gr.OAuthProfile | None):
@@ -60,7 +87,6 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
 
     agent_code = f"https://huggingface.co/spaces/{space_id}/tree/main" if space_id else "https://huggingface.co"
 
-    # Fetch Questions
     try:
         res = requests.get(f"{DEFAULT_API_URL}/questions", timeout=15)
         res.raise_for_status()
@@ -83,7 +109,6 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
         answers_payload.append({"task_id": task_id, "submitted_answer": ans})
         results_log.append({"Task ID": task_id, "Question": q, "Submitted Answer": ans})
 
-    # Submit
     try:
         sub_res = requests.post(
             f"{DEFAULT_API_URL}/submit",