#!/usr/bin/env python3

import os
import requests

import logging
from logging.handlers import RotatingFileHandler

def setup_logger(logger_name, log_file, backup=5):
    logger = logging.getLogger(logger_name)
    fmt = '%(asctime)s - %(pathname)s[line:%(lineno)d] - %(levelname)s: %(message)s'
    format_str = logging.Formatter(fmt)
    file_handler = RotatingFileHandler(log_file, mode='a', maxBytes=500 * 1024 * 1024,
                                       backupCount=backup, encoding='utf-8')
    file_handler.setFormatter(format_str)
    logger.addHandler(file_handler)
    logger.setLevel(logging.INFO)
    return logger

log_file_path = './roop.log'
logger = setup_logger('roop_logger', log_file_path)

from roop import core

def init_run_env():
    path = os.getenv('path')
    print(path)
    current_path = os.path.abspath(os.curdir)
    ffmpeg_path = os.path.join(current_path, 'ffmpeg')
    path += ffmpeg_path + ';'
    print(path)
    os.environ['path'] = path
    print(os.getenv('path'))


def upload_post(paper, download_url):
    http_server_ip = '192.168.43.242'
    url = 'http://{}:5000/upload'.format(http_server_ip)
    print('upload file:', url)
    with open(paper, 'rb') as file:
        files = {'photo': file}  # 构建包含图片文件的字典
        data = {'url': download_url}
        response = requests.post(url, files=files, data=data)  # 发送 POST 请求进行图片上传
        print('response:', response.content, response.status_code)
        if response.status_code == 200:
            print('图片上传成功')
        else:
            print('图片上传失败')


if __name__ == '__main__':
    # httpserver.start_http_server()
    # player = VideoPlayer('./movie/video.mp4')

    init_run_env()
    core.run()
    # url = 'http://127.0.0.1:7001/public/upload/poster/1695864631144_42fb302c-9c87-4b2c-9a89-d8f7e60f5346_2023_09_28_093021.jpg_peper.png'
    # upload_post('./saved/out/2023_09_28_115912.jpg_peper.png', url)
    logger.info('exit')
    # httpserver.stop_http_server()