# -*- coding:utf-8 -*-
from voice import voice_fix
from voice import voice_get
from pic import pic_fix
from data import data_fix
import os
import requests
import warnings
from flask import Flask, request, jsonify
import json
warnings.filterwarnings("ignore")


app = Flask(__name__)
app.config.from_object(__name__)
app.config["JSON_AS_ASCII"] = False


@app.route('/welcome', methods=['GET'])
def welcome():
    return 0


@app.route('/voice', methods=['POST'])
def voice():
    if request.method == 'POST':
        body = request.json
        flag = body['flag']
        user = body['user']
        if flag == '1':
            print('开始录音，小声时结束')
            voice_get.listen()
            text = voice_fix.fix()
            headers = {
                "Content-Type": "application/json",
                "Authorization": "Bearer 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"
            }
            url = 'https://open.aiproxy.xyz/v1/chat/completions'
            data = {
                "model": "gpt-3.5-turbo",
                "messages": [{"role": user, "content": text}]
            }
            response = requests.post(url, json=data, headers=headers)
            response = response.text
            response = response.encode('utf-8')
            response = json.loads(response)
            print(response)
            return response
        else:
            return {'error': 'log has something wrong'}
    else:
        return {'error': 'web connect wrong'}


# @app.route('/pic')
# def pic():
#     return 0


# @app.route('/login', methods=['POST'])
# def test():
#     if request.method == 'POST':
#         text = request.json
#         print(text)
#         headers = {
#                 "Content-Type": "application/json",
#                 "Authorization": "Bearer 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"
#             }
#         url = 'https://open.aiproxy.xyz/v1/chat/completions'
#         data = {
#                 "model": "gpt-3.5-turbo",
#                 "messages": [{"role": "user", "content": text['content']}]
#             }
#         response = requests.post(url, json=data, headers=headers)
#     print(response.text)
#     return response.json()


if __name__ == '__main__':
    app.run()