Update app.py

app.py

@@ -1,189 +1,110 @@
 import gradio as gr
-from transformers import pipeline
-from ddgs import DDGS
-import re
+from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel
 
-class ImprovedAIAgent:
+class SimpleAIAgent:
     def __init__(self):
-        print("Starting Improved AI Agent...")
+        print("Initializing AI Agent...")
         
-        # Используем более умную модель для фактов
-        self.model = pipeline(
-            "text-generation", 
-            model="microsoft/DialoGPT-medium",  # Более качественная чем small
-            max_length=300,
-            temperature=0.3,  # Меньше креативности, больше фактов
-            do_sample=True
-        )
+        # Используем языковую модель от Hugging Face
+        self.model = HfApiModel("Qwen/Qwen2.5-Coder-32B-Instruct")
         
-        # Инициализируем поиск
-        self.ddgs = DDGS()
-        print("Improved AI Agent ready!")
-    
-    def search_web(self, query):
-        """Поиск в интернете через DuckDuckGo"""
-        try:
-            results = list(self.ddgs.text(query, max_results=3))
-            if results:
-                # Собираем информацию из нескольких результатов
-                combined_info = []
-                for result in results[:2]:  # Берем первые 2 результата
-                    text = result.get('body', '')
-                    if text and len(text) > 10:  # Фильтруем слишком короткие
-                        combined_info.append(text[:200])  # Ограничиваем длину
-                
-                if combined_info:
-                    return " | ".join(combined_info)
-            return None
-        except Exception as e:
-            print(f"Search error: {e}")
-            return None
-    
-    def is_factual_question(self, question):
-        """Определяем, является ли вопрос фактологическим"""
-        question_lower = question.lower()
+        # Инструмент для поиска в интернете
+        self.search_tool = DuckDuckGoSearchTool()
         
-        factual_keywords = [
-            'what', 'who', 'when', 'where', 'which', 'how many', 'how much',
-            'capital', 'population', 'president', 'country', 'city',
-            'define', 'definition', 'is the', 'are the'
-        ]
+        # Создаем агента который может искать в интернете
+        self.agent = CodeAgent(
+            tools=[self.search_tool],
+            model=self.model,
+            max_steps=4
+        )
         
-        return any(keyword in question_lower for keyword in factual_keywords)
-    
+        print("AI Agent ready!")
+
     def chat(self, message, history):
-        print(f"User: {message}")
+        """
+        Основная функция для общения с агентом
+        """
+        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:
-            # Для фактологических вопросов используем поиск
-            if self.is_factual_question(message):
-                search_result = self.search_web(message)
-                
-                if search_result:
-                    # Промпт с найденной информацией
-                    prompt = f"""Based on this information: {search_result}
-                    
-                    Question: {message}
-                    
-                    Provide a concise and accurate answer:"""
-                else:
-                    # Если поиск не сработал, используем общий промпт
-                    prompt = f"""Answer this question accurately: {message}
-                    
-                    If you don't know the exact answer, say so."""
-            else:
-                # Для не-фактологических вопросов используем обычный чат
-                prompt = f"User: {message}\nAssistant:"
-            
-            # Генерируем ответ
-            response = self.model(
-                prompt,
-                max_length=400,
-                num_return_sequences=1,
-                temperature=0.3,  # Низкая температура для фактов
-                do_sample=True,
-                pad_token_id=50256,
-                repetition_penalty=1.3
-            )
-            
-            # Извлекаем ответ
-            full_text = response[0]['generated_text']
+            # Получаем ответ от агента
+            response = self.agent.run(prompt)
             
-            if "Answer:" in full_text:
-                answer = full_text.split("Answer:")[-1].strip()
-            elif "Assistant:" in full_text:
-                answer = full_text.split("Assistant:")[-1].strip()
-            else:
-                answer = full_text.replace(prompt, "").strip()
+            # Очищаем ответ от технических деталей
+            clean_response = self.clean_answer(response)
+            print(f"Agent replied: {clean_response[:100]}...")
             
