Update app.py

app.py

@@ -15,6 +15,7 @@ from smolagents import (
 
 from pypdf import PdfReader
 
+# --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 MODEL_ID = "Qwen/Qwen2.5-32B-Instruct"
 
@@ -39,8 +40,10 @@ class BasicAgent:
         def download_file(url: str) -> str:
             """
             Скачивает файл по URL.
+
             Args:
                 url (str): URL файла
+
             Returns:
                 str: Путь или ошибка
             """
@@ -63,8 +66,10 @@ class BasicAgent:
         def read_pdf(path: str) -> str:
             """
             Читает PDF.
+
             Args:
                 path (str): Путь
+
             Returns:
                 str: Текст (до 4000 символов)
             """
@@ -79,9 +84,11 @@ class BasicAgent:
         def read_excel(path: str, sheet: str = None) -> str:
             """
             Читает Excel.
+
             Args:
                 path (str): Путь
                 sheet (str, optional): Лист
+
             Returns:
                 str: Таблица или ошибка
             """
@@ -104,24 +111,20 @@ class BasicAgent:
     def __call__(self, question: str) -> str:
         print(f"Вопрос: {question[:120]}...")
 
-        # Обрезка длинных вопросов
+        # Обрезка
         if len(question) > 2500:
             question = question[:2500] + "\n[Обрезано из-за длины.]"
 
-        # Хак для файлов/видео/аудио/attached
+        # Хак для файлов/видео/аудио
         q = question.lower()
-        if any(k in q for k in [".mp3", "audio", "recording", "voice", "youtube.com", "video", "attached", "file", "excel", "pdf", "image", "jpg", "png"]):
-            question += "\nЕсли есть URL или attached файл — скачай, прочитай и отвечай только по фактам из него. Не придумывай числа, имена или данные."
-
-        # Хак для шахмат
-        if "chess" in q or "image" in q or ".jpg" in q or ".png" in q:
-            question += "\nЕсли есть URL изображения — скачай и опиши позицию или ищи похожую. Не придумывай ход."
+        if any(k in q for k in [".mp3", "audio", "recording", "voice", "youtube.com", "video", "attached", "file", "excel", "pdf", "image"]):
+            question += "\nЕсли есть URL — скачай файл/видео и читай/ищи транскрипт/описание. Не придумывай."
 
         try:
             result = self.agent.run(question)
             answer = str(result).strip()
 
-            # Жёсткая очистка + защита от длинных ответов
+            # Жёсткая очистка
             prefixes = [
                 "Final Answer", "Final answer", "Answer:", "The answer is",
                 "So the final answer is", "```", "boxed{", "}", "[/INST]", "</s>",
@@ -136,8 +139,8 @@ class BasicAgent:
                 answer = answer[1:-1].strip()
 
             answer = answer.strip()
-            # Если ответ слишком длинный или подозрительный — обрезаем
-            if len(answer) > 300 or "придум" in answer.lower() or answer.count(",") > 20:
+            # Если ответ слишком длинный или странный — обрезаем
+            if len(answer) > 300 or "придум" in answer.lower():
                 answer = answer[:150] + "..." if len(answer) > 150 else answer
 
             print(f"Ответ: {answer[:150]}...")