Merge branch 'main' into text-edition

Reviewed-on: #3
Reviewed-by: gav <a@a.aa>

.gitattributes

@@ -0,0 +1,2 @@
+*.png filter=lfs diff=lfs merge=lfs -text
+*.jpg filter=lfs diff=lfs merge=lfs -text

.gitignore

@@ -2,3 +2,5 @@
 tmp
 .env
 docker-compose-dev.yml
+*.DS_Store*
+.idea/

locales.py

@@ -4,8 +4,8 @@ from vkwave.bots import Keyboard, ButtonColor
 MENU = """С помощью бота вы можете:
 ⚙ пройти тест и получить индивидуальный NFT-талисман, сгенерированный специально для вас нейросетью;
 ⚙ вступить в чат абитуриентов, студентов и преподавателей;
-⚙ пообщаться со студентом (можно выбрать, с кем);
-⚙ пообщаться с руководителем ИШ или преподавателем (можно выбрать, с кем);"""
+⚙ пообщаться со студентом (можно выбрать с кем);
+⚙ пообщаться с руководителем ИШ или преподавателем (можно выбрать с кем)."""
 MENU_KB = Keyboard()
 MENU_KB.add_text_button(text="Пройти тест", payload={"test": "-1"}, color=ButtonColor.POSITIVE)
 MENU_KB.add_link_button(text="Чат абитуриентов", link="https://vk.me/join/AJQ1dw97/SBEQYIyQdZfG69y")
@@ -102,9 +102,11 @@ FAV_THEME = "Какая школьная дисциплина нравится
 FAV_THEME_KB = Keyboard()
 FAV_THEME_KB.add_text_button(text="Математика", payload={"q": "Математика"}, color=ButtonColor.PRIMARY)
 FAV_THEME_KB.add_text_button(text="Русский/Литература", payload={"q": "русскийлитра"}, color=ButtonColor.PRIMARY)
-FAV_THEME_KB.add_text_button(text="Информатика/программирование", payload={"q": "Инфа"}, color=ButtonColor.PRIMARY)
 FAV_THEME_KB.add_row()
+FAV_THEME_KB.add_text_button(text="Информатика/программирование", payload={"q": "Инфа"}, color=ButtonColor.PRIMARY)
 FAV_THEME_KB.add_text_button(text="Физика", payload={"q": "Физика"}, color=ButtonColor.PRIMARY)
+FAV_THEME_KB.add_row()
+FAV_THEME_KB.add_text_button(text="История/обществознание", payload={"q": "Инфа"}, color=ButtonColor.PRIMARY)
 FAV_THEME_KB.add_text_button(text="Другой", payload={"q": "other"}, color=ButtonColor.PRIMARY)
 
 # 4
@@ -131,7 +133,7 @@ ZODIAC = "Кто вы по знаку зодиака?"
 # CONTACT = "Введите почту или ник в телеграмме, по которому мы сможем прислать вам результаты"
 
 # last
-LAST_MESSAGE = "Теперь у вас есть ваш персональный NFT-талисман! Он поможет вам при сдаче экзаменов и поступлении в желаемый ВУЗ 🌟"
+LAST_MESSAGE = "Теперь у вас есть ваш персональный NFT-талисман! Он поможет вам при сдаче экзаменов и поступлении в желаемый университет 🌟"
 LAST_MESSAGE_KB = Keyboard()
 LAST_MESSAGE_KB.add_text_button(text="Вернуться на главную", payload={}, color=ButtonColor.POSITIVE)
 

nft_svc/app.py

@@ -1,34 +1,14 @@
 import io
 import random
+import os
 
 from PIL import Image
 from flask import Flask, request, jsonify
 import matplotlib.image as mpimg
-import numpy as np
-import numba
-import pickle
 import base64
 
 app = Flask(__name__)
-
-
-@numba.njit
-def optimized_mandelbrot(n_rows, n_columns, iterations, cx, cy):
-    x_cor = np.linspace(-2, 2, n_rows)
-    y_cor = np.linspace(-2, 2, n_columns)
-    output = np.zeros((n_rows,n_columns))
-    c = cx + 1j * cy
-    for i in range(n_rows):
-        for j in range(n_columns):
-            z = x_cor[i] + y_cor[j] * 1j
-            count = 0
-            for k in range(iterations):
-                z = (z*z) + c
-                count += 1
-                if np.abs(z) > 4:
-                    break
-            output[i, j] = count
-    return output.T
+IMAGE_PATH = 'pics'
 
 
 def open_image_as_array(path):
@@ -40,19 +20,19 @@ def get_encoded_img(arr):
     img_byte_arr = io.BytesIO()
     img.save(img_byte_arr, format='PNG')
     encoded_img = base64.encodebytes(img_byte_arr.getvalue()).decode('ascii')
-    return encoded_img
+    return str(encoded_img)
 
 
 @app.route('/getImage', methods=['GET'])
 def get_image():
-    with open('data_100.pickle', 'rb') as f:
-        images = pickle.load(f)
-    buffered = io.BytesIO()
-    random.choice(images).save(buffered, format="JPEG")
-    img_str = base64.b64encode(buffered.getvalue())
+    # print(random.choice(os.listdir(IMAGE_PATH)))
+    image = mpimg.imread(os.path.join(IMAGE_PATH, random.choice(os.listdir(IMAGE_PATH))))
+    # buffered = io.BytesIO()
+    # image.save(buffered, format="JPEG")
+    # img_str = base64.b64encode(buffered.getvalue())
     return jsonify({
         "code": 0,
-        "image": str(img_str.decode('ascii')),
+        "image": get_encoded_img(image),
         "first_time": 1
     })