-            # Очищаем ответ
-            clean_answer = self.clean_response(answer)
-            
-            # Если ответ явно неправильный, даем запасной ответ
-            if self.is_wrong_answer(clean_answer, message):
-                clean_answer = "I'm not sure about that. Let me search for more information."
-                search_fallback = self.search_web(message)
-                if search_fallback:
-                    clean_answer = f"Based on what I found: {search_fallback[:300]}..."
-            
-            print(f"Assistant: {clean_answer[:100]}...")
-            return clean_answer
+            return clean_response
             
         except Exception as e:
+            error_msg = f"Sorry, I encountered an error: {str(e)}"
             print(f"Error: {e}")
-            return "Let me check that for you. Could you rephrase your question?"
-    
-    def is_wrong_answer(self, answer, question):
-        """Проверяем, явно ли неправильный ответ"""
-        answer_lower = answer.lower()
-        question_lower = question.lower()
-        
-        # Если вопрос про столицу, но ответ не похож на название города
-        if 'capital' in question_lower:
-            # Проверяем, содержит ли ответ название города
-            if not any(indicator in answer_lower for indicator in ['city', 'capital', 'is']):
-                return True
-            # Если ответ содержит "ambassador" или подобное для вопроса о столице - это ошибка
-            if 'ambassador' in answer_lower:
-                return True
-        
-        return False
-    
-    def clean_response(self, text):
-        """Очистка ответа"""
-        # Убираем технические части
-        text = re.sub(r'Based on this information:.*?Provide a concise answer:', '', text, flags=re.DOTALL)
-        text = re.sub(r'Question:.*?Answer:', '', text, flags=re.DOTALL)
-        
-        # Берем первую релевантную часть
-        sentences = text.split('.')
-        clean_sentences = []
-        
-        for sentence in sentences:
-            sentence = sentence.strip()
-            if sentence and len(sentence) > 5:
-                clean_sentences.append(sentence)
-                if len(clean_sentences) >= 2:  # Ограничиваем 2 предложениями
-                    break
+            return error_msg
+
+    def clean_answer(self, answer):
+        """
+        Убираем техническую информацию из ответа агента
+        """
+        lines = answer.split('\n')
+        clean_lines = []
         
-        result = '. '.join(clean_sentences)
-        if result and not result.endswith('.'):
-            result += '.'
+        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():
+                clean_lines.append(line)
         
-        if len(result) > 300:
-            result = result[:297] + "..."
+        # Собираем обратно в текст
+        result = '\n'.join(clean_lines).strip()
         
-        return result if result else "I need more information to answer that accurately."
+        # Если ответ слишком длинный, обрезаем
+        if len(result) > 1500:
+            result = result[:1497] + "..."
+            
+        return result if result else "I couldn't find a good answer to that question."
 
-# Создаем агента
-agent = ImprovedAIAgent()
+# Создаем экземпляр агента
+ai_agent = SimpleAIAgent()
 
-# Создаем интерфейс
-with gr.Blocks() as app:
-    gr.Markdown("# Improved AI Assistant")
-    gr.Markdown("I can answer factual questions using web search")
-    
-    chatbot = gr.Chatbot(
-        height=400,
-        show_copy_button=True
-    )
+# Создаем интерфейс чата
+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.")
     
+    # Создаем чат-интерфейс
+    chatbot = gr.Chatbot(height=400)
     msg = gr.Textbox(
         label="Your question",
-        placeholder="Ask me factual questions...",
+        placeholder="Ask me anything...",
         lines=2
     )
-    
-    clear = gr.Button("Clear Chat")
+    clear_btn = gr.Button("Clear Chat")
     
     def respond(message, chat_history):
-        bot_message = agent.chat(message, chat_history)
-        chat_history.append((message, bot_message))
+        # Получаем ответ от агента
+        bot_response = ai_agent.chat(message, chat_history)
+        # Добавляем в историю чата
+        chat_history.append((message, bot_response))
         return "", chat_history
     
+    # Обработчики событий
     msg.submit(respond, [msg, chatbot], [msg, chatbot])
-    clear.click(lambda: None, None, chatbot, queue=False)
+    clear_btn.click(lambda: None, None, chatbot, queue=False)
 
+# Запускаем приложение
 if __name__ == "__main__":
-    print("Starting Improved AI Assistant...")
-    app.launch()
+    print("Starting AI Chat Assistant...")
+    chat_interface.launch(share=True)