Upload 2 files

Danbooru Prompt Selector/TEST2024/prompt_selector_auto_1215_testv9.exe

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:fe74b40a778bdf46e0e7fa36feecbd72627c251990a0b889c8be6e606b00c0b6
-size 151211498
+oid sha256:2de13ef437ae80bdf617829959f017d41cf55d8162935beac38a43ed370442c6
+size 151554073

Danbooru Prompt Selector/TEST2024/prompt_selector_auto_1215_testv9.py

@@ -32,6 +32,10 @@ import character_dictionary as cd
 import pandas as pd
 import webbrowser
 from collections import OrderedDict
+import platform
+from openpyxl import Workbook
+from openpyxl.styles import Alignment
+from openpyxl.drawing.image import Image as OpenpyxlImage
 
 BASE_URL="https://api.novelai.net"
 copyright_keys = copyright_list_reformatted.copyright_list
@@ -85,21 +89,78 @@ def generate_image(access_token, prompt, model, action, parameters):
 
 def history_queue(image, prompt, seed):
     global history
+    #new_image = image.thumbnail((192, 192)) # 가로 세로 100픽셀로 조정
     history.append([image, prompt, seed])
-    if len(history) > 5:
-        history.pop(0)
 
 def show_history():
     global history
 
+    def save_to_excel():
+        """
+        contributor : highnoon1
+        function : 히스토리 엑셀로 저장 (무제한 보기는 성능 이슈로 제외)
+        """
+        global start_time
+        workbook = Workbook()
+        sheet = workbook.active
+
+        alignment = Alignment(wrapText=True)
+        sheet['A1'].alignment = alignment
+        sheet['A1'] = 'Prompt'
+        sheet['B1'] = 'Seed'
+        sheet['C1'] = 'Image'
+
+        image_objects = []  # BytesIO 객체를 추적하기 위한 리스트
+
+        # history 데이터를 반복하며 Excel에 저장
+        for index, (pil_image, prompt, seed) in enumerate(history, start=1):
+            
+            row_num = index + 1  # 행 번호 (Excel 행은 1부터 시작)
+
+            # 프롬프트와 시드 삽입
+            prompt_cell = sheet.cell(row=row_num, column=1, value=prompt)
+            sheet.cell(row=row_num, column=2, value=seed)
+            prompt_cell.alignment = Alignment(wrap_text=True)
+
+            # PIL 이미지를 BytesIO 객체에 PNG 형식으로 저장
+            pil_image = resize_image_to_fit(pil_image,192)
+            output = io.BytesIO()
+            pil_image.save(output, format='PNG')
+            output.seek(0)
+            image_objects.append(output)  # 리스트에 추가
+
+            # Openpyxl 이미지 생성 및 시트에 추가
+            img = OpenpyxlImage(output)
+            img.anchor = f'C{row_num}'  # 이미지를 C열에 고정
+            sheet.add_image(img)
+
+            # 이미지 크기에 따라 행 높이 및 열 너비 조정
+            scale_factor = 0.75
+            sheet.row_dimensions[row_num].height = pil_image.height * scale_factor
+            sheet.column_dimensions['C'].width = pil_image.width * scale_factor / 7
+
+        # A열과 B열 크기 조정
+        sheet.column_dimensions['A'].width = 50
+        sheet.column_dimensions['B'].width = 15
+
+        # 워크북 저장
+        workbook.save(f"history_with_images_{start_time}.xlsx")
+
+        # 모든 BytesIO 객체 닫기
+        for img_obj in image_objects:
+            img_obj.close()
+            if os.path.exists(f"history_with_images_{start_time}.xlsx"):
+                os.startfile(f"history_with_images_{start_time}.xlsx")
+
+        print("엑셀 파일에 저장됨")
+
     # 새로운 toplevel 창 생성
     history_window = tk.Toplevel(window)
     history_window.title("History")
 
     # history_window에서 썸네일과 텍스트를 표시할 grid 설정
-    for index, (image, prompt, seed) in enumerate(history):
-        # 이미지를 썸네일로 변환
-        image.thumbnail((192, 192))  # 가로 세로 100픽셀로 조정
+    for index, (image, prompt, seed) in enumerate(history[-5:]):
+        image.thumbnail((192, 192)) # 가로 세로 100픽셀로 조정
         photo = ImageTk.PhotoImage(image)
         
         # 썸네일 이미지 라벨 생성
@@ -113,6 +174,9 @@ def show_history():
         text_widget.insert('1.0', text_prompt)
         text_widget.grid(row=index, column=1)
         text_widget.config(state='disabled')  # 텍스트 위젯을 읽기 전용으로 설정
+    
+    save_button = tk.Button(history_window, text=f"Save to Excel ({len(history)})", command=save_to_excel)
+    save_button.grid(row=50)
 
 def generate(width, height, positive, negative, button):
     global temp_clipboard_image
@@ -227,7 +291,8 @@ def generate(width, height, positive, negative, button):
         button.config(state=tk.NORMAL)
 
     global delay_offset,cached_rows
-    time.sleep(delay_offset)
+    if delay_offset >= 0:
+        time.sleep(delay_offset)
 
     running_flag = False
     last_generation_seed = params['seed']
@@ -237,7 +302,12 @@ def generate(width, height, positive, negative, button):
         random_function()
         if len(cached_rows) == 0:
             mac_var.set(0)
-        time.sleep(random.uniform(5.1, 8.5))
+        sleep_time = random.uniform(5.1, 8.5)
+        if delay_offset < 0:
+            sleep_time += delay_offset
+            if sleep_time < 0.5:
+                sleep_time = 0.5
+        time.sleep(sleep_time)
         button.config(state=tk.DISABLED)
         window.event_generate(GENERATE_EVENT, when="tail")
     else:
@@ -1899,7 +1969,8 @@ def generate_turbo(_seed, turbo_count, tb_access_token,width, height, positive,
     entry_seed_value.set(last_generation_seed)
 
     global delay_offset
-    time.sleep(delay_offset)
+    if delay_offset >= 0:
+        time.sleep(delay_offset)
 
     stopped = False
     if (turbo_stop_bit):
@@ -2354,7 +2425,7 @@ def show_automation_option():
         if delay_entry.get():
             try:
                 delay_offset = round(float(delay_entry.get()), 1)
-                delay_offset_label.config(text="Dealy offset : "+str(delay_offset))
+                delay_offset_label.config(text="Delay offset : "+str(delay_offset))
             except ValueError as e:
                 print(e)
                 delay_offset = 0