Update app.py

app.py

@@ -13,6 +13,9 @@ if 'blur_option' not in st.session_state:
 # Initialize Gradio client
 client_nsfw = Client("https://ozoneasai-falconsai-nsfw-image-detection.hf.space/--replicas/0fyhj/")
 
+# File uploader interface
+uploaded_files = st.file_uploader("画像ファイルをアップロードしてください", type=["jpg", "jpeg", "png"], accept_multiple_files=True)
+
 # Initialize pagination
 if 'page' not in st.session_state:
     st.session_state.page = 1
@@ -118,8 +121,8 @@ def display_images(images):
                         st.experimental_rerun()
 
 # Main part of the app
-if 'uploaded_files' in st.session_state:
-    save_uploaded_files(st.session_state.uploaded_files)
+if uploaded_files:
+    save_uploaded_files(uploaded_files)
 
 files_per_page = 20
 num_pages = (len(file_paths) - 1) // files_per_page + 1
@@ -162,7 +165,7 @@ if st.button("CSVに保存"):
         csv_writer = csv.writer(csv_file)
         csv_writer.writerow(['ファイル名', 'NSFWの予測', 'Goodの値', '作成時間'])
         for i, file_path in enumerate(file_paths):
-            gradio_nsfw_prediction = gradio_nsfw_predictions[i]
+            gradio_nsfw_prediction = gradio_nsfw_predictions[i][1]  # Accessing the prediction result from the tuple
             good_value = good_values[i]
             timestamp = timestamps[i]
             csv_writer.writerow([os.path.basename(file_path), gradio_nsfw_prediction, good_value, timestamp])