mirror/DeepCreamPy
1
0
Fork 0
mirror of https://github.com/Deepshift/DeepCreamPy.git synced 2024-11-29 05:10:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

improved naming for decensored files

Browse Source
This commit is contained in:
smethan 2018-11-01 09:47:54 -05:00
parent 3ab32da970
commit f44a0e9e53
1 changed files with 10 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
ui.py
View File

@ -37,6 +37,7 @@ from tkinter import messagebox
from tkinter import filedialog
import framework
import decensor
import os
class PaintApplication(framework.Framework):
@ -178,10 +179,16 @@ class PaintApplication(framework.Framework):
combined_img = Image.alpha_composite(self.canvas.img.convert('RGBA'), self.drawn_img)
decensorer = decensor.Decensor()
print(self.file_name)
decensorer.decensor_image(combined_img.convert('RGB'),combined_img.convert('RGB'), self.file_name + ".png")
save_path = self.file_name + ".png"
orig_name = self.file_name
path, file = os.path.split(self.file_name)
name, ext = os.path.splitext(file)
name = name + "_decensored"
self.file_name = os.path.join(path, name + ext)
print(self.file_name)
decensorer.decensor_image(combined_img.convert('RGB'),combined_img.convert('RGB'), self.file_name)
messagebox.showinfo(
"Decensoring", "Decensoring complete! image saved to {save_path}".format(save_path=save_path))
"Decensoring", "Decensoring complete! image saved to {save_path}".format(save_path=self.file_name))
self.file_name = orig_name
def on_about_menu_clicked(self, event=None):
# messagebox.showinfo(
# "Decensoring", "Decensoring in progress.")