Update app.py

app.py

@@ -1,83 +1,196 @@
 import gradio as gr
-from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel
+from transformers import pipeline
+from ddgs import DDGS
+import re
 
-class SimpleAIAgent:
+class ImprovedAIAgent:
     def __init__(self):
-        print("Initializing AI Agent...")
+        print("Initializing Improved AI Agent...")
         
-        # Используем языковую модель от Hugging Face
-        self.model = HfApiModel("Qwen/Qwen2.5-Coder-32B-Instruct")
-        
-        # Инструмент для поиска в интернете
-        self.search_tool = DuckDuckGoSearchTool()
-        
-        # Создаем агента который может искать в интернете
-        self.agent = CodeAgent(
-            tools=[self.search_tool],
-            model=self.model,
-            max_steps=4
+        # Используем более подходящую модель для вопросов и ответов
+        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!")
 
+    def needs_web_search(self, question):
+        """Определяем, нужен ли поиск в интернете для этого вопроса"""
+        question_lower = question.lower()
+        
+        # Ключевые слова, которые требуют актуальной информации
+        search_keywords = [
+            'current', 'today', 'now', 'latest', 'recent', 'weather',
+            'news', '2024', '2023', 'update', 'breaking',
+            'how to', 'tutorial', 'guide', 'steps to'
+        ]
+        
+        # Вопросы, начинающиеся с what, who, when, where (фактологические)
+        factual_patterns = [
+            'what is', 'who is', 'when did', 'where is', 'which country',
+            'capital of', 'population of', 'president of'
+        ]
+        
+        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}")
         
-        # Создаем инструкцию для агента
-        prompt = f"""
-        The user asked: {message}
-        
-        Please provide a helpful and accurate answer.
-        If you need current information, use the search tool to find it online.
-        Keep your response clear and conversational.
-        """
-        
         try:
-            # Получаем ответ от агента
-            response = self.agent.run(prompt)
+            # Определяем, нужен ли поиск в интернете
+            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
+            )
             
-            # Очищаем ответ от технических деталей
-            clean_response = self.clean_answer(response)
-            print(f"Agent replied: {clean_response[:100]}...")
+            # Извлекаем ответ
+            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 self.is_nonsense_response(clean_response, message):
+                clean_response = "I'm not sure about that. Could you provide more context or rephrase your question?"
+            
+            print(f"Assistant: {clean_response[:100]}...")
             return clean_response
             
         except Exception as e:
-            error_msg = f"Sorry, I encountered an error: {str(e)}"
             print(f"Error: {e}")
-            return error_msg
+            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()
+        
+        # Если ответ повторяет вопрос
+        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'
+        ]
+        
+        if any(phrase in response_lower for phrase in nonsense_phrases):
+            return True
+            
+        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:
-            # Пропускаем строки про инструменты и процесс поиска
-            lower_line = line.lower()
-            if any(word in lower_line for word in ['tool:', 'searching', 'step', 'using tool']):
-                continue
-                
-            # Пропускаем пустые строки в начале
-            if line.strip():
+            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 = '\n'.join(clean_lines).strip()
+        result = ' '.join(clean_lines)
+        
+        # Убираем лишние пробелы
+        result = re.sub(r'\s+', ' ', result).strip()
         
         # Если ответ слишком длинный, обрезаем
-        if len(result) > 1500:
-            result = result[:1497] + "..."
+        if len(result) > 800:
+            result = result[:797] + "..."
             
-        return result if result else "I couldn't find a good answer to that question."
+        return result if result else "I couldn't find a clear answer to that question."
 
 # Создаем экземпляр агента
-ai_agent = SimpleAIAgent()
+ai_agent = ImprovedAIAgent()
 
 # Создаем интерфейс чата
 with gr.Blocks(title="My AI Assistant") as chat_interface: