Update app.py

app.py

@@ -1,223 +1,241 @@
-import gradio as gr
-from transformers import pipeline
-from ddgs import DDGS
-import re
-
-class ImprovedAIAgent:
-    def __init__(self):
-        print("Initializing Improved AI Agent...")
-        
-        # Используем более подходящую модель для вопросов и ответов
-        self.model = pipeline(
-            "text-generation",
-            model="microsoft/DialoGPT-medium",  # Более качественная чем small
-            max_length=400,
-            temperature=0.7,
-            do_sample=True
-        )
-        
-        # Инициализируем поиск
-        self.ddgs = DDGS()
-        print("AI Agent ready!")
+from smolagents import CodeAgent, HfApiModel, load_tool, tool, DuckDuckGoSearchTool
+import datetime
+import requests
+import pytz
+import yaml
+import base64
+import os
+import io
+from PIL import Image
+from tools.final_answer import FinalAnswerTool
+from Gradio_UI import GradioUI
+
+# 1. функции-инструменты
 
-    def needs_web_search(self, question):
-        """Определяем, нужен ли поиск в интернете для этого вопроса"""
-        question_lower = question.lower()
+@tool
+def web_search(query: str) -> str:
+    """Search the web for information using DuckDuckGo.
+    
+    Args:
+        query: The search query to look up
         
-        # Ключевые слова, которые требуют актуальной информации
-        search_keywords = [
-            'current', 'today', 'now', 'latest', 'recent', 'weather',
-            'news', '2024', '2023', 'update', 'breaking',
-            'how to', 'tutorial', 'guide', 'steps to'
-        ]
+    Returns:
+        Search results from DuckDuckGo
+    """
+    try:
+        search_tool = DuckDuckGoSearchTool()
+        results = search_tool(query)
+        if not results or "No results found" in results:
+            return f"Поиск не дал результатов для запроса: '{query}'. Попробуйте другой запрос."
+        return f"**Результаты поиска по '{query}':**\n\n{results}"
+    except Exception as e:
+        return f"Ошибка поиска: {str(e)}"
+
+@tool
+def get_weather(city: str) -> str:
+    """Get current weather for a city using reliable weather services.
+    
+    Args:
+        city: The name of the city to get weather for (e.g., 'Moscow', 'London')
         
-        # Вопросы, начинающиеся с what, who, when, where (фактологические)
-        factual_patterns = [
-            'what is', 'who is', 'when did', 'where is', 'which country',
-            'capital of', 'population of', 'president of'
-        ]
+    Returns:
+        Current weather information in Celsius
+    """
+    try:
+        url = f"https://wttr.in/{city}?format=%C+%t+%h+%w&m&lang=ru"
+        response = requests.get(url, timeout=10)
         
-        return (any(keyword in question_lower for keyword in search_keywords) or
-                any(question_lower.startswith(pattern) for pattern in factual_patterns))
-
-    def search_web(self, query):
-        """Поиск в интернете через DuckDuckGo"""
-        try:
-            results = list(self.ddgs.text(query, max_results=2))
-            if results:
-                # Собираем информацию из результатов
-                combined_info = []
-                for result in results:
-                    text = result.get('body', '')
-                    if text and len(text) > 20:  # Фильтруем слишком короткие
-                        combined_info.append(text[:300])  # Ограничиваем длину
-                
-                if combined_info:
-                    return " ".join(combined_info)
-            return None
-        except Exception as e:
-            print(f"Search error: {e}")
-            return None
-
-    def chat(self, message, history):
-        """
-        Основная функция для общения с агентом
-        """
-        print(f"User asked: {message}")
+        if response.status_code == 200 and response.text.strip():
+            data = response.text.strip()
+            if '°F' in data:
+                data = data.replace('°F', '°C')
+            return f"Погода в {city}:\n{data}"
+       
+        url2 = f"https://wttr.in/{city}?format=%C+%t+%h+%w+%P&m&lang=ru"
+        response2 = requests.get(url2, timeout=10)
         
-        try:
-            # Определяем, нужен ли поиск в интернете
-            if self.needs_web_search(message):
-                search_result = self.search_web(message)
-                
-                if search_result:
-                    # Используем найденную информацию
-                    prompt = f"""Based on this information: {search_result}
-                    
-                    Please answer this question: {message}
-                    
-                    Provide a clear and accurate answer:"""
-                else:
-                    # Если поиск не дал результатов
-                    prompt = f"""Please answer this question: {message}
-                    
-                    If you don't know the answer, please say so."""
-            else:
-                # Для общих вопросов используем обычный промпт
-                prompt = f"User: {message}\nAssistant:"
-            
-            # Генерируем ответ
-            response = self.model(
-                prompt,
-                max_length=500,
-                num_return_sequences=1,
-                temperature=0.7,
-                do_sample=True,
-                pad_token_id=50256,
-                repetition_penalty=1.2
-            )
-            
-            # Извлекаем ответ
-            full_text = response[0]['generated_text']
-            
-            # Извлекаем только ответ ассистента
-            if "Assistant:" in full_text:
-                answer = full_text.split("Assistant:")[-1].strip()
-            elif "answer:" in full_text.lower():
-                # Ищем после "answer:"
-                answer_match = re.search(r'answer:\s*(.*)', full_text, re.IGNORECASE)
-                answer = answer_match.group(1).strip() if answer_match else full_text
-            else:
-                # Убираем промпт из ответа
-                answer = full_text.replace(prompt, "").strip()
-            
-            # Очищаем ответ
-            clean_response = self.clean_answer(answer)
+        if response2.status_code == 200 and response2.text.strip():
+            data = response2.text.strip()
+            if '°F' in data:
+                data = data.replace('°F', '°C')
+            return f"Погода в {city}:\n{data}"
             
