chatgraph 0.6.0__tar.gz → 0.6.1__tar.gz

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

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

{chatgraph-0.6.0 → chatgraph-0.6.1}/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.1}/chatgraph/gRPC/gRPCCall.py

@@ -74,13 +74,26 @@ class RouterServiceClient:
 
         try:
             response = self.send_message_stub.SendImage(request)
-            if not response.status:
+            if not response.status and response.message != "arquivo não encontrado":
                 print(f"Erro ao chamar SendImage: {response.message}")
+            elif response.message == "arquivo não encontrado":
+                print("Arquivo não encontrado, Carregando arquivo...")
             return response
         except grpc.RpcError as e:
             print(f"Erro ao chamar SendImage: {e}")
             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 UploadFile: {response.message}")
+            return response
+        except grpc.RpcError as e:
+            print(f"Erro ao chamar UploadFile: {e}")
+            return None
+
     def transfer_to_human(self, transfer_request_data):
         request = chatbot_pb2.TransferToHumanRequest(**transfer_request_data)
         try:

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

@@ -1,5 +1,5 @@
 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
@@ -176,25 +176,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:
 

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

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "chatgraph"
-version = "0.6.0"
+version = "0.6.1"
 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(),
-        }