Update app.py

app.py

@@ -1,14 +1,13 @@
 from flask import Flask, request, jsonify
 from transformers import pipeline
 import logging
-import re
 
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
 app = Flask(__name__)
 
-# Инициализация pipeline с правильными параметрами
+# Инициализация pipeline
 try:
     chatbot = pipeline(
         "text-generation",
@@ -16,93 +15,56 @@ try:
         device_map="auto",
         model_kwargs={"torch_dtype": "auto"}
     )
-    logger.info("Russian model loaded successfully")
+    logger.info("Model loaded successfully")
 except Exception as e:
-    logger.error(f"Error loading Russian model: {e}")
-    chatbot = pipeline(
-        "text-generation", 
-        model="microsoft/DialoGPT-medium",
-        device_map="auto"
-    )
-    logger.info("Fallback English model loaded")
-
-def clean_response(text, user_message):
-    """Очистка ответа от мусора"""
-    # Удаляем промпт если он остался
-    text = text.replace(f"Пользователь: {user_message}", "").replace("Ассистент:", "").strip()
-    
-    # Удаляем HTML entities
-    text = re.sub(r'&[a-z]+;', '', text)
-    
-    # Удаляем URL
-    text = re.sub(r'http\S+', '', text)
-    
-    # Удаляем странные числовые последовательности
-    text = re.sub(r'\d{8,}', '', text)
-    
-    # Берем только первую осмысленную часть до следующего "Пользователь:" или цифр
-    if "Пользователь:" in text:
-        text = text.split("Пользователь:")[0]
-    
-    # Удаляем лишние переносы и пробелы
-    text = re.sub(r'\n+', ' ', text)
-    text = re.sub(r'\s+', ' ', text).strip()
-    
-    # Обрезаем слишком длинные ответы
-    if len(text) > 200:
-        sentences = re.split(r'[.!?]', text)
-        if len(sentences) > 1:
-            text = '.'.join(sentences[:2]) + '.'
-        else:
-            text = text[:200] + '...'
-    
-    return text
+    logger.error(f"Error: {e}")
+    chatbot = None
 
 @app.route('/health', methods=['GET'])
 def health_check():
-    return jsonify({"status": "healthy", "message": "Service is running"})
+    return jsonify({"status": "healthy", "model_loaded": chatbot is not None})
 
 @app.route('/chat', methods=['POST'])
 def chat():
     try:
+        if not chatbot:
+            return jsonify({"error": "Model not loaded"}), 500
+            
         data = request.get_json()
+        user_message = data.get('message', '').strip()
         
-        if not data or 'message' not in data:
-            return jsonify({"error": "Missing 'message' in request"}), 400
-        
-        user_message = data['message'].strip()
-        logger.info(f"Received message: {user_message}")
-        
-        # Создаем четкий промпт для модели
-        prompt = f"""Ты - полезный AI ассистент. Дай полный ответ и по делу на вопрос пользователя, но не слишком длинный.
+        if not user_message:
+            return jsonify({"error": "Empty message"}), 400
+
+        logger.info(f"Received: {user_message}")
+
+        # Четкий промпт для адекватного общения
+        prompt = f"""Ты - профессиональный помощник по Telegram стикерам. Отвечай вежливо и по делу.
 
 Пользователь: {user_message}
-Ассистент:"""
-        
-        # Генерация с строгими параметрами
+Помощник:"""
+
         response = chatbot(
             prompt,
-            max_new_tokens=100,  # Ограничиваем длину ответа
-            temperature=0.3,     # Меньше случайности
+            max_new_tokens=100,
+            temperature=0.7,
             do_sample=True,
-            top_p=0.85,
-            repetition_penalty=1.2,
+            top_p=0.9,
+            repetition_penalty=1.1,
             num_return_sequences=1,
             pad_token_id=chatbot.tokenizer.eos_token_id,
-            truncation=True,     # Явное указание усечения
-            no_repeat_ngram_size=2  # Предотвращает повторение фраз
+            truncation=True
         )
-        
-        # Извлекаем и очищаем ответ
+
+        # Извлекаем ответ
         generated_text = response[0]['generated_text']
         assistant_response = generated_text.replace(prompt, "").strip()
-        assistant_response = clean_response(assistant_response, user_message)
         
-        # Если ответ пустой или слишком короткий - дефолтный ответ
-        if not assistant_response or len(assistant_response) < 2:
-            assistant_response = "Я AI ассистент. Чем могу помочь?"
+        # Очищаем ответ
+        if "Пользователь:" in assistant_response:
+            assistant_response = assistant_response.split("Пользователь:")[0].strip()
         
-        logger.info(f"Generated response: {assistant_response}")
+        logger.info(f"Response: {assistant_response}")
         
         return jsonify({
             "response": assistant_response,
@@ -110,18 +72,12 @@ def chat():
         })
         
     except Exception as e:
-        logger.error(f"Error in /chat: {e}")
-        return jsonify({"error": "Internal server error"}), 500
+        logger.error(f"Error: {e}")
+        return jsonify({"error": "Internal error"}), 500
 
 @app.route('/', methods=['GET'])
 def home():
-    return jsonify({
-        "message": "Chat API is running", 
-        "endpoints": {
-            "health": "GET /health",
-            "chat": "POST /chat"
-        }
-    })
+    return jsonify({"message": "Sticker assistant API"})
 
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=7860, debug=False)