FilesExpand file tree

 master

/

utils.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

76 lines (58 loc) · 2.71 KB

 master

/

utils.py

Copy path

Top

File metadata and controls

• Code
• Blame

76 lines (58 loc) · 2.71 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

from clarifai_grpc.channel.clarifai_channel import ClarifaiChannel

from clarifai_grpc.grpc.api import service_pb2_grpc, service_pb2, resources_pb2

from clarifai_grpc.grpc.api.status import status_code_pb2

from random import randint

from telegram import ReplyKeyboardMarkup, KeyboardButton, InlineKeyboardButton, InlineKeyboardMarkup

import settings

def get_bot_number(user_number):

return randint(user_number - 10, user_number + 10)

def play_random_numbers(user_number, bot_number):

if user_number > bot_number:

message = f"Ваше число {user_number}, мое {bot_number}, вы выиграли"

elif user_number == bot_number:

message = f"Ваше число {user_number}, мое {bot_number}, ничья"

else:

message = f"Ваше число {user_number}, мое {bot_number}, вы проиграли"

return message

def main_keyboard():

return ReplyKeyboardMarkup([

['Прислать котика', KeyboardButton('Мои координаты', request_location=True), 'Заполнить анкету']

])

def has_object_on_image(file_name, object_name):

channel = ClarifaiChannel.get_grpc_channel()

app = service_pb2_grpc.V2Stub(channel)

metadata = (('authorization', f'Key {settings.CLARIFAI_API_KEY}'),)

with open(file_name, 'rb') as f:

file_data = f.read()

image = resources_pb2.Image(base64=file_data)

request = service_pb2.PostModelOutputsRequest(

model_id='aaa03c23b3724a16a56b629203edc62c',

inputs=[

resources_pb2.Input(

data=resources_pb2.Data(image=image)

)

])

response = app.PostModelOutputs(request, metadata=metadata)

# print(response)

return check_response_for_object(response, object_name)

def check_response_for_object(response, object_name):

if response.status.code == status_code_pb2.SUCCESS:

for concept in response.outputs[0].data.concepts:

if concept.name == object_name and concept.value >= 0.9:

return True

else:

print(f'Ошибка распознования картинки {response.outputs[0].status.details}')

return False

def cat_rating_inline_keyboard(image_name):

callback_text = f"rating|{image_name}|"

keyboard = [

[

InlineKeyboardButton('Нравится', callback_data=callback_text + '1'),

InlineKeyboardButton('Не нравится', callback_data=callback_text + '-1')

]

]

return InlineKeyboardMarkup(keyboard)

if __name__ == '__main__':

print(has_object_on_image('images/cat1.jpg', 'cat'))

print(has_object_on_image('images/not_cat.jpg', 'cat'))

print(has_object_on_image('images/not_cat2.jpg', 'dog'))