forked from HSE_team/BetterCallPraskovia
styling add
This commit is contained in:
parent
570f0b7ea7
commit
ef71c67683
@ -115,8 +115,9 @@ class DocumentUseCases:
|
||||
await self._send_telegram_notification(
|
||||
telegram_id,
|
||||
"🔄 <b>Начинаю индексацию документа...</b>\n\n"
|
||||
f"Документ: <b>{title}</b>\n"
|
||||
"Это может занять некоторое время."
|
||||
f"📄 <b>Документ:</b> {title}\n\n"
|
||||
f"Это может занять некоторое время.\n"
|
||||
f"Вы получите уведомление по завершении."
|
||||
)
|
||||
|
||||
chunks = await self.rag_service.index_document(document)
|
||||
@ -124,9 +125,12 @@ class DocumentUseCases:
|
||||
await self._send_telegram_notification(
|
||||
telegram_id,
|
||||
"✅ <b>Индексация завершена!</b>\n\n"
|
||||
f"Документ: <b>{title}</b>\n"
|
||||
f"Проиндексировано чанков: <b>{len(chunks)}</b>\n\n"
|
||||
"Теперь вы можете задавать вопросы по этому документу."
|
||||
f"━━━━━━━━━━━━━━━━━━━━━━━━━━━━\n\n"
|
||||
f"📄 <b>Документ:</b> {title}\n"
|
||||
f"📊 <b>Проиндексировано чанков:</b> {len(chunks)}\n\n"
|
||||
f"━━━━━━━━━━━━━━━━━━━━━━━━━━━━\n\n"
|
||||
f"💡 <b>Теперь вы можете задавать вопросы по этому документу!</b>\n"
|
||||
f"Просто напишите ваш вопрос, и я найду ответ на основе загруженного документа."
|
||||
)
|
||||
except Exception as e:
|
||||
print(f"Ошибка при автоматической индексации документа {document.document_id}: {e}")
|
||||
|
||||
@ -3,6 +3,8 @@ from aiogram.types import Message
|
||||
from tg_bot.config.settings import settings
|
||||
from tg_bot.domain.user_service import UserService, User
|
||||
from tg_bot.application.services.rag_service import RAGService
|
||||
import re
|
||||
|
||||
|
||||
router = Router()
|
||||
rag_service = RAGService()
|
||||
@ -60,22 +62,37 @@ async def process_premium_question(message: Message, user: User, question_text:
|
||||
|
||||
# Беседа уже сохранена в бэкенде через API /rag/question
|
||||
|
||||
import re
|
||||
formatted_answer = answer
|
||||
formatted_answer = re.sub(r'\*\*(.+?)\*\*', r'<b>\1</b>', formatted_answer)
|
||||
formatted_answer = re.sub(r'^(\d+)\.\s+', r'\1. ', formatted_answer, flags=re.MULTILINE)
|
||||
formatted_answer = formatted_answer.replace("- ", "• ")
|
||||
|
||||
response = (
|
||||
f"<b>Ваш вопрос:</b>\n"
|
||||
f"<i>{question_text[:200]}</i>\n\n"
|
||||
f"<b>Ответ:</b>\n{answer}\n\n"
|
||||
f"━━━━━━━━━━━━━━━━━━━━━━━━━━━━\n\n"
|
||||
f"💬 <b>Ответ:</b>\n\n"
|
||||
f"{formatted_answer}\n\n"
|
||||
)
|
||||
|
||||
if sources:
|
||||
response += f"<b>Источники:</b>\n"
|
||||
response += f"━━━━━━━━━━━━━━━━━━━━━━━━━━━━\n\n"
|
||||
response += f"📚 <b>Источники:</b>\n"
|
||||
for idx, source in enumerate(sources[:5], 1):
|
||||
title = source.get('title', 'Без названия')
|
||||
response += f"{idx}. {title}\n"
|
||||
response += "\n<i>Используйте /mycollections для просмотра всех коллекций</i>\n\n"
|
||||
try:
|
||||
from urllib.parse import unquote
|
||||
title = unquote(title)
|
||||
except:
|
||||
pass
|
||||
response += f" {idx}. {title}\n"
|
||||
response += "\n<i>💡 Используйте /mycollections для просмотра всех коллекций</i>\n\n"
|
||||
|
||||
response += (
|
||||
f"<b>Статус:</b> Premium (вопросов безлимитно)\n"
|
||||
f"<b>Всего вопросов:</b> {user.questions_used}"
|
||||
f"━━━━━━━━━━━━━━━━━━━━━━━━━━━━\n\n"
|
||||
f"✨ <b>Статус:</b> Premium (вопросов безлимитно)\n"
|
||||
f"📊 <b>Всего вопросов:</b> {user.questions_used}"
|
||||
)
|
||||
|
||||
except Exception as e:
|
||||
@ -83,9 +100,12 @@ async def process_premium_question(message: Message, user: User, question_text:
|
||||
response = (
|
||||
f"<b>Ваш вопрос:</b>\n"
|
||||
f"<i>{question_text[:200]}</i>\n\n"
|
||||
f"Ошибка при генерации ответа. Попробуйте позже.\n\n"
|
||||
f"<b>Статус:</b> Premium\n"
|
||||
f"<b>Всего вопросов:</b> {user.questions_used}"
|
||||
f"━━━━━━━━━━━━━━━━━━━━━━━━━━━━\n\n"
|
||||
f"❌ <b>Ошибка при генерации ответа.</b>\n"
|
||||
f"Попробуйте позже.\n\n"
|
||||
f"━━━━━━━━━━━━━━━━━━━━━━━━━━━━\n\n"
|
||||
f"✨ <b>Статус:</b> Premium\n"
|
||||
f"📊 <b>Всего вопросов:</b> {user.questions_used}"
|
||||
)
|
||||
|
||||
await message.answer(response, parse_mode="HTML")
|
||||
@ -109,40 +129,56 @@ async def process_free_question(message: Message, user: User, question_text: str
|
||||
|
||||
# Уже все сохранили через /rag/question
|
||||
|
||||
formatted_answer = answer
|
||||
formatted_answer = re.sub(r'\*\*(.+?)\*\*', r'<b>\1</b>', formatted_answer)
|
||||
formatted_answer = re.sub(r'^(\d+)\.\s+', r'\1. ', formatted_answer, flags=re.MULTILINE)
|
||||
formatted_answer = formatted_answer.replace("- ", "• ")
|
||||
response = (
|
||||
f"<b>Ваш вопрос:</b>\n"
|
||||
f"<i>{question_text[:200]}</i>\n\n"
|
||||
f"<b>Ответ:</b>\n{answer}\n\n"
|
||||
f"━━━━━━━━━━━━━━━━━━━━━━━━━━━━\n\n"
|
||||
f"💬 <b>Ответ:</b>\n\n"
|
||||
f"{formatted_answer}\n\n"
|
||||
)
|
||||
|
||||
if sources:
|
||||
response += f"<b>Источники:</b>\n"
|
||||
response += f"━━━━━━━━━━━━━━━━━━━━━━━━━━━━\n\n"
|
||||
response += f"📚 <b>Источники:</b>\n"
|
||||
for idx, source in enumerate(sources[:5], 1):
|
||||
title = source.get('title', 'Без названия')
|
||||
response += f"{idx}. {title}\n"
|
||||
response += "\n<i>Используйте /mycollections для просмотра всех коллекций</i>\n\n"
|
||||
try:
|
||||
from urllib.parse import unquote
|
||||
title = unquote(title)
|
||||
except:
|
||||
pass
|
||||
response += f" {idx}. {title}\n"
|
||||
response += "\n<i>💡 Используйте /mycollections для просмотра всех коллекций</i>\n\n"
|
||||
|
||||
response += (
|
||||
f"<b>Статус:</b> Бесплатный доступ\n"
|
||||
f"<b>Использовано вопросов:</b> {user.questions_used}/{settings.FREE_QUESTIONS_LIMIT}\n"
|
||||
f"<b>Осталось бесплатных:</b> {remaining}\n\n"
|
||||
f"━━━━━━━━━━━━━━━━━━━━━━━━━━━━\n\n"
|
||||
f"📊 <b>Статус:</b> Бесплатный доступ\n"
|
||||
f"📈 <b>Использовано вопросов:</b> {user.questions_used}/{settings.FREE_QUESTIONS_LIMIT}\n"
|
||||
f"🎯 <b>Осталось бесплатных:</b> {remaining}\n\n"
|
||||
)
|
||||
|
||||
if remaining <= 3 and remaining > 0:
|
||||
response += f"<i>Осталось мало вопросов! Для продолжения используйте /buy</i>\n\n"
|
||||
response += f"⚠️ <i>Осталось мало вопросов! Для продолжения используйте /buy</i>\n\n"
|
||||
|
||||
response += f"<i>Для безлимитного доступа: /buy</i>"
|
||||
response += f"💎 <i>Для безлимитного доступа: /buy</i>"
|
||||
|
||||
except Exception as e:
|
||||
print(f"Error generating answer: {e}")
|
||||
response = (
|
||||
f"<b>Ваш вопрос:</b>\n"
|
||||
f"<i>{question_text[:200]}</i>\n\n"
|
||||
f"Ошибка при генерации ответа. Попробуйте позже.\n\n"
|
||||
f"<b>Статус:</b> Бесплатный доступ\n"
|
||||
f"<b>Использовано вопросов:</b> {user.questions_used}/{settings.FREE_QUESTIONS_LIMIT}\n"
|
||||
f"<b>Осталось бесплатных:</b> {remaining}\n\n"
|
||||
f"<i>Для безлимитного доступа: /buy</i>"
|
||||
f"━━━━━━━━━━━━━━━━━━━━━━━━━━━━\n\n"
|
||||
f"❌ <b>Ошибка при генерации ответа.</b>\n"
|
||||
f"Попробуйте позже.\n\n"
|
||||
f"━━━━━━━━━━━━━━━━━━━━━━━━━━━━\n\n"
|
||||
f"📊 <b>Статус:</b> Бесплатный доступ\n"
|
||||
f"📈 <b>Использовано вопросов:</b> {user.questions_used}/{settings.FREE_QUESTIONS_LIMIT}\n"
|
||||
f"🎯 <b>Осталось бесплатных:</b> {remaining}\n\n"
|
||||
f"💎 <i>Для безлимитного доступа: /buy</i>"
|
||||
)
|
||||
|
||||
await message.answer(response, parse_mode="HTML")
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user