chatgraph 0.5.3__py3-none-any.whl → 0.6.1__py3-none-any.whl

chatgraph/types/end_types.py

@@ -1,4 +1,3 @@
-
 class RedirectResponse:
     """
     Representa uma resposta que redireciona o fluxo do chatbot para uma nova rota.
@@ -16,6 +15,7 @@ class RedirectResponse:
         """
         self.route = route
 
+
 class EndChatResponse:
     """
     Representa uma resposta que indica o fim do chatbot.
@@ -25,21 +25,35 @@ class EndChatResponse:
         observations (str): As observações finais do chatbot.
     """
 
-    def __init__(self, tabulation_id: str, observations:str) -> None:
-        '''
+    def __init__(self, tabulation_id: str, observations: str) -> None:
+        """
         Finzaliza e tabula as informações do chatbot.
-        '''
+        """
         self.tabulation_id = tabulation_id
         self.observations = observations
 
+
 class TransferToHuman:
     """
     Representa uma transferencia para um atendente humano.
     """
-    def __init__(self, campaign_id: str, observations:str) -> None:
-        '''
+
+    def __init__(self, campaign_id: str, observations: str) -> None:
+        """
         Finzaliza e tabula as informações do chatbot.
-        '''
+        """
         self.campaign_id = campaign_id
         self.observations = observations
-    
+
+
+class TransferToMenu:
+    """
+    Representa uma transferencia para outro Menu.
+    """
+
+    def __init__(self, menu: str, user_message: str) -> None:
+        """
+        Finzaliza e tabula as informações do chatbot.
+        """
+        self.menu = menu.lower()
+        self.user_message = user_message

chatgraph/types/image.py

@@ -0,0 +1,92 @@
+from datetime import datetime, timedelta
+import hashlib
+import base64
+from chatgraph.types.message_types import Message
+
+
+class ImageData:
+    """
+    Representa dados de uma imagem, seja por URL ou arquivo local. Caso fornecido um url e um caminho de arquivo, o url será utilizado.
+    Atributos:
+        type (str): Tipo de imagem, "link" para URL ou "file" para arquivo local.
+        url (str): URL da imagem, se aplicável.
+        image_path (str): Caminho do arquivo de imagem, se aplicável.
+        expiration (int): Tempo de expiração em minutos, 0 para sem expiração.
+    """
+
+    def __init__(self, *, url: str = None, image_path: str = None, expiration: int = 0):
+
+        if not url and not image_path:
+            raise ValueError("URL or image path must be provided.")
+
+        self.type = "link" if url else "file"
+        self.url = url
+        self.expiration = expiration
+
+        self.image_bytes = None
+        self.file_extension = None
+        if self.type == "file":
+            if not image_path:
+                raise ValueError("Image path must be provided for file type.")
+            self.file_extension = image_path.split(".")[-1]
+            with open(image_path, "rb") as f:
+                self.image_bytes = f.read()
+
+        # Create hash of image bytes if available, otherwise use URL
+        hash_input = self.image_bytes if self.image_bytes else url.encode("utf-8")
+
+        # Create SHA-256 hash and encode in base64
+        self.image_id = base64.b64encode(hashlib.sha256(hash_input).digest()).decode(
+            "utf-8"
+        )
+
+    def get_upload_dict(self):
+        dict_data = {}
+        if self.expiration > 0:
+            now = datetime.now()
+            expiration_time = now + timedelta(minutes=self.expiration)
+            dict_data["expiration"] = expiration_time.strftime("%Y-%m-%d %H:%M:%S")
+
+        if self.type == "file":
+            dict_data["file_type"] = self.type
+            dict_data["file_content"] = self.image_bytes
+            dict_data["file_extension"] = self.file_extension
+        else:
+            dict_data["file_type"] = self.type
+            dict_data["file_url"] = self.url
+
+        return dict_data
+
+    def get_dict(self):
+        dict_data = {
+            "image_id": self.image_id,
+            "file_type": self.type,
+        }
+        if self.type == "file":
+            dict_data["file_content"] = self.image_bytes
+            dict_data["file_extension"] = self.file_extension
+        else:
+            dict_data["file_url"] = self.url
+        return dict_data
+
+
+class ImageMessage:
+    def __init__(self, image: ImageData, message: Message = None):
+        if not isinstance(image, ImageData):
+            raise TypeError("Expected an instance of ImageData.")
+
+        self.type = "image"
+        self.image = image
+        self.message = message
+
+    def to_dict(self):
+        if not self.message:
+            return {
+                "file_id": self.image.image_id,
+                "message": {},
+            }
+
+        return {
+            "file_id": self.image.image_id,
+            "message": self.message.to_dict(),
+        }

chatgraph/types/request_types.py

@@ -1,4 +1,5 @@
 from chatgraph.gRPC.gRPCCall import RouterServiceClient
+from chatgraph.types.image import ImageData, ImageMessage
 from chatgraph.types.message_types import Message, Button, MessageTypes, messageTypes
 from typing import Optional
 from datetime import datetime
@@ -157,7 +158,7 @@ class UserCall:
 
     def __str__(self):
         return f"UserCall({self.type}, {self.__type_message}, {self.__content_message}, {self.__user_state})"
-    
+
     def send(
         self,
         message: messageTypes | Message,
@@ -174,11 +175,39 @@ class UserCall:
                 detail=str(message),
             )
 
+        if isinstance(message, ImageData):
+            message = ImageMessage(
+                image=message,
+            )
+
         if isinstance(message, Message):
             self.__send(message)
+        elif isinstance(message, ImageMessage):
+            self.__send_image(message)
         else:
             raise ValueError("Tipo de mensagem inválido.")
 
+    def __upload_file(self, image: ImageData) -> None:
+        dict_image = image.get_upload_dict()
+        response = self.__router_client.upload_file(dict_image)
+
+        if not response.status:
+            raise ValueError("Erro ao fazer upload do arquivo.")
+        else:
+            print("Arquivo enviado com sucesso.")
+
+    def __send_image(self, message: ImageMessage) -> None:
+        dict_message = message.to_dict()
+        dict_message["message"]["chat_id"] = self.__user_state.chatID.to_dict()
+        response = self.__router_client.send_image(dict_message)
+
+        if not response.status and response.message != "arquivo não encontrado":
+            raise ValueError("Erro ao enviar imagem.")
+        elif response.message == "arquivo não encontrado":
+            self.__upload_file(message.image)
+            print("tentando enviar imagem novamente...")
+            self.__send_image(message)
+
     def __send(self, message: Message) -> None:
 
         dict_message = message.to_dict()
@@ -192,26 +221,41 @@ class UserCall:
             raise ValueError("Erro ao enviar mensagem de botões.")
 
     def transfer_to_human(self, message: str, campaign_name: str) -> None:
-        
+
         campaign = self.__router_client.get_campaign_id({"name": campaign_name})
-        
+
         response = self.__router_client.transfer_to_human(
             {
                 "chat_id": self.__user_state.chatID.to_dict(),
                 "campaign_id": campaign.id,
+                "observation": message,
             }
         )
 
         if not response.status:
             raise ValueError("Erro ao transferir chat para humano.")
 
+    def transfer_to_menu(self, menu: str, message: str) -> None:
+
+        response = self.__router_client.transfer_to_menu(
+            {
+                "chat_id": self.__user_state.chatID.to_dict(),
+                "menu": menu,
+                "user_message": message,
+            }
+        )
+
+        if not response.status:
+            raise ValueError("Erro ao transferir chat para menu.")
+
     def end_chat(self, message: str, tabulation_name: str) -> None:
         tabulation = self.__router_client.get_tabulation_id({"name": tabulation_name})
-        
+
         response = self.__router_client.end_chat(
             {
                 "chat_id": self.__user_state.chatID.to_dict(),
                 "tabulation_id": tabulation.id,
+                "observation": message,
             }
         )
 
@@ -263,15 +307,15 @@ class UserCall:
     @property
     def protocol(self):
         return self.__user_state.protocol
-    
+
     @property
     def observation(self):
         return self.__user_state.observation
-    
+
     @property
     def type_message(self):
         return self.__type_message
-    
+
     @property
     def content_message(self):
         return self.__content_message
@@ -279,12 +323,16 @@ class UserCall:
     @menu.setter
     def menu(self, menu):
 
-        self.update_user_state(menu, self.__user_state.route, self.__user_state.observation)
+        self.update_user_state(
+            menu, self.__user_state.route, self.__user_state.observation
+        )
 
     @route.setter
     def route(self, route):
 
-        self.update_user_state(self.__user_state.menu, route, self.__user_state.observation)
+        self.update_user_state(
+            self.__user_state.menu, route, self.__user_state.observation
+        )
 
     @observation.setter
     def observation(self, observation):

{chatgraph-0.5.3.dist-info → chatgraph-0.6.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: chatgraph
-Version: 0.5.3
+Version: 0.6.1
 Summary: A user-friendly chatbot library
 Home-page: https://github.com/irissonnlima/chatgraph
 License: MIT
@@ -16,6 +16,7 @@ Requires-Dist: grpcio-tools (>=1.67.0,<2.0.0)
 Requires-Dist: matplotlib (>=3.10.0,<4.0.0)
 Requires-Dist: networkx (>=3.4.2,<4.0.0)
 Requires-Dist: pika (>=1.3.2,<2.0.0)
+Requires-Dist: protobuf (>=6.31.1,<7.0.0)
 Requires-Dist: python-dotenv (>=1.0.1,<2.0.0)
 Requires-Dist: rich (>=13.8.1,<14.0.0)
 Requires-Dist: typer (>=0.12.5,<0.13.0)

chatgraph-0.6.1.dist-info/RECORD

@@ -0,0 +1,23 @@
+chatgraph/__init__.py,sha256=cQk87BOUYKvOjgCkS5Vexy9EYeLOeuQn6vR1vXx1Akc,935
+chatgraph/auth/credentials.py,sha256=xsMEpLQnc66novPjL6upocMcnUnvFJ7yxINzUenkxmc,2388
+chatgraph/bot/chatbot_model.py,sha256=wLVXehmHEWItRwvl6J8HM96lC69FXmqsNG8KqplBboI,7373
+chatgraph/bot/chatbot_router.py,sha256=bVbm9dBoC2OesetXlQJpCrkaiLM3GUro0GrOyCSgreI,2586
+chatgraph/cli/__init__.py,sha256=tfgYhGoKy1nD4STN4xDh6J4VV55RICk7v1GZmzAg-bM,7413
+chatgraph/error/chatbot_error.py,sha256=4sEcW8vz0eQk2QDmDygucVM4caHliZW5iH7XJvmLBuk,1897
+chatgraph/error/route_error.py,sha256=CY8m82ap7-Sr-DXPsolltRW50TqD__5RyXBmNNJCWr8,793
+chatgraph/gRPC/gRPCCall.py,sha256=KXLhxfX7Mrm48BVuWhTA4xw4T4vghXnljvE37FAxmfQ,6402
+chatgraph/messages/message_consumer.py,sha256=yHhfLYjtmXQYG44o7SNO6XUaeM4-jH8BpIpPHI-mt8Y,6116
+chatgraph/pb/router.proto,sha256=0FDhMe7GrabeQz7QRJN8PN2mcuRjA5YrGt6TC9GWBak,3488
+chatgraph/pb/router_pb2.py,sha256=eSo0ncYDSG3ryIfWNFzqmG3anx-VmQ2nGXBy1gPnXmQ,7962
+chatgraph/pb/router_pb2_grpc.py,sha256=iaQQsYPc8Tihuf4sWEI48DhvhtCYtDqFBHhwRtE57ps,29886
+chatgraph/types/background_task.py,sha256=j-Bpd1l_LCDiWukhfG8wNr1F0_GUaOL_J2_BCUnTof4,928
+chatgraph/types/end_types.py,sha256=ulVq3XybRs-EGcOxw6e1FF6U2p3nbtPfOv3uRe6oJfo,1638
+chatgraph/types/image.py,sha256=-Lf9ns8_M4jmROoFnHlZiYeScYD_GhF1pgIGLVjsRtI,3216
+chatgraph/types/message_types.py,sha256=__LOk09Xfvojum0xiq9n157_3QSSLNaDMc02iq8dGpk,3434
+chatgraph/types/request_types.py,sha256=pGpJhzk6LGSrREGsG2gC14SoS9DTpUO5w9MzmLUg7PU,10500
+chatgraph/types/route.py,sha256=FO5INy5UXgicuQ8FKEZKcPv6WS5YH10dPx2OaPu_0sE,2978
+chatgraph-0.6.1.dist-info/entry_points.txt,sha256=bO9_Q-PqE5vCNNo6ke_-3j2gHfKQMDGnKDTkNuCdBuA,48
+chatgraph-0.6.1.dist-info/LICENSE,sha256=rVJozpRzDlplOpvI8A1GvmfVS0ReYdZvMWc1j2jV0v8,1090
+chatgraph-0.6.1.dist-info/METADATA,sha256=avh_0Xu-Mfkhzr4b5v9mKZyJcd2CprrvUIMBnlWWr6g,12908
+chatgraph-0.6.1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+chatgraph-0.6.1.dist-info/RECORD,,

chatgraph-0.5.3.dist-info/RECORD

@@ -1,22 +0,0 @@
-chatgraph/__init__.py,sha256=QtS8l6UjQ7N9gPNruTaPKUTNP8-EvTYeiTBoiSKGp7E,781
-chatgraph/auth/credentials.py,sha256=xsMEpLQnc66novPjL6upocMcnUnvFJ7yxINzUenkxmc,2388
-chatgraph/bot/chatbot_model.py,sha256=PhWCDDTIfT92rwS-USciugVjXuCDKk44NJbiOfEhvRk,7051
-chatgraph/bot/chatbot_router.py,sha256=bVbm9dBoC2OesetXlQJpCrkaiLM3GUro0GrOyCSgreI,2586
-chatgraph/cli/__init__.py,sha256=tfgYhGoKy1nD4STN4xDh6J4VV55RICk7v1GZmzAg-bM,7413
-chatgraph/error/chatbot_error.py,sha256=4sEcW8vz0eQk2QDmDygucVM4caHliZW5iH7XJvmLBuk,1897
-chatgraph/error/route_error.py,sha256=CY8m82ap7-Sr-DXPsolltRW50TqD__5RyXBmNNJCWr8,793
-chatgraph/gRPC/gRPCCall.py,sha256=d8nobBb8yDSKl_eFd1ZUfozvpyDZSoul2HlBIO8Q288,5308
-chatgraph/messages/message_consumer.py,sha256=yHhfLYjtmXQYG44o7SNO6XUaeM4-jH8BpIpPHI-mt8Y,6116
-chatgraph/pb/router.proto,sha256=pw3hTwjdMN8-yJIAlKHNigVKYqM-w2nme9-2bCTBiJ0,2853
-chatgraph/pb/router_pb2.py,sha256=jw-Rg05NMwgOKrRsoubFFb82v_O80KgWY4tIqwUf2Aw,6571
-chatgraph/pb/router_pb2_grpc.py,sha256=KxhVxCMVD9tpuX7CyXqIkzZ4P3zfXeVT92Bt_Hi7mrE,24109
-chatgraph/types/background_task.py,sha256=j-Bpd1l_LCDiWukhfG8wNr1F0_GUaOL_J2_BCUnTof4,928
-chatgraph/types/end_types.py,sha256=--Ty2gM_y7J-yRAvZV26e4HMUpoguAMAhfOIS9-kQTk,1316
-chatgraph/types/message_types.py,sha256=__LOk09Xfvojum0xiq9n157_3QSSLNaDMc02iq8dGpk,3434
-chatgraph/types/request_types.py,sha256=BeUI2l9XZpBf5c8xyPvKI68WZ5u6n98cC8DmUfWKlQ0,8788
-chatgraph/types/route.py,sha256=FO5INy5UXgicuQ8FKEZKcPv6WS5YH10dPx2OaPu_0sE,2978
-chatgraph-0.5.3.dist-info/entry_points.txt,sha256=bO9_Q-PqE5vCNNo6ke_-3j2gHfKQMDGnKDTkNuCdBuA,48
-chatgraph-0.5.3.dist-info/LICENSE,sha256=rVJozpRzDlplOpvI8A1GvmfVS0ReYdZvMWc1j2jV0v8,1090
-chatgraph-0.5.3.dist-info/METADATA,sha256=X4TtQbT5Bcpbaes_XIrLjuoELAMuYU3y105R8F7Ilnw,12866
-chatgraph-0.5.3.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-chatgraph-0.5.3.dist-info/RECORD,,