diff --git a/nft_svc/app.py b/nft_svc/app.py index 8382649..0c10697 100644 --- a/nft_svc/app.py +++ b/nft_svc/app.py @@ -1,11 +1,13 @@ -import base64 import io +import random 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__) @@ -43,12 +45,14 @@ def get_encoded_img(arr): @app.route('/getImage', methods=['GET']) def get_image(): - name = request.args.get('name') - fractal = optimized_mandelbrot(1000, 1000, np.random.randint(2, 251), np.random.uniform(-1, 1), np.random.uniform(-1, 1)) - img = get_encoded_img(fractal) + 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()) return jsonify({ "code": 0, - "image": img, + "image": str(img_str), "first_time": 1 }) diff --git a/nft_svc/data_100.pickle b/nft_svc/data_100.pickle new file mode 100644 index 0000000..2ad9145 Binary files /dev/null and b/nft_svc/data_100.pickle differ