chatgraph 0.6.0__tar.gz → 0.6.2__tar.gz

{chatgraph-0.6.0 → chatgraph-0.6.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: chatgraph
-Version: 0.6.0
+Version: 0.6.2
 Summary: A user-friendly chatbot library
 Home-page: https://github.com/irissonnlima/chatgraph
 License: MIT

{chatgraph-0.6.0 → chatgraph-0.6.2}/chatgraph/__init__.py

@@ -12,7 +12,7 @@ from .types.end_types import (
 from .types.message_types import Message, Button
 from .types.route import Route
 from .types.background_task import BackgroundTask
-from .types.image import ImageData, SendImage
+from .types.image import ImageData, ImageMessage
 
 __all__ = [
     "ChatbotApp",
@@ -32,5 +32,5 @@ __all__ = [
     "Button",
     "BackgroundTask",
     "ImageData",
-    "SendImage",
+    "ImageMessage",
 ]

{chatgraph-0.6.0 → chatgraph-0.6.2}/chatgraph/bot/chatbot_model.py

@@ -2,6 +2,7 @@ import inspect
 from functools import wraps
 from logging import debug, error
 import asyncio
+import re
 
 from ..error.chatbot_error import ChatbotMessageError
 from ..messages.message_consumer import MessageConsumer
@@ -16,6 +17,11 @@ from ..types.end_types import (
 from ..types.route import Route
 from .chatbot_router import ChatbotRouter
 from ..types.background_task import BackgroundTask
+from .default_functions import voltar
+
+DEFAULT_FUNCTION: dict[str, callable] = {
+    r"^\s*(voltar)\s*$": voltar,
+}
 
 
 class ChatbotApp:
@@ -23,16 +29,22 @@ class ChatbotApp:
     Classe principal para a aplicação do chatbot, gerencia as rotas e a lógica de processamento de mensagens.
     """
 
-    def __init__(self, message_consumer: MessageConsumer = None):
+    def __init__(
+        self,
+        message_consumer: MessageConsumer = None,
+        default_functions: dict[str, callable] = DEFAULT_FUNCTION,
+    ):
         """
         Inicializa a classe ChatbotApp com um estado de usuário e um consumidor de mensagens.
 
         Args:
             message_consumer (MessageConsumer): O consumidor de mensagens que lida com a entrada de mensagens no sistema.
+            default_functions (dict[str, callable]): Dicionário de funções padrão que podem ser usadas antes das rotas.
         """
         if not message_consumer:
             message_consumer = MessageConsumer.load_dotenv()
 
+        self.default_functions = default_functions
         self.__message_consumer = message_consumer
         self.__routes = {}
 
@@ -106,7 +118,20 @@ class ChatbotApp:
         route = userCall.route.lower()
         route_handler = route.split(".")[-1]
 
-        handler = self.__routes.get(route_handler, None)
+        matchDefault = False
+
+        for regex, func in self.default_functions.items():
+            if re.match(regex, userCall.content_message):
+                matchDefault = True
+                debug(f"Função padrão encontrada: {func.__name__} para a rota {route}")
+                handler = {
+                    "function": func,
+                    "params": {UserCall: "userCall", Route: "route"},
+                }
+                break
+
+        if not matchDefault:
+            handler = self.__routes.get(route_handler, None)
 
         if not handler:
             raise ChatbotMessageError(user_id, f"Rota não encontrada para {route}!")
@@ -127,6 +152,9 @@ class ChatbotApp:
             loop = asyncio.get_running_loop()
             userCall_response = await loop.run_in_executor(None, lambda: func(**kwargs))
 
+        if matchDefault:
+            userCall.content_message = ""
+
         if isinstance(userCall_response, (list, tuple)):
             for response in userCall_response:
                 await self.__process_func_response(response, userCall, route=route)
@@ -166,6 +194,7 @@ class ChatbotApp:
                 userCall.end_chat,
                 userCall_response.observations,
                 userCall_response.tabulation_id,
+                userCall_response.tabulation_name,
             )
             return
 
@@ -175,6 +204,7 @@ class ChatbotApp:
                 userCall.transfer_to_human,
                 userCall_response.observations,
                 userCall_response.campaign_id,
+                userCall_response.campaign_name,
             )
             return
 

chatgraph-0.6.2/chatgraph/bot/default_functions.py

@@ -0,0 +1,24 @@
+from ..types.request_types import UserCall
+from ..types.message_types import Message, Button
+from ..types.end_types import (
+    RedirectResponse,
+    EndChatResponse,
+    TransferToHuman,
+    TransferToMenu,
+)
+from ..types.route import Route
+
+
+async def voltar(route: Route, userCall: UserCall) -> tuple:
+    """
+    Função para voltar à rota anterior.
+    Args:
+        route (Route): A rota atual do chatbot.
+        usercall (UserCall): O objeto UserCall associado à chamada do usuário.
+    """
+
+    previous = route.get_previous()
+    userCall.console.print(
+        f"Voltando rota. ({route.current}) -> ({previous.current})", style="bold yellow"
+    )
+    return RedirectResponse(previous.current_node)

{chatgraph-0.6.0 → chatgraph-0.6.2}/chatgraph/gRPC/gRPCCall.py

@@ -1,6 +1,7 @@
 import os
 import grpc
 import json
+from rich.console import Console
 
 import chatgraph.pb.router_pb2 as chatbot_pb2
 import chatgraph.pb.router_pb2_grpc as chatbot_pb2_grpc
@@ -22,15 +23,22 @@ class RouterServiceClient:
         self.transfer_stub = chatbot_pb2_grpc.TransferStub(self.channel)
         self.end_chat_stub = chatbot_pb2_grpc.EndChatStub(self.channel)
 
+        self.console = Console()
+
     def insert_update_user_state(self, user_state_data):
         request = chatbot_pb2.UserState(**user_state_data)
         try:
             response = self.user_state_stub.InsertUpdateUserState(request)
             if not response.status:
-                print(f"Erro ao chamar SendMessage: {response.message}")
+                self.console.print(
+                    f"Erro ao chamar InsertUpdateUserState: {response.message}",
+                    style="bold red",
+                )
             return response
         except grpc.RpcError as e:
-            print(f"Erro ao chamar InsertUpdateUserState: {e}")
+            self.console.print(
+                f"Erro ao chamar InsertUpdateUserState: {e}", style="bold red"
+            )
             return None
 
     def delete_user_state(self, chat_id_data):
@@ -38,10 +46,12 @@ class RouterServiceClient:
         try:
             response = self.user_state_stub.DeleteUserState(request)
             if not response.status:
-                print(f"Erro ao chamar SendMessage: {response.message}")
+                self.console.print(
+                    f"Erro ao chamar SendMessage: {response.message}", style="bold red"
+                )
             return response
         except grpc.RpcError as e:
-            print(f"Erro ao chamar DeleteUserState: {e}")
+            self.console.print(f"Erro ao chamar DeleteUserState: {e}", style="bold red")
             return None
 
     def get_user_state(self, chat_id_data):
@@ -50,7 +60,7 @@ class RouterServiceClient:
             response = self.user_state_stub.GetUserState(request)
             return response
         except grpc.RpcError as e:
-            print(f"Erro ao chamar GetUserState: {e}")
+            self.console.print(f"Erro ao chamar GetUserState: {e}", style="bold red")
             return None
 
     def send_message(self, message_data):
@@ -61,10 +71,12 @@ class RouterServiceClient:
         try:
             response = self.send_message_stub.SendMessage(request)
             if not response.status:
-                print(f"Erro ao chamar SendMessage: {response.message}")
+                self.console.print(
+                    f"Erro ao chamar SendMessage: {response.message}", style="bold red"
+                )
             return response
         except grpc.RpcError as e:
-            print(f"Erro ao chamar SendMessage: {e}")
+            self.console.print(f"Erro ao chamar SendMessage: {e}", style="bold red")
             return None
 
     def send_image(self, message_data):
@@ -74,11 +86,28 @@ class RouterServiceClient:
 
         try:
             response = self.send_message_stub.SendImage(request)
+            if not response.status and response.message != "arquivo não encontrado":
+                self.console.print(
+                    f"Erro ao chamar SendImage: {response.message}", style="bold red"
+                )
+            elif response.message == "arquivo não encontrado":
+                print("Arquivo não encontrado, Carregando arquivo...")
+            return response
+        except grpc.RpcError as e:
+            self.console.print(f"Erro ao chamar SendImage: {e}", style="bold red")
+            return None
+
+    def upload_file(self, file_data):
+        request = chatbot_pb2.UploadFileRequest(**file_data)
+        try:
+            response = self.send_message_stub.UploadFile(request)
             if not response.status:
-                print(f"Erro ao chamar SendImage: {response.message}")
+                self.console.print(
+                    f"Erro ao chamar UploadFile: {response.message}", style="bold red"
+                )
             return response
         except grpc.RpcError as e:
-            print(f"Erro ao chamar SendImage: {e}")
+            self.console.print(f"Erro ao chamar UploadFile: {e}", style="bold red")
             return None
 
     def transfer_to_human(self, transfer_request_data):
@@ -86,10 +115,12 @@ class RouterServiceClient:
         try:
             response = self.transfer_stub.TransferToHuman(request)
             if not response.status:
-                print(f"Erro ao chamar SendMessage: {response.message}")
+                self.console.print(
+                    f"Erro ao chamar SendMessage: {response.message}", style="bold red"
+                )
             return response
         except grpc.RpcError as e:
-            print(f"Erro ao chamar TransferToHuman: {e}")
+            self.console.print(f"Erro ao chamar TransferToHuman: {e}", style="bold red")
             return None
 
     def transfer_to_menu(self, transfer_request_data):
@@ -97,10 +128,13 @@ class RouterServiceClient:
         try:
             response = self.transfer_stub.TransferToMenu(request)
             if not response.status:
-                print(f"Erro ao chamar TransferToMenu: {response.message}")
+                self.console.print(
+                    f"Erro ao chamar TransferToMenu: {response.message}",
+                    style="bold red",
+                )
             return response
         except grpc.RpcError as e:
-            print(f"Erro ao chamar TransferToMenu: {e}")
+            self.console.print(f"Erro ao chamar TransferToMenu: {e}", style="bold red")
             return None
 
     def end_chat(self, end_chat_request_data):
@@ -108,10 +142,12 @@ class RouterServiceClient:
         try:
             response = self.end_chat_stub.EndChat(request)
             if not response.status:
-                print(f"Erro ao chamar SendMessage: {response.message}")
+                self.console.print(
+                    f"Erro ao chamar SendMessage: {response.message}", style="bold red"
+                )
             return response
         except grpc.RpcError as e:
-            print(f"Erro ao chamar EndChat: {e}")
+            self.console.print(f"Erro ao chamar EndChat: {e}", style="bold red")
             return None
 
     def get_campaign_id(self, campaign_name):
@@ -120,7 +156,7 @@ class RouterServiceClient:
             response = self.transfer_stub.GetCampaignID(request)
             return response
         except grpc.RpcError as e:
-            print(f"Erro ao chamar GetCampaignID: {e}")
+            self.console.print(f"Erro ao chamar GetCampaignID: {e}", style="bold red")
             return None
 
     def get_all_campaigns(self):
@@ -129,7 +165,7 @@ class RouterServiceClient:
             response = self.transfer_stub.GetAllCampaigns(request)
             return response
         except grpc.RpcError as e:
-            print(f"Erro ao chamar GetAllCampaigns: {e}")
+            self.console.print(f"Erro ao chamar GetAllCampaigns: {e}", style="bold red")
             return None
 
     def get_tabulation_id(self, tabulation_name):
@@ -138,7 +174,7 @@ class RouterServiceClient:
             response = self.end_chat_stub.GetTabulationID(request)
             return response
         except grpc.RpcError as e:
-            print(f"Erro ao chamar GetTabulationID: {e}")
+            self.console.print(f"Erro ao chamar GetTabulationID: {e}", style="bold red")
             return None
 
     def get_all_tabulations(self):
@@ -147,5 +183,7 @@ class RouterServiceClient:
             response = self.end_chat_stub.GetAllTabulations(request)
             return response
         except grpc.RpcError as e:
-            print(f"Erro ao chamar GetAllTabulations: {e}")
+            self.console.print(
+                f"Erro ao chamar GetAllTabulations: {e}", style="bold red"
+            )
             return None

{chatgraph-0.6.0 → chatgraph-0.6.2}/chatgraph/pb/router.proto

@@ -7,6 +7,7 @@ option go_package = "./chatbot";
 ///// Serviços de Estado do Usuário /////
 service UserStateService {
     rpc InsertUpdateUserState(UserState) returns (RequestStatus);
+    rpc SetRoute(RouteRequest) returns (RequestStatus);
     rpc DeleteUserState(ChatID) returns (RequestStatus);
     rpc GetUserState(ChatID) returns (UserState);
     rpc GetAllUserStates(Void) returns (UserStateList);
@@ -60,6 +61,10 @@ message UserStateList {
     repeated UserState user_states = 1;
 }
 
+message RouteRequest {
+    ChatID chat_id = 1;
+    string route = 2;
+}
 ///// Mensagens para Serviços de Mensagens /////
 message TextMessage {
     string type = 1;
@@ -89,10 +94,11 @@ message FileMessage {
 }
 
 message UploadFileRequest {
-    string file_id = 1;
-    string file_url = 2;
-    string file_type = 3;
-    bytes file_content = 4;
+  string file_url = 1;
+  string file_type = 2;
+  string file_extension = 3;
+  string expiration = 4;
+  bytes file_content = 5;
 }
 
 ///// Mensagens para Serviços de Transfer /////
@@ -105,7 +111,8 @@ message TransferToHumanRequest{
 message TransferToMenuRequest{
     ChatID chat_id = 1;
     string menu = 2;
-    string user_message = 3;
+    string route = 3;
+    string user_message = 4;
 }
 
 message TabulationName{

chatgraph-0.6.2/chatgraph/pb/router_pb2.py

@@ -0,0 +1,79 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: router.proto
+# Protobuf Python Version: 6.31.0
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC, 6, 31, 0, "", "router.proto"
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
+    b'\n\x0crouter.proto\x12\x07\x63hatbot"\x06\n\x04Void"0\n\rRequestStatus\x12\x0e\n\x06status\x18\x01 \x01(\x08\x12\x0f\n\x07message\x18\x02 \x01(\t"-\n\x06\x43hatID\x12\x0f\n\x07user_id\x18\x01 \x01(\t\x12\x12\n\ncompany_id\x18\x02 \x01(\t"q\n\tUserState\x12 \n\x07\x63hat_id\x18\x01 \x01(\x0b\x32\x0f.chatbot.ChatID\x12\x0c\n\x04menu\x18\x02 \x01(\t\x12\r\n\x05route\x18\x03 \x01(\t\x12\x10\n\x08protocol\x18\x04 \x01(\t\x12\x13\n\x0bobservation\x18\x05 \x01(\t"8\n\rUserStateList\x12\'\n\x0buser_states\x18\x01 \x03(\x0b\x32\x12.chatbot.UserState"?\n\x0cRouteRequest\x12 \n\x07\x63hat_id\x18\x01 \x01(\x0b\x32\x0f.chatbot.ChatID\x12\r\n\x05route\x18\x02 \x01(\t"j\n\x0bTextMessage\x12\x0c\n\x04type\x18\x01 \x01(\t\x12\x0b\n\x03url\x18\x02 \x01(\t\x12\x10\n\x08\x66ilename\x18\x03 \x01(\t\x12\r\n\x05title\x18\x04 \x01(\t\x12\x0e\n\x06\x64\x65tail\x18\x05 \x01(\t\x12\x0f\n\x07\x63\x61ption\x18\x06 \x01(\t"5\n\x06\x42utton\x12\x0c\n\x04type\x18\x01 \x01(\t\x12\r\n\x05title\x18\x02 \x01(\t\x12\x0e\n\x06\x64\x65tail\x18\x03 \x01(\t"\x9d\x01\n\x07Message\x12 \n\x07\x63hat_id\x18\x01 \x01(\x0b\x32\x0f.chatbot.ChatID\x12%\n\x07message\x18\x02 \x01(\x0b\x32\x14.chatbot.TextMessage\x12 \n\x07\x62uttons\x18\x03 \x03(\x0b\x32\x0f.chatbot.Button\x12\'\n\x0e\x64isplay_button\x18\x04 \x01(\x0b\x32\x0f.chatbot.Button"A\n\x0b\x46ileMessage\x12!\n\x07message\x18\x01 \x01(\x0b\x32\x10.chatbot.Message\x12\x0f\n\x07\x66ile_id\x18\x02 \x01(\t"z\n\x11UploadFileRequest\x12\x10\n\x08\x66ile_url\x18\x01 \x01(\t\x12\x11\n\tfile_type\x18\x02 \x01(\t\x12\x16\n\x0e\x66ile_extension\x18\x03 \x01(\t\x12\x12\n\nexpiration\x18\x04 \x01(\t\x12\x14\n\x0c\x66ile_content\x18\x05 \x01(\x0c"d\n\x16TransferToHumanRequest\x12 \n\x07\x63hat_id\x18\x01 \x01(\x0b\x32\x0f.chatbot.ChatID\x12\x13\n\x0b\x63\x61mpaign_id\x18\x02 \x01(\t\x12\x13\n\x0bobservation\x18\x03 \x01(\t"l\n\x15TransferToMenuRequest\x12 \n\x07\x63hat_id\x18\x01 \x01(\x0b\x32\x0f.chatbot.ChatID\x12\x0c\n\x04menu\x18\x02 \x01(\t\x12\r\n\x05route\x18\x03 \x01(\t\x12\x14\n\x0cuser_message\x18\x04 \x01(\t"\x1e\n\x0eTabulationName\x12\x0c\n\x04name\x18\x01 \x01(\t"B\n\x11TabulationDetails\x12\n\n\x02id\x18\x01 \x01(\t\x12\x0c\n\x04name\x18\x02 \x01(\t\x12\x13\n\x0blast_update\x18\x03 \x01(\t"B\n\x0fTabulationsList\x12/\n\x0btabulations\x18\x01 \x03(\x0b\x32\x1a.chatbot.TabulationDetails"^\n\x0e\x45ndChatRequest\x12 \n\x07\x63hat_id\x18\x01 \x01(\x0b\x32\x0f.chatbot.ChatID\x12\x15\n\rtabulation_id\x18\x02 \x01(\t\x12\x13\n\x0bobservation\x18\x03 \x01(\t"\x1c\n\x0c\x43\x61mpaignName\x12\x0c\n\x04name\x18\x01 \x01(\t"@\n\x0f\x43\x61mpaignDetails\x12\n\n\x02id\x18\x01 \x01(\t\x12\x0c\n\x04name\x18\x02 \x01(\t\x12\x13\n\x0blast_update\x18\x03 \x01(\t"<\n\rCampaignsList\x12+\n\tcampaigns\x18\x01 \x03(\x0b\x32\x18.chatbot.CampaignDetails2\xbe\x02\n\x10UserStateService\x12\x43\n\x15InsertUpdateUserState\x12\x12.chatbot.UserState\x1a\x16.chatbot.RequestStatus\x12\x39\n\x08SetRoute\x12\x15.chatbot.RouteRequest\x1a\x16.chatbot.RequestStatus\x12:\n\x0f\x44\x65leteUserState\x12\x0f.chatbot.ChatID\x1a\x16.chatbot.RequestStatus\x12\x33\n\x0cGetUserState\x12\x0f.chatbot.ChatID\x1a\x12.chatbot.UserState\x12\x39\n\x10GetAllUserStates\x12\r.chatbot.Void\x1a\x16.chatbot.UserStateList2\xfd\x01\n\x0bSendMessage\x12\x37\n\x0bSendMessage\x12\x10.chatbot.Message\x1a\x16.chatbot.RequestStatus\x12\x39\n\tSendImage\x12\x14.chatbot.FileMessage\x1a\x16.chatbot.RequestStatus\x12\x38\n\x08SendFile\x12\x14.chatbot.FileMessage\x1a\x16.chatbot.RequestStatus\x12@\n\nUploadFile\x12\x1a.chatbot.UploadFileRequest\x1a\x16.chatbot.RequestStatus2\x9c\x02\n\x08Transfer\x12\x38\n\x0fGetAllCampaigns\x12\r.chatbot.Void\x1a\x16.chatbot.CampaignsList\x12@\n\rGetCampaignID\x12\x15.chatbot.CampaignName\x1a\x18.chatbot.CampaignDetails\x12J\n\x0fTransferToHuman\x12\x1f.chatbot.TransferToHumanRequest\x1a\x16.chatbot.RequestStatus\x12H\n\x0eTransferToMenu\x12\x1e.chatbot.TransferToMenuRequest\x1a\x16.chatbot.RequestStatus2\xcb\x01\n\x07\x45ndChat\x12<\n\x11GetAllTabulations\x12\r.chatbot.Void\x1a\x18.chatbot.TabulationsList\x12\x46\n\x0fGetTabulationID\x12\x17.chatbot.TabulationName\x1a\x1a.chatbot.TabulationDetails\x12:\n\x07\x45ndChat\x12\x17.chatbot.EndChatRequest\x1a\x16.chatbot.RequestStatusB\x0bZ\t./chatbotb\x06proto3'
+)
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, "router_pb2", _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+    _globals["DESCRIPTOR"]._loaded_options = None
+    _globals["DESCRIPTOR"]._serialized_options = b"Z\t./chatbot"
+    _globals["_VOID"]._serialized_start = 25
+    _globals["_VOID"]._serialized_end = 31
+    _globals["_REQUESTSTATUS"]._serialized_start = 33
+    _globals["_REQUESTSTATUS"]._serialized_end = 81
+    _globals["_CHATID"]._serialized_start = 83
+    _globals["_CHATID"]._serialized_end = 128
+    _globals["_USERSTATE"]._serialized_start = 130
+    _globals["_USERSTATE"]._serialized_end = 243
+    _globals["_USERSTATELIST"]._serialized_start = 245
+    _globals["_USERSTATELIST"]._serialized_end = 301
+    _globals["_ROUTEREQUEST"]._serialized_start = 303
+    _globals["_ROUTEREQUEST"]._serialized_end = 366
+    _globals["_TEXTMESSAGE"]._serialized_start = 368
+    _globals["_TEXTMESSAGE"]._serialized_end = 474
+    _globals["_BUTTON"]._serialized_start = 476
+    _globals["_BUTTON"]._serialized_end = 529
+    _globals["_MESSAGE"]._serialized_start = 532
+    _globals["_MESSAGE"]._serialized_end = 689
+    _globals["_FILEMESSAGE"]._serialized_start = 691
+    _globals["_FILEMESSAGE"]._serialized_end = 756
+    _globals["_UPLOADFILEREQUEST"]._serialized_start = 758
+    _globals["_UPLOADFILEREQUEST"]._serialized_end = 880
+    _globals["_TRANSFERTOHUMANREQUEST"]._serialized_start = 882
+    _globals["_TRANSFERTOHUMANREQUEST"]._serialized_end = 982
+    _globals["_TRANSFERTOMENUREQUEST"]._serialized_start = 984
+    _globals["_TRANSFERTOMENUREQUEST"]._serialized_end = 1092
+    _globals["_TABULATIONNAME"]._serialized_start = 1094
+    _globals["_TABULATIONNAME"]._serialized_end = 1124
+    _globals["_TABULATIONDETAILS"]._serialized_start = 1126
+    _globals["_TABULATIONDETAILS"]._serialized_end = 1192
+    _globals["_TABULATIONSLIST"]._serialized_start = 1194
+    _globals["_TABULATIONSLIST"]._serialized_end = 1260
+    _globals["_ENDCHATREQUEST"]._serialized_start = 1262
+    _globals["_ENDCHATREQUEST"]._serialized_end = 1356
+    _globals["_CAMPAIGNNAME"]._serialized_start = 1358
+    _globals["_CAMPAIGNNAME"]._serialized_end = 1386
+    _globals["_CAMPAIGNDETAILS"]._serialized_start = 1388
+    _globals["_CAMPAIGNDETAILS"]._serialized_end = 1452
+    _globals["_CAMPAIGNSLIST"]._serialized_start = 1454
+    _globals["_CAMPAIGNSLIST"]._serialized_end = 1514
+    _globals["_USERSTATESERVICE"]._serialized_start = 1517
+    _globals["_USERSTATESERVICE"]._serialized_end = 1835
+    _globals["_SENDMESSAGE"]._serialized_start = 1838
+    _globals["_SENDMESSAGE"]._serialized_end = 2091
+    _globals["_TRANSFER"]._serialized_start = 2094
+    _globals["_TRANSFER"]._serialized_end = 2378
+    _globals["_ENDCHAT"]._serialized_start = 2381
+    _globals["_ENDCHAT"]._serialized_end = 2584
+# @@protoc_insertion_point(module_scope)

{chatgraph-0.6.0 → chatgraph-0.6.2}/chatgraph/pb/router_pb2_grpc.py

@@ -43,6 +43,12 @@ class UserStateServiceStub(object):
             response_deserializer=router__pb2.RequestStatus.FromString,
             _registered_method=True,
         )
+        self.SetRoute = channel.unary_unary(
+            "/chatbot.UserStateService/SetRoute",
+            request_serializer=router__pb2.RouteRequest.SerializeToString,
+            response_deserializer=router__pb2.RequestStatus.FromString,
+            _registered_method=True,
+        )
         self.DeleteUserState = channel.unary_unary(
             "/chatbot.UserStateService/DeleteUserState",
             request_serializer=router__pb2.ChatID.SerializeToString,
@@ -72,6 +78,12 @@ class UserStateServiceServicer(object):
         context.set_details("Method not implemented!")
         raise NotImplementedError("Method not implemented!")
 
+    def SetRoute(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details("Method not implemented!")
+        raise NotImplementedError("Method not implemented!")
+
     def DeleteUserState(self, request, context):
         """Missing associated documentation comment in .proto file."""
         context.set_code(grpc.StatusCode.UNIMPLEMENTED)
@@ -98,6 +110,11 @@ def add_UserStateServiceServicer_to_server(servicer, server):
             request_deserializer=router__pb2.UserState.FromString,
             response_serializer=router__pb2.RequestStatus.SerializeToString,
         ),
+        "SetRoute": grpc.unary_unary_rpc_method_handler(
+            servicer.SetRoute,
+            request_deserializer=router__pb2.RouteRequest.FromString,
+            response_serializer=router__pb2.RequestStatus.SerializeToString,
+        ),
         "DeleteUserState": grpc.unary_unary_rpc_method_handler(
             servicer.DeleteUserState,
             request_deserializer=router__pb2.ChatID.FromString,
@@ -157,6 +174,36 @@ class UserStateService(object):
             _registered_method=True,
         )
 
+    @staticmethod
+    def SetRoute(
+        request,
+        target,
+        options=(),
+        channel_credentials=None,
+        call_credentials=None,
+        insecure=False,
+        compression=None,
+        wait_for_ready=None,
+        timeout=None,
+        metadata=None,
+    ):
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            "/chatbot.UserStateService/SetRoute",
+            router__pb2.RouteRequest.SerializeToString,
+            router__pb2.RequestStatus.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True,
+        )
+
     @staticmethod
     def DeleteUserState(
         request,

{chatgraph-0.6.0 → chatgraph-0.6.2}/chatgraph/types/end_types.py

@@ -25,11 +25,20 @@ 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,
+        tabulation_name: str | None = None,
+        observations: str | None = None,
+    ) -> None:
         """
         Finzaliza e tabula as informações do chatbot.
         """
+        if not tabulation_id and not tabulation_name:
+            raise ValueError("tabulation_id or tabulation_name must be provided.")
+
         self.tabulation_id = tabulation_id
+        self.tabulation_name = tabulation_name
         self.observations = observations
 
 
@@ -38,11 +47,17 @@ class TransferToHuman:
     Representa uma transferencia para um atendente humano.
     """
 
-    def __init__(self, campaign_id: str, observations: str) -> None:
+    def __init__(
+        self,
+        campaign_id: str,
+        campaign_name: str | None = None,
+        observations: str | None = None,
+    ) -> None:
         """
         Finzaliza e tabula as informações do chatbot.
         """
         self.campaign_id = campaign_id
+        self.campaign_name = campaign_name
         self.observations = observations
 
 

chatgraph-0.6.2/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-0.6.0 → chatgraph-0.6.2}/chatgraph/types/request_types.py

@@ -1,9 +1,10 @@
 from chatgraph.gRPC.gRPCCall import RouterServiceClient
-from chatgraph.types.image import ImageData, SendImage
+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
 import json, os
+from rich.console import Console
 
 
 class ChatID:
@@ -155,6 +156,7 @@ class UserCall:
         self.__grpc_uri = grpc_uri
 
         self.__router_client = RouterServiceClient(self.__grpc_uri)
+        self.console = Console()
 
     def __str__(self):
         return f"UserCall({self.type}, {self.__type_message}, {self.__content_message}, {self.__user_state})"
@@ -176,25 +178,37 @@ class UserCall:
             )
 
         if isinstance(message, ImageData):
-            message = SendImage(
+            message = ImageMessage(
                 image=message,
-                message=Message(type="message", detail=""),
             )
 
         if isinstance(message, Message):
             self.__send(message)
-        elif isinstance(message, SendImage):
+        elif isinstance(message, ImageMessage):
             self.__send_image(message)
         else:
             raise ValueError("Tipo de mensagem inválido.")
 
-    def __send_image(self, message: SendImage) -> None:
+    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:
+        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:
 
@@ -208,14 +222,20 @@ class UserCall:
         if not response.status:
             raise ValueError("Erro ao enviar mensagem de botões.")
 
-    def transfer_to_human(self, message: str, campaign_name: str) -> None:
+    def transfer_to_human(
+        self, message: str, campaign_id: str = None, campaign_name: str = None
+    ) -> None:
 
-        campaign = self.__router_client.get_campaign_id({"name": campaign_name})
+        if campaign_id is None and campaign_name is None:
+            raise ValueError("Você deve informar o ID ou o nome da campanha.")
+        if not campaign_id:
+            campaign = self.__router_client.get_campaign_id({"name": campaign_name})
+            campaign_id = campaign.id
 
         response = self.__router_client.transfer_to_human(
             {
                 "chat_id": self.__user_state.chatID.to_dict(),
-                "campaign_id": campaign.id,
+                "campaign_id": campaign_id,
                 "observation": message,
             }
         )
@@ -236,13 +256,19 @@ class UserCall:
         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})
+    def end_chat(self, message: str, tabulation_id: str, tabulation_name: str) -> None:
+        if tabulation_id is None and tabulation_name is None:
+            raise ValueError("Você deve informar o ID ou o nome da tabulação.")
+        if not tabulation_id:
+            tabulation = self.__router_client.get_tabulation_id(
+                {"name": tabulation_name}
+            )
+            tabulation_id = tabulation.id
 
         response = self.__router_client.end_chat(
             {
                 "chat_id": self.__user_state.chatID.to_dict(),
-                "tabulation_id": tabulation.id,
+                "tabulation_id": tabulation_id,
                 "observation": message,
             }
         )
@@ -328,3 +354,7 @@ class UserCall:
         self.update_user_state(
             self.__user_state.menu, self.__user_state.route, observation
         )
+
+    @content_message.setter
+    def content_message(self, content_message: str):
+        self.__content_message = content_message

{chatgraph-0.6.0 → chatgraph-0.6.2}/chatgraph/types/route.py

@@ -10,7 +10,7 @@ class Route:
         routes (list[str]): A lista de todas as rotas disponíveis no fluxo.
     """
 
-    def __init__(self, current: str, routes: list[str], separator:str='.'):
+    def __init__(self, current: str, routes: list[str], separator: str = "."):
         """
         Inicializa a rota com a rota atual e a lista de rotas disponíveis.
 
@@ -24,20 +24,20 @@ class Route:
         self.separator = separator
 
     @property
-    def previous(self)->'Route':
+    def previous(self) -> "Route":
         """
         Retorna a rota anterior a atual do usuário
         """
         return self.get_previous()
-    
+
     @property
-    def current_node(self)->str:
+    def current_node(self) -> str:
         """
         Retorna o nó atual do usuário
         """
         return self.current.split(self.separator)[-1]
-    
-    def get_previous(self) -> 'Route':
+
+    def get_previous(self) -> "Route":
         """
         Retorna o caminho anterior ao caminho atual.
 
@@ -47,13 +47,18 @@ class Route:
         Returns:
             str: O caminho anterior à rota atual.
         """
-        if self.current == 'start':
+        if self.current == "start":
             return Route(self.current, self.routes, self.separator)
 
-        previous_route = self.separator.join(self.current.split(self.separator)[:-1])
+        rotas_dedup = self.separator.join(
+            dict.fromkeys(self.current.split(self.separator))
+        )
+
+        previous_route = self.separator.join(rotas_dedup.split(self.separator)[:-1])
+
         return Route(previous_route, self.routes, self.separator)
 
-    def get_next(self, next_part: str) -> 'Route':
+    def get_next(self, next_part: str) -> "Route":
         """
         Monta e retorna o próximo caminho com base na parte fornecida.
 
@@ -69,8 +74,8 @@ class Route:
         next_part = next_part.strip().lower()
         next_route = f"{self.current.rstrip(self.separator)}.{next_part}"
         if next_part not in self.routes:
-            raise RouteError(f'Rota não encontrada: {next_part}')
-        
+            raise RouteError(f"Rota não encontrada: {next_part}")
+
         return Route(next_route, self.routes, self.separator)
 
     def __str__(self):
@@ -80,7 +85,7 @@ class Route:
         Returns:
             str: A representação em string da rota atual.
         """
-        return f'Route(current={self.current})'
+        return f"Route(current={self.current})"
 
     def __repr__(self):
         """

{chatgraph-0.6.0 → chatgraph-0.6.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "chatgraph"
-version = "0.6.0"
+version = "0.6.2"
 description = "A user-friendly chatbot library"
 authors = ["Irisson N. Lima <irisson.lima@verdecard.com.br>"]
 readme = "README.md"

chatgraph-0.6.0/chatgraph/pb/router_pb2.py

@@ -1,81 +0,0 @@
-# -*- coding: utf-8 -*-
-# Generated by the protocol buffer compiler.  DO NOT EDIT!
-# NO CHECKED-IN PROTOBUF GENCODE
-# source: router.proto
-# Protobuf Python Version: 6.31.0
-"""Generated protocol buffer code."""
-from google.protobuf import descriptor as _descriptor
-from google.protobuf import descriptor_pool as _descriptor_pool
-from google.protobuf import runtime_version as _runtime_version
-from google.protobuf import symbol_database as _symbol_database
-from google.protobuf.internal import builder as _builder
-_runtime_version.ValidateProtobufRuntimeVersion(
-    _runtime_version.Domain.PUBLIC,
-    6,
-    31,
-    0,
-    '',
-    'router.proto'
-)
-# @@protoc_insertion_point(imports)
-
-_sym_db = _symbol_database.Default()
-
-
-
-
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x0crouter.proto\x12\x07\x63hatbot\"\x06\n\x04Void\"0\n\rRequestStatus\x12\x0e\n\x06status\x18\x01 \x01(\x08\x12\x0f\n\x07message\x18\x02 \x01(\t\"-\n\x06\x43hatID\x12\x0f\n\x07user_id\x18\x01 \x01(\t\x12\x12\n\ncompany_id\x18\x02 \x01(\t\"q\n\tUserState\x12 \n\x07\x63hat_id\x18\x01 \x01(\x0b\x32\x0f.chatbot.ChatID\x12\x0c\n\x04menu\x18\x02 \x01(\t\x12\r\n\x05route\x18\x03 \x01(\t\x12\x10\n\x08protocol\x18\x04 \x01(\t\x12\x13\n\x0bobservation\x18\x05 \x01(\t\"8\n\rUserStateList\x12\'\n\x0buser_states\x18\x01 \x03(\x0b\x32\x12.chatbot.UserState\"j\n\x0bTextMessage\x12\x0c\n\x04type\x18\x01 \x01(\t\x12\x0b\n\x03url\x18\x02 \x01(\t\x12\x10\n\x08\x66ilename\x18\x03 \x01(\t\x12\r\n\x05title\x18\x04 \x01(\t\x12\x0e\n\x06\x64\x65tail\x18\x05 \x01(\t\x12\x0f\n\x07\x63\x61ption\x18\x06 \x01(\t\"5\n\x06\x42utton\x12\x0c\n\x04type\x18\x01 \x01(\t\x12\r\n\x05title\x18\x02 \x01(\t\x12\x0e\n\x06\x64\x65tail\x18\x03 \x01(\t\"\x9d\x01\n\x07Message\x12 \n\x07\x63hat_id\x18\x01 \x01(\x0b\x32\x0f.chatbot.ChatID\x12%\n\x07message\x18\x02 \x01(\x0b\x32\x14.chatbot.TextMessage\x12 \n\x07\x62uttons\x18\x03 \x03(\x0b\x32\x0f.chatbot.Button\x12\'\n\x0e\x64isplay_button\x18\x04 \x01(\x0b\x32\x0f.chatbot.Button\"A\n\x0b\x46ileMessage\x12!\n\x07message\x18\x01 \x01(\x0b\x32\x10.chatbot.Message\x12\x0f\n\x07\x66ile_id\x18\x02 \x01(\t\"_\n\x11UploadFileRequest\x12\x0f\n\x07\x66ile_id\x18\x01 \x01(\t\x12\x10\n\x08\x66ile_url\x18\x02 \x01(\t\x12\x11\n\tfile_type\x18\x03 \x01(\t\x12\x14\n\x0c\x66ile_content\x18\x04 \x01(\x0c\"d\n\x16TransferToHumanRequest\x12 \n\x07\x63hat_id\x18\x01 \x01(\x0b\x32\x0f.chatbot.ChatID\x12\x13\n\x0b\x63\x61mpaign_id\x18\x02 \x01(\t\x12\x13\n\x0bobservation\x18\x03 \x01(\t\"]\n\x15TransferToMenuRequest\x12 \n\x07\x63hat_id\x18\x01 \x01(\x0b\x32\x0f.chatbot.ChatID\x12\x0c\n\x04menu\x18\x02 \x01(\t\x12\x14\n\x0cuser_message\x18\x03 \x01(\t\"\x1e\n\x0eTabulationName\x12\x0c\n\x04name\x18\x01 \x01(\t\"B\n\x11TabulationDetails\x12\n\n\x02id\x18\x01 \x01(\t\x12\x0c\n\x04name\x18\x02 \x01(\t\x12\x13\n\x0blast_update\x18\x03 \x01(\t\"B\n\x0fTabulationsList\x12/\n\x0btabulations\x18\x01 \x03(\x0b\x32\x1a.chatbot.TabulationDetails\"^\n\x0e\x45ndChatRequest\x12 \n\x07\x63hat_id\x18\x01 \x01(\x0b\x32\x0f.chatbot.ChatID\x12\x15\n\rtabulation_id\x18\x02 \x01(\t\x12\x13\n\x0bobservation\x18\x03 \x01(\t\"\x1c\n\x0c\x43\x61mpaignName\x12\x0c\n\x04name\x18\x01 \x01(\t\"@\n\x0f\x43\x61mpaignDetails\x12\n\n\x02id\x18\x01 \x01(\t\x12\x0c\n\x04name\x18\x02 \x01(\t\x12\x13\n\x0blast_update\x18\x03 \x01(\t\"<\n\rCampaignsList\x12+\n\tcampaigns\x18\x01 \x03(\x0b\x32\x18.chatbot.CampaignDetails2\x83\x02\n\x10UserStateService\x12\x43\n\x15InsertUpdateUserState\x12\x12.chatbot.UserState\x1a\x16.chatbot.RequestStatus\x12:\n\x0f\x44\x65leteUserState\x12\x0f.chatbot.ChatID\x1a\x16.chatbot.RequestStatus\x12\x33\n\x0cGetUserState\x12\x0f.chatbot.ChatID\x1a\x12.chatbot.UserState\x12\x39\n\x10GetAllUserStates\x12\r.chatbot.Void\x1a\x16.chatbot.UserStateList2\xfd\x01\n\x0bSendMessage\x12\x37\n\x0bSendMessage\x12\x10.chatbot.Message\x1a\x16.chatbot.RequestStatus\x12\x39\n\tSendImage\x12\x14.chatbot.FileMessage\x1a\x16.chatbot.RequestStatus\x12\x38\n\x08SendFile\x12\x14.chatbot.FileMessage\x1a\x16.chatbot.RequestStatus\x12@\n\nUploadFile\x12\x1a.chatbot.UploadFileRequest\x1a\x16.chatbot.RequestStatus2\x9c\x02\n\x08Transfer\x12\x38\n\x0fGetAllCampaigns\x12\r.chatbot.Void\x1a\x16.chatbot.CampaignsList\x12@\n\rGetCampaignID\x12\x15.chatbot.CampaignName\x1a\x18.chatbot.CampaignDetails\x12J\n\x0fTransferToHuman\x12\x1f.chatbot.TransferToHumanRequest\x1a\x16.chatbot.RequestStatus\x12H\n\x0eTransferToMenu\x12\x1e.chatbot.TransferToMenuRequest\x1a\x16.chatbot.RequestStatus2\xcb\x01\n\x07\x45ndChat\x12<\n\x11GetAllTabulations\x12\r.chatbot.Void\x1a\x18.chatbot.TabulationsList\x12\x46\n\x0fGetTabulationID\x12\x17.chatbot.TabulationName\x1a\x1a.chatbot.TabulationDetails\x12:\n\x07\x45ndChat\x12\x17.chatbot.EndChatRequest\x1a\x16.chatbot.RequestStatusB\x0bZ\t./chatbotb\x06proto3')
-
-_globals = globals()
-_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
-_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'router_pb2', _globals)
-if not _descriptor._USE_C_DESCRIPTORS:
-  _globals['DESCRIPTOR']._loaded_options = None
-  _globals['DESCRIPTOR']._serialized_options = b'Z\t./chatbot'
-  _globals['_VOID']._serialized_start=25
-  _globals['_VOID']._serialized_end=31
-  _globals['_REQUESTSTATUS']._serialized_start=33
-  _globals['_REQUESTSTATUS']._serialized_end=81
-  _globals['_CHATID']._serialized_start=83
-  _globals['_CHATID']._serialized_end=128
-  _globals['_USERSTATE']._serialized_start=130
-  _globals['_USERSTATE']._serialized_end=243
-  _globals['_USERSTATELIST']._serialized_start=245
-  _globals['_USERSTATELIST']._serialized_end=301
-  _globals['_TEXTMESSAGE']._serialized_start=303
-  _globals['_TEXTMESSAGE']._serialized_end=409
-  _globals['_BUTTON']._serialized_start=411
-  _globals['_BUTTON']._serialized_end=464
-  _globals['_MESSAGE']._serialized_start=467
-  _globals['_MESSAGE']._serialized_end=624
-  _globals['_FILEMESSAGE']._serialized_start=626
-  _globals['_FILEMESSAGE']._serialized_end=691
-  _globals['_UPLOADFILEREQUEST']._serialized_start=693
-  _globals['_UPLOADFILEREQUEST']._serialized_end=788
-  _globals['_TRANSFERTOHUMANREQUEST']._serialized_start=790
-  _globals['_TRANSFERTOHUMANREQUEST']._serialized_end=890
-  _globals['_TRANSFERTOMENUREQUEST']._serialized_start=892
-  _globals['_TRANSFERTOMENUREQUEST']._serialized_end=985
-  _globals['_TABULATIONNAME']._serialized_start=987
-  _globals['_TABULATIONNAME']._serialized_end=1017
-  _globals['_TABULATIONDETAILS']._serialized_start=1019
-  _globals['_TABULATIONDETAILS']._serialized_end=1085
-  _globals['_TABULATIONSLIST']._serialized_start=1087
-  _globals['_TABULATIONSLIST']._serialized_end=1153
-  _globals['_ENDCHATREQUEST']._serialized_start=1155
-  _globals['_ENDCHATREQUEST']._serialized_end=1249
-  _globals['_CAMPAIGNNAME']._serialized_start=1251
-  _globals['_CAMPAIGNNAME']._serialized_end=1279
-  _globals['_CAMPAIGNDETAILS']._serialized_start=1281
-  _globals['_CAMPAIGNDETAILS']._serialized_end=1345
-  _globals['_CAMPAIGNSLIST']._serialized_start=1347
-  _globals['_CAMPAIGNSLIST']._serialized_end=1407
-  _globals['_USERSTATESERVICE']._serialized_start=1410
-  _globals['_USERSTATESERVICE']._serialized_end=1669
-  _globals['_SENDMESSAGE']._serialized_start=1672
-  _globals['_SENDMESSAGE']._serialized_end=1925
-  _globals['_TRANSFER']._serialized_start=1928
-  _globals['_TRANSFER']._serialized_end=2212
-  _globals['_ENDCHAT']._serialized_start=2215
-  _globals['_ENDCHAT']._serialized_end=2418
-# @@protoc_insertion_point(module_scope)

chatgraph-0.6.0/chatgraph/types/image.py

@@ -1,36 +0,0 @@
-import hashlib
-import base64
-from chatgraph.types.message_types import Message
-
-
-class ImageData:
-    def __init__(self, url: str, image_bytes: bytes = None):
-        if not url and not image_bytes:
-            raise ValueError("URL or image bytes must be provided.")
-
-        self.url = url
-        self.image_bytes = image_bytes
-
-        # Create hash of image bytes if available, otherwise use URL
-        hash_input = image_bytes if 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"
-        )
-
-
-class SendImage:
-    def __init__(self, image: ImageData, message: Message):
-        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):
-        return {
-            "file_id": self.image.image_id,
-            "message": self.message.to_dict(),
-        }