-            # Проверяем, не является ли ответ бессмысленным
-            if self.is_nonsense_response(clean_response, message):
-                clean_response = "I'm not sure about that. Could you provide more context or rephrase your question?"
+        return f"Данные о погоде в {city} временно недоступны. Попробуйте позже или проверьте на сайте gismeteo.ru"
             
-            print(f"Assistant: {clean_response[:100]}...")
-            return clean_response
-            
-        except Exception as e:
-            print(f"Error: {e}")
-            return "I'm here to help! What would you like to know?"
-
-    def is_nonsense_response(self, response, question):
-        """Проверяем, является ли ответ бессмысленным"""
-        if not response or len(response) < 5:
-            return True
-            
-        response_lower = response.lower()
-        question_lower = question.lower()
+    except Exception as e:
+        return f"Ошибка при получении погоды: {str(e)}"
+
+@tool
+def get_weather_detailed(city: str) -> str:
+    """Get detailed weather forecast for a city.
+    
+    Args:
+        city: The name of the city
         
-        # Если ответ повторяет вопрос
-        if question_lower in response_lower and len(response) < len(question) + 10:
-            return True
-            
-        # Если ответ содержит явно неподходящие фразы
-        nonsense_phrases = [
-            'the user asked', 'please answer', 'based on this information',
-            'provide a clear', 'i cannot answer', 'as an ai'
-        ]
+    Returns:
+        Detailed weather information in Celsius
+    """
+    try:
+        url = f"https://wttr.in/{city}?m&lang=ru"
+        response = requests.get(url, timeout=10)
         
-        if any(phrase in response_lower for phrase in nonsense_phrases):
-            return True
+        if response.status_code == 200:
+            lines = response.text.split('\n')
+            short_forecast = '\n'.join(lines[:8])  # Первые 8 строк
             
-        return False
-
-    def clean_answer(self, answer):
-        """
-        Убираем техническую информацию из ответа агента
-        """
-        if not answer:
-            return "I don't have an answer for that question."
-        
-        # Убираем технические фразы
-        clean_patterns = [
-            r'based on this information:.*?provide a clear answer:',
-            r'please answer this question:.*?if you don\'t know',
-            r'user:.*?assistant:',
-        ]
-        
-        for pattern in clean_patterns:
-            answer = re.sub(pattern, '', answer, flags=re.IGNORECASE | re.DOTALL)
-        
-        # Разбиваем на строки и фильтруем
-        lines = answer.split('\n')
-        clean_lines = []
-        
-        for line in lines:
-            line = line.strip()
-            if line and len(line) > 3:
-                # Пропускаем технические строки
-                if any(word in line.lower() for word in ['tool:', 'searching', 'step', 'using tool']):
-                    continue
-                clean_lines.append(line)
-        
-        # Собираем обратно в текст
-        result = ' '.join(clean_lines)
-        
-        # Убираем лишние пробелы
-        result = re.sub(r'\s+', ' ', result).strip()
-        
-        # Если ответ слишком длинный, обрезаем
-        if len(result) > 800:
-            result = result[:797] + "..."
+            short_forecast = short_forecast.replace('°F', '°C')
+            return f"Подробный прогноз для {city}:\n{short_forecast}"
+        else:
+            return f"Не удалось получить подробный прогноз для {city}"
             
