Update app.py

app.py

@@ -22,6 +22,7 @@ MAX_RESULTS = 5
 # Новый клиент Mistral
 mistral_client = Mistral(api_key=os.getenv("MISTRAL_API_KEY"))
 
+# Обновленный SYSTEM_PROMPT с улучшенным форматированием
 SYSTEM_PROMPT = """
 Ты PrintMaster, сервисный инженер по печатной технике. Правила:
 1. Формат ответа строго:
@@ -29,8 +30,8 @@ SYSTEM_PROMPT = """
    **Решение:** [пошаговые действия]
 2. В решении включай ВСЕ шаги из найденной информации
 3. Шаги решения нумеруй цифрами с точкой (1., 2., 3.)
-4. Не добавляй дополнительные разделы после "Источники"
-5. Не дублируй разделы
+4. Всегда завершай ответ разделом **Источники информации** (без нумерации)
+5. Не добавляй дополнительные разделы после "Источники информации"
 6. Будь краток, но исчерпывающе
 7. Используй точные технические термины
 8. Указывай артикулы деталей при замене
@@ -192,19 +193,22 @@ def web_search(query: str) -> tuple:
 
 
 def clean_response(response: str) -> str:
-    response = re.sub(r'</?assistant>|<\|system\|>|</s>', '', response, flags=re.IGNORECASE)
-    response = re.sub(r'(\*\*Проблема:\*\*.+?)(\*\*Проблема:\*\*)', r'\1', response, flags=re.DOTALL)
-    response = re.sub(r'(\*\*Решение:\*\*.+?)(\*\*Решение:\*\*)', r'\1', response, flags=re.DOTALL)
+    # Удаляем все упоминания источников с нумерацией
+    response = re.sub(r'Источники:\s*(\[\d+\](,\s*)?+', '', response)
     response = re.sub(r'(\*\*Источники:\*\*.+?)(\*\*Источники:\*\*)', r'\1', response, flags=re.DOTALL)
+    
+    # Оставляем только раздел "Источники информации"
+    response = re.sub(r'(\*\*Источники информация?\*\*)', '', response, flags=re.IGNORECASE)
+    
+    # Убираем лишние переносы
     response = re.sub(r'\n\s*\n', '\n\n', response)
     response = re.sub(r'[ \t]+', ' ', response)
     response = re.sub(r'\s*\[Решение\]\s*', '', response)
-    response = re.sub(r'\*\*Источники:\*\*\s*$', '', response)
-
-    last_dot = response.rfind('.')
-    if last_dot != -1:
-        response = response[:last_dot + 1]
-
+    
+    # Убираем дублирование разделов
+    response = re.sub(r'(\*\*Проблема:\*\*.+?)(\*\*Проблема:\*\*)', r'\1', response, flags=re.DOTALL)
+    response = re.sub(r'(\*\*Решение:\*\*.+?)(\*\*Решение:\*\*)', r'\1', response, flags=re.DOTALL)
+    
     return response.strip()
 
 
@@ -278,16 +282,15 @@ def process_query(prompt: str):
             critical_steps.append("При необходимости замените неисправные компоненты")
 
         if critical_steps:
-            solution_section = re.search(r'\*\*Решение:\*\*(.+?)(\*\*Источники:\*\*|$)',
+            solution_section = re.search(r'\*\*Решение:\*\*(.+?)(\*\*Источники информация?\*\*|$)',
                                          final_response,
                                          flags=re.DOTALL)
             if solution_section:
                 updated_solution = solution_section.group(1) + "\n" + "\n".join(critical_steps)
                 final_response = final_response.replace(solution_section.group(1), updated_solution)
 
-        if sources:
-            sources_ref = ", ".join([f"[{i+1}]" for i in range(len(sources))])
-            final_response += f"\n\n**Источники:** {sources_ref}"
+        # Всегда добавляем раздел "Источники информации"
+        final_response += "\n\n**Источники информации**"
 
         message_queue.put(('response_end', final_response))
         message_queue.put(('sources', json.dumps(sources)))
@@ -332,4 +335,4 @@ def stream():
 
 
 if __name__ == '__main__':
-    app.run(host='0.0.0.0', port=7860, debug=False)
+    app.run(host='0.0.0.0', port=7860, debug=False)