Adding login and feedback to agent

interface/app.py

@@ -33,8 +33,8 @@ def display_page_message(message):
 
 
 def main():
-    app = Chatbot()
     login = Login()
+    app = Chatbot()
 
     if not st.session_state.user_authorized:
         login.mount_login_page()

interface/chatbot.py

@@ -11,7 +11,8 @@ from interface.auxiliary_functions import (
     extract_content,
     extract_links,
     fetch_webpage_content,
-    url_to_suggestion
+    url_to_suggestion,
+    connect_to_services
 )
 
 class Chatbot:
@@ -25,6 +26,24 @@ class Chatbot:
                 "links": []
             }]
         
+        if "chat" not in st.session_state and st.session_state.get('username'):
+            chat = st.session_state.services["edgedb_client"].query('''
+                Select Chat {
+                    id
+                } filter .user.username = <str>$username and .prototipo.id = <uuid>$prototipo_id
+                ''',
+                username=st.session_state.username,
+                prototipo_id=st.session_state.prototipo_id
+            )
+
+            if not chat:
+                chat = self._create_default_chat()
+            else:
+                chat = chat[0]
+                self.update_messages(chat)
+            
+            st.session_state.chat = chat
+        
         # Conteúdo dos botões do sidebar
         if "topics" not in st.session_state:
             st.session_state["topics"] = [
@@ -40,6 +59,43 @@ class Chatbot:
         # Pergunta do usuário no chatbot
         self.response = ""
         self.links = []
+    
+    def _create_default_chat(self):
+        chat = st.session_state.services["edgedb_client"].query('''
+                    INSERT Chat {
+                        user := (SELECT User FILTER .username = <str>$username),
+                        prototipo := (SELECT Prototipo FILTER .id = <uuid>$prototipo_id)
+                    }
+                    ''',
+                    username=st.session_state.username,
+                    prototipo_id=st.session_state.prototipo_id
+                )
+        message = st.session_state.services["edgedb_client"].query('''
+            SELECT (
+                INSERT Message {
+                    content := "Como eu posso ajudar?",
+                    role := "assistant",
+                    chat := (SELECT Chat FILTER .id = <uuid>$chat_id)
+                }
+            ) {
+                content,
+                role                                             
+            }
+        ''', chat_id=chat[0].id)
+        # st.session_state.chat_history = message
+        return chat[0]
+    
+    def update_messages(self, chat):
+        messages = st.session_state.services["edgedb_client"].query('''
+            SELECT Message {
+                id,
+                content,
+                role
+            } 
+            FILTER .chat.id = <uuid>$chat_id
+            ORDER BY .created_at ASC;
+        ''', chat_id=chat.id)
+        # st.session_state.chat_history = messages
 
     def create_sidebar(self):
         """Cria o sidebar com tópicos e opções."""
@@ -134,6 +190,20 @@ class Chatbot:
             all_links = extracted_links + original_links
             self.links.extend(all_links)
 
+    def _save_msg_to_db(self, msg, role):
+        message = st.session_state.services["edgedb_client"].query('''
+            SELECT (
+                INSERT Message {
+                    content := <str>$content,
+                    chat := (SELECT Chat FILTER .id = <uuid>$chat_id),
+                    role := <str>$role
+                }
+            ) {
+                id
+            }
+        ''', content=msg, chat_id=st.session_state.chat.id, role=role)
+        return message
+
     def display_feedback(self):
         user_input = st.session_state.chat_history[-2]["content"]
         bot_output = st.session_state.chat_history[-1]["content"]
@@ -151,23 +221,27 @@ class Chatbot:
             if rate is not None:
                 if st.button("Enviar Avaliação"):
                     try:
+                        msg = self._save_msg_to_db(bot_output, "assistant")
+
                         feedback_rate = rate + 1
                         # TODO Colocar nessa parte a estrutura para adicionar o feedback_data ao banco de dados
-                        # st.session_state.services["edgedb_client"].query('''
-                        #     INSERT Feedback {
-                        #         rating := <int16>$rating,
-                        #         content := <str>$content,
-                        #         message := <str>$content
-                        #     }
-                        # ''',
-                        # message_id=bot_output,
-                        # rating=feedback_rate,
-                        # content=text_feedback,
-                        # )
+                        st.session_state.services["edgedb_client"].query('''
+                            INSERT Feedback {
+                                rating := <int16>$rating,
+                                content := <str>$content,
+                                message := (SELECT Message FILTER .id = <uuid>$message_id),
+                                prototipo := (SELECT Prototipo FILTER .id = <uuid>$prototipo_id)
+                            }
+                        ''',
+                        message_id=msg[-1].id,
+                        rating=feedback_rate,
+                        content=text_feedback,
+                        prototipo_id=st.session_state.prototipo_id
+                        )
                         # st.session_state.chat_history
                         st.success(f"Avaliação enviada!")
                     except Exception as e:
                         print(e)
                         st.error("Erro ao enviar avaliação!")
 
-                    st.session_state.is_feedback_active = False # Desativando o feedback
+                    st.session_state.is_feedback_active = False # Desativando o feedback

interface/login.py

@@ -11,12 +11,6 @@ from auxiliary_functions import connect_to_services
 
 class Login:
     def __init__(self):
-        if "username" not in st.session_state:
-            st.session_state.username = ""
-        
-        if "user_authorized" not in st.session_state:
-            st.session_state.user_authorized = False
-        
         if "services" not in st.session_state:
             oa_client, qdrant_client, edgedb_client =  connect_to_services()
             st.session_state.services = {
@@ -24,7 +18,20 @@ class Login:
                 "qdrant_client": qdrant_client, 
                 "edgedb_client": edgedb_client
             }
+        
+        if "prototipo" not in st.session_state:
+            prototipo = st.session_state.services["edgedb_client"].query('''
+                Select Prototipo {
+                    id
+                } filter .name = <str>'agente'
+            ''')
+            st.session_state.prototipo_id = prototipo[0].id
 
+        if "username" not in st.session_state:
+            st.session_state.username = ""
+        
+        if "user_authorized" not in st.session_state:
+            st.session_state.user_authorized = False
 
         self.username = ""
         self.password = ""
@@ -36,22 +43,35 @@ class Login:
         with st.form("login_form"):
             st.title("Login")
 
-            username = st.text_input('E-mail:')
+            username = st.text_input('Nome de usuário:')
             password = st.text_input('Senha:', type='password')
             submitted = st.form_submit_button("Entrar")
-            st.text("Não possui uma conta?")
+
+            # Instruções de criação de conta
+            st.subheader("Não tem uma conta?")
+            st.write(
+                "Crie uma conta utilizando o seguinte formato para o nome de usuário: "
+                "@[PrimeiroNome][ÚltimoNome] - Certifique-se de usar um identificador para cada protótipo."
+            )
+            
+            # Avisos sobre a criação de contas
+            st.warning(
+                "Lembre-se: você deverá criar **duas contas distintas**, uma para cada protótipo. "
+                "Para diferenciá-las, adicione um identificador no nome de usuário."
+            )
+            st.write("Exemplo: `@GabrielSilva-agente`")
             create_account = st.form_submit_button("Criar uma conta")
 
             if create_account:
                 if not username:
-                    st.warning("Digite um email para cadastrar a conta.")
+                    st.warning("Digite um nome de usuário para cadastrar a conta.")
                 
                 if not password:
                     st.warning("Digite uma senha para cadastrar a conta.")
                 
                 if username and password:
-                    if "@" not in username:
-                        st.warning("Digite um email válido para cadastrar a conta.")
+                    if username[0] != "@":
+                        st.warning("Acrescente um '@' no início do nome de usuário para validar a conta.")
                     
                     else:
                         user = self.validate_credentials(username, password, return_just_user=True)
@@ -65,11 +85,13 @@ class Login:
                             st.session_state.services["edgedb_client"].query('''
                                 INSERT User {
                                     username := <str>$username,
-                                    password := <str>$password
+                                    password := <str>$password,
+                                    prototipo := (SELECT Prototipo FILTER .id = <uuid>$prototipo_id)
                                 }
                             ''',
                                 username=username, \
                                 password=hashed_password_str, \
+                                prototipo_id=st.session_state.prototipo_id
                             )
 
                             st.success("Conta cadastrada com sucesso.")