Spaces:
Sleeping
Sleeping
| import re | |
| from flask import Blueprint, render_template, request | |
| from services.genai_service import configure_genai | |
| from logs import get_user_history, save_log | |
| chat_bp = Blueprint('chat', __name__) | |
| model = configure_genai() | |
| def chat(user_id): | |
| """ | |
| Rota para interação do usuário com a IA. | |
| """ | |
| history = get_user_history(user_id) | |
| if not history: | |
| return ''' | |
| <html> | |
| <head> | |
| <meta http-equiv="refresh" content="2;url=/" /> | |
| </head> | |
| <body> | |
| <h2>Ops, algo deu errado por aqui.</h2> | |
| <p>Vou te levar de volta para a página inicial.</p> | |
| </body> | |
| </html> | |
| ''' | |
| response = None | |
| if request.method == 'POST': | |
| question = request.form.get('question') | |
| session = model.start_chat(enable_automatic_function_calling=True) | |
| prompt = ( | |
| f"Você é uma especialista em livros e pode conversar livremente " | |
| f"sobre obras literárias, personagens, enredos, gêneros e autores." | |
| f" O nome do usuário é {history.get('name', f'Usuário {user_id}')}" | |
| f". As preferências dele são: " | |
| f"{', '.join(history.get('preferences', [])) or 'nenhuma'}. " | |
| f"Histórico: {history}. " | |
| f"Pergunta: {question} " | |
| f"Responda na mesma língua da pergunta, com até 500 caracteres. " | |
| f"Fale diretamente com o usuário." | |
| ) | |
| gemini_response = session.send_message(prompt) | |
| response = gemini_response.text | |
| save_log(user_id, history, response) | |
| formatted_response = None | |
| if response: | |
| formatted_response = response.replace('\n', '<br>') | |
| formatted_response = re.sub( | |
| r'(?<=<br>)\s*(Ficção|Não[- ]Ficção|Ciência|Fantasia)\s*:(?=<br>)', | |
| r'<b>\1:</b>', | |
| f"<br>{formatted_response}<br>" | |
| )[4:-4] | |
| formatted_response = formatted_response.replace('**', '') | |
| formatted_response = formatted_response.replace('* ', '• ') | |
| return render_template( | |
| 'chat.html', | |
| user_id=user_id, | |
| response=formatted_response, | |
| history=history | |
| ) | |