Update app.py

app.py

@@ -57,109 +57,120 @@ logging.basicConfig(
     ]
 )
 
-
 def generate_search_query(prompt: str) -> dict:
-    BRAND_ALIASES = {
-        "hewlett-packard": "HP", "hewlett packard": "HP", "hp": "HP", "хп": "HP", "эйчпи": "HP",
-        "коника": "Konica Minolta", "конике": "Konica Minolta", "коника минолта": "Konica Minolta",
-        "кэнон": "Canon", "кэноне": "Canon", "канон": "Canon", "каноне": "Canon", 
-        "кенон": "Canon", "кеноне": "Canon", "canon": "Canon",
-        "ксерокс": "Xerox", "ксероксе": "Xerox", "херокс": "Xerox", "хероксе": "Xerox", "херох": "Xerox",
-        "epson": "Epson", "эпсон": "Epson", "brother": "Brother", "братер": "Brother", "братэр": "Brother",
-        "kyocera": "Kyocera", "киосера": "Kyocera", "ricoh": "Ricoh", "рико": "Ricoh", "lexmark": "Lexmark",
-        "лексмарк": "Lexmark", "oki": "OKI", "оки": "OKI", "sharp": "Sharp", "шарп": "Sharp",
-        "samsung": "Samsung", "самсунг": "Samsung", "dell": "Dell", "делл": "Dell"
+    # Создаем расширенный системный промт с требованием англоязычного вывода
+    system_prompt = """
+    You are a technical expert. Extract structured data from the user's query and generate an English search query.
+    Return data in strict JSON format with these fields:
+    - brand: English brand name (HP, Canon, Konica Minolta, etc.)
+    - model: equipment model (model name only)
+    - error_code: error code (if present)
+    - problem_description: brief English problem description (1-2 sentences)
+    - search_query: full English search query for technical documentation
+    
+    Important rules:
+    1. All fields MUST be in English
+    2. For brands use official English names
+    3. Remove brand mentions and the word "error" from model name
+    4. If error code is specified - include it in search_query
+    5. Problem description should be concise technical terms (max 7 words)
+    6. Search query should be optimized for technical documentation search
+    
+    Examples:
+    
+    Query: "Коника Минольта bizhub 368 выдает ошибку C-2557"
+    Response: {
+        "brand": "Konica Minolta",
+        "model": "bizhub 368",
+        "error_code": "C-2557",
+        "problem_description": "C-2557 error code",
+        "search_query": "Konica Minolta bizhub 368 C-2557 error code technical solution"
     }
-
-    # Улучшенное регулярное выражение для кодов ошибок
-    error_pattern = r'\b(?:[A-Za-z]{1,4}[- _]?)?\d{2,5}[A-Za-z]?\b|\bC[- ]?\d{4,5}\b|\bE[- ]?[A-Z]\d{2,3}\b'
     
-    messages = [
-        {
-            "role": "system", 
-            "content": """Извлеки информацию из запроса в формате JSON:
-            - brand: английское название бренда (HP, Canon, Konica Minolta и т.д.)
-            - model: модель оборудования
-            - error_code: код ошибки (если есть)
-            - problem_description: краткое описание проблемы на русском
-            
-            Примеры:
-            Запрос: "HP LaserJet 1020 не включается"
-            Ответ: {"brand": "HP", "model": "LaserJet 1020", "error_code": "", "problem_description": "Принтер не включается"}
-            
-            Запрос: "Canon imagePROGRAF TM-300 ошибка E07"
-            Ответ: {"brand": "Canon", "model": "imagePROGRAF TM-300", "error_code": "E07", "problem_description": "Ошибка E07"}
-            """
-        },
-        {"role": "user", "content": prompt}
-    ]
-
+    Query: "Не печатает HP LaserJet Pro M404dn"
+    Response: {
+        "brand": "HP",
+        "model": "LaserJet Pro M404dn",
+        "error_code": "",
+        "problem_description": "printer not printing",
+        "search_query": "HP LaserJet Pro M404dn not printing technical fix"
+    }
+    
+    Query: "Hewlett-Packard Color LaserJet MFP E77825 Ошибка 63.00.41"
+    Response: {
+        "brand": "HP",
+        "model": "Color LaserJet MFP E77825",
+        "error_code": "63.00.41",
+        "problem_description": "error 63.00.41",
+        "search_query": "HP Color LaserJet MFP E77825 error 63.00.41 technical solution"
+    }
+    
+    Query: "Canon imagePROGRAF TM-200 Ошибка EC12-2F2B"
+    Response: {
+        "brand": "Canon",
+        "model": "imagePROGRAF TM-200",
+        "error_code": "EC12-2F2B",
+        "problem_description": "EC12-2F2B error",
+        "search_query": "Canon imagePROGRAF TM-200 EC12-2F2B error troubleshooting"
+    }
+    """
+    
     try:
+        # Запрашиваем структурированные данные у Mistral
         response = mistral_client.chat.complete(
             model=MISTRAL_MODEL,
-            messages=messages,
+            messages=[
+                {"role": "system", "content": system_prompt},
+                {"role": "user", "content": prompt}
+            ],
             temperature=0.1,
-            max_tokens=250,
+            max_tokens=350,
             response_format={"type": "json_object"}
         )
 
+        # Парсим JSON ответ
         json_data = json.loads(response.choices[0].message.content)
         
-        # Улучшенное извлечение кода ошибки, если Mistral его пропустил
-        if not json_data.get('error_code', ''):
-            error_match = re.search(error_pattern, prompt, re.IGNORECASE)
-            if error_match:
-                json_data['error_code'] = error_match.group(0).strip()
-
-        # Улучшенное извлечение модели, если Mistral ее пропустил
-        if not json_data.get('model', ''):
-            # Ищем последовательность из 2-5 слов, начинающуюся с заглавной буквы или цифр
-            model_match = re.search(r'(\b[A-Z0-9][\w-]+(?:\s+[A-Z0-9][\w-]+){1,4}\b', prompt)
-            if model_match:
-                json_data['model'] = model_match.group(0).strip()
-
-        # Формируем поисковый запрос
-        if json_data.get('error_code', ''):
-            search_query = f"{json_data['brand']} {json_data['model']} ошибка {json_data['error_code']}"
-        else:
-            search_query = f"{json_data['brand']} {json_data['model']} {json_data['problem_description']}"
-
-        json_data['search_query'] = re.sub(r'\s+', ' ', search_query).strip()
+        # Простая валидация и нормализация полей
+        required_fields = ['brand', 'model', 'error_code', 'problem_description', 'search_query']
+        for field in required_fields:
+            if field not in json_data:
+                json_data[field] = ""
+        
+        # Очистка модели от бренда
+        if json_data['brand'] and json_data['model']:
+            json_data['model'] = re.sub(
+                re.escape(json_data['brand']), 
+                '', 
+                json_data['model'], 
+                flags=re.IGNORECASE
+            ).strip()
+        
+        # Обеспечиваем англоязычный поисковый запрос
+        if not json_data['search_query']:
+            # Fallback генерация поискового запроса
+            search_parts = [json_data['brand'], json_data['model']]
+            if json_data['error_code']:
+                search_parts.append(f"error {json_data['error_code']}")
+            if json_data['problem_description']:
+                search_parts.append(json_data['problem_description'])
+            json_data['search_query'] = " ".join(search_parts).strip() + " technical solution"
+        
         return json_data
 
     except Exception as e:
-        message_queue.put(('log', f"❌ Ошибка извлечения данных: {str(e)}"))
-        
-        # Расширенный fallback для брендов
-        brand = ""
-        prompt_lower = prompt.lower()
-        for name, alias in BRAND_ALIASES.items():
-            if name in prompt_lower:
-                brand = alias
-                break
-                
-        # Извлечение кода ошибки
-        error_match = re.search(error_pattern, prompt, re.IGNORECASE)
-        error_code = error_match.group(0).strip() if error_match else ""
+        # Простой fallback на случай ошибки
+        error_msg = f"❌ Ошибка извлечения данных: {str(e)}"
+        message_queue.put(('log', error_msg))
         
-        # Извлечение модели (более простое)
-        model_parts = []
-        for word in prompt.split():
-            if word[0].isupper() or word.isupper() or word.isdigit():
-                model_parts.append(word)
-            elif model_parts:
-                break
-        model = " ".join(model_parts) if model_parts else ""
-
         return {
-            'brand': brand,
-            'model': model,
-            'error_code': error_code,
-            'problem_description': prompt,
-            'search_query': prompt
+            'brand': "",
+            'model': "",
+            'error_code': "",
+            'problem_description': "",
+            'search_query': prompt  # Используем оригинальный запрос как fallback
         }
 
-
 def web_search(query: str) -> tuple:
     try:
         message_queue.put(('log', f"🔍 Провожу поиск по запросу: {query}"))
@@ -169,8 +180,8 @@ def web_search(query: str) -> tuple:
             "api_key": os.getenv("SERPAPI_KEY"),
             "engine": "google",
             "q": query,
-            "hl": "ru",
-            "gl": "ru",
+            "hl": "en",
+            "gl": "us",
             "num": 10,
             "safe": "off",
         }