Update app.py

app.py

@@ -5,14 +5,10 @@ from smolagents.models import TransformersModel
 class SimpleAIAgent:
     def __init__(self):
         print("Initializing AI Agent...")
-        
-        # Используем языковую модель от Hugging Face
-        self.model = TranformersModel("microsoft/DialoGPT-medium")
-        
-        # Инструмент для поиска в интернете
+        self.model = TransformersModel("microsoft/DialoGPT-medium")
+
         self.search_tool = DuckDuckGoSearchTool()
-        
-        # Создаем агента который может искать в интернете
+
         self.agent = CodeAgent(
             tools=[self.search_tool],
             model=self.model,
@@ -26,8 +22,7 @@ class SimpleAIAgent:
         Основная функция для общения с агентом
         """
         print(f"User asked: {message}")
-        
-        # Создаем инструкцию для агента
+
         prompt = f"""
         The user asked: {message}
         
@@ -37,10 +32,8 @@ class SimpleAIAgent:
         """
         
         try:
-            # Получаем ответ от агента
             response = self.agent.run(prompt)
-            
-            # Очищаем ответ от технических деталей
+ 
             clean_response = self.clean_answer(response)
             print(f"Agent replied: {clean_response[:100]}...")
             
@@ -59,33 +52,26 @@ class SimpleAIAgent:
         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():
                 clean_lines.append(line)
-        
-        # Собираем обратно в текст
+
         result = '\n'.join(clean_lines).strip()
-        
-        # Если ответ слишком длинный, обрезаем
+
         if len(result) > 1500:
             result = result[:1497] + "..."
             
         return result if result else "I couldn't find a good answer to that question."
 
-# Создаем экземпляр агента
 ai_agent = SimpleAIAgent()
 
-# Создаем интерфейс чата
 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",
@@ -95,17 +81,13 @@ with gr.Blocks(title="My AI Assistant") as chat_interface:
     clear_btn = gr.Button("Clear Chat")
     
     def respond(message, chat_history):
-        # Получаем ответ от агента
         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_btn.click(lambda: None, None, chatbot, queue=False)
 
-# Запускаем приложение
 if __name__ == "__main__":
     print("Starting AI Chat Assistant...")
     chat_interface.launch(share=True)