Karl/comfy_fm24_newgens
1
0
Fork 0
mirror of https://github.com/karl0ss/comfy_fm24_newgens.git synced 2025-04-29 04:13:40 +01:00
Code Issues Packages Projects Releases Wiki Activity

reworked to work with processed_players

Browse Source
This commit is contained in:
Karl Hudgell 2024-12-16 15:57:02 +00:00
parent 81d05515b3
commit 659c40be28
1 changed files with 9 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
comfy_fm_newgen.py
View File

@ -12,7 +12,7 @@ import logging.config
from tqdm import tqdm from tqdm import tqdm
from lib.rtf_parser import RTF_Parser from lib.rtf_parser import RTF_Parser
from lib.remove_bg import remove_bg_from_files_in_dir from lib.remove_bg import remove_bg_from_file_list
from lib.generate_xml import create_config_xml, append_to_config_xml from lib.generate_xml import create_config_xml, append_to_config_xml
from lib.resize_images import resize_images from lib.resize_images import resize_images
from lib.xml_reader import extract_from_values from lib.xml_reader import extract_from_values
@ -23,14 +23,14 @@ from comfy_api_simplified import ComfyApiWrapper, ComfyWorkflowWrapper
logging.config.dictConfig(LOGGING_CONFIG) logging.config.dictConfig(LOGGING_CONFIG)
cut = 10 cut = 2
update = False update = True
use_gpu = False use_gpu = False
# Load user configurations # Load user configurations
user_config = configparser.ConfigParser() user_config = configparser.ConfigParser()
try: try:
user_config.read("config.cfg") user_config.read("user_config.cfg")
output_folder = user_config["general"]["output_dir"] output_folder = user_config["general"]["output_dir"]
logging.debug("Configuration loaded successfully.") logging.debug("Configuration loaded successfully.")
except KeyError as e: except KeyError as e:
@ -122,11 +122,11 @@ def post_process_images(output_folder, update, processed_players, football_manag
""" """
try: try:
# Resize images to desired dimensions # Resize images to desired dimensions
resize_images(output_folder) resize_images(output_folder, processed_players)
logging.debug("Images resized successfully.") logging.debug("Images resized successfully.")
# Remove background from images using GPU if available # Remove background from images using GPU if available
remove_bg_from_files_in_dir(output_folder, use_gpu=use_gpu) remove_bg_from_file_list(output_folder,processed_players, use_gpu=use_gpu)
logging.debug("Background removed from images.") logging.debug("Background removed from images.")
# Update or create configuration XML # Update or create configuration XML
@ -134,7 +134,7 @@ def post_process_images(output_folder, update, processed_players, football_manag
append_to_config_xml(output_folder, processed_players, football_manager_version) append_to_config_xml(output_folder, processed_players, football_manager_version)
logging.debug("Configuration XML updated.") logging.debug("Configuration XML updated.")
else: else:
create_config_xml(output_folder, football_manager_version) create_config_xml(output_folder,processed_players, football_manager_version)
logging.debug("Configuration XML created.") logging.debug("Configuration XML created.")
except Exception as e: except Exception as e:
logging.error(f"Post-processing failed: {e}") logging.error(f"Post-processing failed: {e}")
@ -172,11 +172,11 @@ def main():
# Load configurations # Load configurations
try: try:
with open("config.json", "r") as f: with open("app_config.json", "r") as f:
app_config = json.load(f) app_config = json.load(f)
logging.debug("Application configuration loaded successfully.") logging.debug("Application configuration loaded successfully.")
except FileNotFoundError: except FileNotFoundError:
logging.error("config.json file not found.") logging.error("app_config.json file not found.")
sys.exit(1) sys.exit(1)
# Check for processed # Check for processed