Upload 2 files

Danbooru Prompt Selector/TEST/prompt_selector_auto_1129_TESTv2.exe

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:d98d1aef959375ad0f1bf89ef9a58b79e207afd1e47b9e9922cab5a4e2955ab0
-size 81747886
+oid sha256:d97e584e1267513494032f9b82dcdd5509fecd4d52b3963a692cdd7244271739
+size 81748020

Danbooru Prompt Selector/TEST/prompt_selector_auto_1129_TESTv2.py

@@ -26,6 +26,7 @@ from PIL import Image, ImageOps, ImageFilter,ImageTk
 from IPython.display import display
 import win32clipboard
 from io import BytesIO
+from datetime import datetime
 
 BASE_URL="https://api.novelai.net"
 
@@ -112,6 +113,17 @@ def generate(width, height, positive, negative, button):
         else:
             return image
 
+    def log_error(e, output_file_path="output_file_path"):
+        # 현재 시간을 얻습니다
+        current_time = datetime.now().strftime("%m/%d %H:%M:%S")
+
+        # 에러 로그 메시지
+        error_message = f"#### Error occured at {current_time} ####\nError: {e}\n############################################\n"
+
+        # 지정된 출력 폴더의 error_log.txt 파일에 쓰기
+        with open(f"error_log.txt", "a") as file:
+            file.write(error_message)
+
     global access_token
     try:
         zipped_bytes = generate_image(access_token, positive, "nai-diffusion-3", "generate", params)
@@ -134,8 +146,12 @@ def generate(width, height, positive, negative, button):
         output_file_path = "output_image.jpg"
         i.save(output_file_path)
     except Exception as e:
-        text_output.insert(tk.END, f"Error: {e}", fg='red')
-        button.config(state=tk.NORMAL)
+        #text_output.insert(tk.END, f"Error: {e}", fg='red')
+        log_error(e, "path_to_output_folder")
+        time.sleep(random.uniform(3.0, 9.0))
+        button.config(state=tk.DISABLED)
+        window.event_generate(GENERATE_EVENT, when="tail")
+
     time.sleep(random.uniform(2.3, 6.5))
     button.config(state=tk.NORMAL)
     if mac_var.get():