Update app.py

app.py

@@ -1,5 +1,4 @@
 print("[1/7] Importing streamlit...", end=" ", flush=True)
-
 import streamlit as st
 print("done")
 print("[2/7] Importing something...", end=" ", flush=True)
@@ -13,23 +12,18 @@ import zipfile
 import base64
 print("done")
 print("[3/7] Importing deepdanbooru...", end=" ", flush=True)
-
 import deepdanbooru as dd
 print("done")
 print("[4/7] Importing huggingface_hub...", end=" ", flush=True)
-
 import huggingface_hub
 print("done")
 print("[5/7] Importing tensorflow...", end=" ", flush=True)
-
 import tensorflow as tf
 print("done")
 print("[6/7] Importing numpy...", end=" ", flush=True)
-
 import numpy as np
 print("done")
 print("[6/7] Importing transformers...", end=" ", flush=True)
-
 from transformers import pipeline
 print("done")
 
@@ -81,12 +75,25 @@ def blur_image(image: Image.Image, blur_enabled: bool, nsfw_score: float) -> Ima
             image = image.filter(ImageFilter.BLUR)
     return image
 
-# blur_toggle_keyの初期値を設定
+# ブラーの有効/無効をトグルで制御する関数
 blur_toggle_key = "blur_toggle_unique_key_for_sidebar"
 
-# ブラーの有効/無効をトグルで制御する関数
 def get_blur_enabled():
-    return st.sidebar.checkbox("NSFW画像にブラーをかける", value=True, key=blur_toggle_key)
+    # 既に生成されたウィジェットがあればそれを返す
+    if blur_toggle_key in st.session_state:
+        return st.sidebar.checkbox("NSFW画像にブラーをかける", value=True, key=blur_toggle_key)
+
+    # まだウィジェットが生成されていない場合、新しいウィジェットを生成し、セッションステートに保存
+    blur_enabled = st.sidebar.checkbox("NSFW画像にブラーをかける", value=True, key=blur_toggle_key)
+    st.session_state[blur_toggle_key] = blur_enabled
+    return blur_enabled
+
+# ページングと並び替えのためのデータを取得する関数
+def load_data():
+    if os.path.exists(index_file_path):
+        return pd.read_csv(index_file_path)
+    else:
+        return pd.DataFrame(columns=["File Name", "Timestamp", "Tags"])
 
 # アップロードされた写真を保存する関数
 def save_uploaded_photo(uploaded_photo, file_name):
@@ -104,13 +111,6 @@ def save_uploaded_photo(uploaded_photo, file_name):
 
     image.save(os.path.join(photos_folder, file_name), "PNG")
 
-# ページングと並び替えのためのデータを取得する関数
-def load_data():
-    if os.path.exists(index_file_path):
-        return pd.read_csv(index_file_path)
-    else:
-        return pd.DataFrame(columns=["File Name", "Timestamp", "Tags"])
-
 # アップロードされた写真を表示する関数
 def display_photos(photos):
     for photo_info in photos.iterrows():
@@ -217,6 +217,6 @@ def main():
     if st.button("写真をダウンロード (ZIP)"):
         file_paths = [os.path.join(photos_folder, file) for file in os.listdir(photos_folder)]
         download_photos_as_zip(file_paths)
-print("all function defined.")
+
 if __name__ == "__main__":
     main()