-        return result if result else "I couldn't find a clear answer to that question."
+    except Exception as e:
+        return f"Ошибка: {str(e)}"
+
+@tool
+def get_current_time() -> str:
+    """Get current local time.
+    
+    Returns:
+        Current date and time
+    """
+    try:
+        current_time = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+        return f"Текущее время: {current_time}"
+    except Exception as e:
+        return f"Ошибка получения времени: {str(e)}"
+
+@tool
+def get_current_time_in_timezone(timezone: str) -> str:
+    """Get current local time in specified timezone.
+    
+    Args:
+        timezone: A valid timezone (e.g., 'America/New_York', 'Europe/Moscow')
+        
+    Returns:
+        Current time in the specified timezone
+    """
+    try:
+        tz = pytz.timezone(timezone)
+        local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
+        return f"Текущее время в {timezone}: {local_time}"
+    except Exception as e:
+        return f"Ошибка получения времени для часового пояса '{timezone}': {str(e)}"
+
+@tool
+def how_are_you() -> str:
+    """Answer questions about how you are doing.
+    
+    Returns:
+        Friendly response about current status
+    """
+    responses = [
+        "У меня всё отлично! Готов помочь вам с любыми вопросами.",
+        "Прекрасно себя чувствую! Что хотели бы узнать?",
+        "Всё хорошо, спасибо! Чем могу быть полезен?",
+        "Отлично! Готов к работе и жду ваших вопросов.",
+        "Замечательно! Как ваши дела?"
+    ]
+    import random
+    return random.choice(responses)
+
+@tool
+def what_are_you_doing() -> str:
+    """Answer questions about what you are currently doing.
+    
+    Returns:
+        Description of current activities
+    """
+    responses = [
+        "Я общаюсь с вами и готов помочь с поиском информации!",
+        "В данный момент я помогаю пользователям находить ответы на их вопросы.",
+        "Я работаю ассистентом - ищу информацию в интернете, сообщаю о погоде и текущем времени.",
+        "Сейчас я здесь, чтобы помочь вам! Чем могу быть полезен?",
+        "Я анализирую ваш запрос и готовлю полезный ответ."
+    ]
+    import random
+    return random.choice(responses)
 
-# Создаем экземпляр агента
-ai_agent = ImprovedAIAgent()
+@tool
+def calculate_math(expression: str) -> str:
+    """Calculate mathematical expressions safely.
+    
+    Args:
+        expression: A mathematical expression (e.g., '2+2', '5*3/2')
+        
+    Returns:
+        Result of the calculation
+    """
+    try:
+        allowed_chars = set('0123456789+-*/.() ')
+        if all(c in allowed_chars for c in expression):
+            result = eval(expression)
+            return f"Результат: {result}"
+        else:
+            return "Используйте только цифры и +-*/.()"
+    except:
+        return "Ошибка в выражении"
 
-# Создаем интерфейс чата
-with gr.Blocks(title="My AI Assistant") as chat_interface:
-    gr.Markdown("# My AI Assistant")
-    gr.Markdown("Ask me anything! I can search the internet for current information.")
+@tool
+def suggest_weather_sources(city: str) -> str:
+    """Suggest reliable weather sources for a city.
     
-    # Создаем чат-интерфейс
-    chatbot = gr.Chatbot(height=400)
-    msg = gr.Textbox(
-        label="Your question",
-        placeholder="Ask me anything...",
-        lines=2
-    )
-    clear_btn = gr.Button("Clear Chat")
+    Args:
+        city: The name of the city
+        
+    Returns:
+        List of reliable weather sources
+    """
+    sources = [
+        f"Gismeteo: https://www.gismeteo.ru/weather-{city.lower()}-4368/",
+        f"Yandex.Погода: https://yandex.ru/pogoda/{city.lower()}",
+        f"Weather.com: https://weather.com/weather/today/l/{city}",
+        f"AccuWeather: https://www.accuweather.com/ru/ru/{city.lower()}/weather-forecast"
+    ]
     
-    def respond(message, chat_history):
-        # Получаем ответ от агента
-        bot_response = ai_agent.chat(message, chat_history)
-        # Добавляем в историю чата
-        chat_history.append((message, bot_response))
-        return "", chat_history
+    return f"Надежные источники погоды для {city}:\n" + "\n".join(sources)
+
+# 2. остальные объекты
+
+final_answer = FinalAnswerTool()
+
+model = HfApiModel(
+    max_tokens=2096,
+    temperature=0.5,
+    model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
+    custom_role_conversions=None,
+)
+
+with open("prompts.yaml", 'r') as stream:
+    prompt_templates = yaml.safe_load(stream)
+
+if "final_answer" not in prompt_templates:
+    prompt_templates["final_answer"] = {
+        "pre_messages": "Based on my research: ",
+        "post_messages": ""
+    }
     
-    # Обработчики событий
-    msg.submit(respond, [msg, chatbot], [msg, chatbot])
-    clear_btn.click(lambda: None, None, chatbot, queue=False)
-
-# Запускаем приложение
-if __name__ == "__main__":
-    print("Starting AI Chat Assistant...")
-    chat_interface.launch(share=True)
+# Создаем агента с новым именем и обновленными инструментами
+agent = CodeAgent(
+    model=model,
+    tools=[
+        final_answer, 
+        web_search, 
+        get_weather, 
+        get_weather_detailed, 
+        get_current_time,
+        get_current_time_in_timezone, 
+        how_are_you,
+        what_are_you_doing,
+        calculate_math, 
+        suggest_weather_sources
+    ],
+    max_steps=8,
+    verbosity_level=1,
+    grammar=None,
+    planning_interval=None,
+    name="WeatherTimeBot",  # Новое имя агента
+    description="Ассистент для поиска информации, прогноза погоды и определения времени",
+    prompt_templates=prompt_templates
+)
+
+GradioUI(agent).launch()