import requests
import cv2
import numpy as np
import base64
import sys
import json

if __name__ == '__main__':
    path = './data/template01.png'
    img = cv2.imread(path)
    retval, bytes = cv2.imencode('.jpg', img)
    encoded_image = base64.b64encode(bytes).decode('utf-8')
    user_id = '61AB9097'

    #给api_service.py发送请求
    url = "http://i-2.gpushare.com:53412/user/generate"
    payload = json.dumps({
        "user_id": "61AB9097",
        "base_img": encoded_image
    })
    headers = {
        'Content-Type': 'application/json'
    }
    response = requests.request("POST", url, headers=headers, data=payload)

    if response.status_code != 200:
        raise RuntimeError(f"Failed to send request to API service. Status code: {response.status_code}")

    ret_image_b64 = response.json().get('generate_photo_b64')

    if ret_image_b64 is None:
        raise RuntimeError(f"ret image failed!")

    image_array = np.frombuffer(base64.b64decode(ret_image_b64), np.uint8)
    image = cv2.imdecode(image_array, cv2.IMREAD_COLOR)
    cv2.imshow('image', cv2.resize(image, (0, 0), fx=0.5, fy=0.5))
    cv2.waitKey()