Update app.py

app.py

@@ -75,37 +75,33 @@ def paginate_files(files, page, files_per_page):
     return files[start_index:end_index]
 
 # Function to display images with predictions and options
-def display_images(images):
+def display_images(images, rows):
     for i, file_path in enumerate(images):
         file_ext = os.path.splitext(file_path)[1].lower()
 
         # Check if index is within the range
-        if os.path.exists("temp/index.csv") and i < len(images):
-            with open("temp/index.csv", mode='r', newline='', encoding='utf-8') as csv_file:
-                csv_reader = csv.reader(csv_file)
-                rows = list(csv_reader)
-                if i < len(rows):
-                    gradio_nsfw_prediction = rows[i][1]
-                    timestamp = rows[i][2]
-
-                    st.write(f"**Prediction for {os.path.basename(file_path)} (NSFW):** {gradio_nsfw_prediction}")
-                    st.write(f"**Timestamp:** {timestamp}")
-
-                    if st.session_state.blur_option and gradio_nsfw_prediction.lower() == "nsfw":
-                        blurred_img = apply_blur(file_path)
-                        st.image(blurred_img, caption=os.path.basename(file_path), use_column_width=True)
-                    else:
-                        if file_ext in [".jpg", ".png"]:
-                            st.image(Image.open(file_path), caption=os.path.basename(file_path), use_column_width=True)
-                        col1, col2 = st.columns([4, 1])
-                        with col1:
-                            if file_ext in [".jpg", ".png"]:
-                                st.image(Image.open(file_path), caption=os.path.basename(file_path), use_column_width=True)
-                        with col2:
-                            if col2.button("削除", key=file_path):
-                                os.remove(file_path)
-                                # Update the displayed images after deletion
-                                st.experimental_rerun()
+        if os.path.exists("temp/index.csv") and i < len(rows):
+            gradio_nsfw_prediction = rows[i][1]
+            timestamp = rows[i][2]
+
+            st.write(f"**Prediction for {os.path.basename(file_path)} (NSFW):** {gradio_nsfw_prediction}")
+            st.write(f"**Timestamp:** {timestamp}")
+
+            if st.session_state.blur_option and gradio_nsfw_prediction.lower() == "nsfw":
+                blurred_img = apply_blur(file_path)
+                st.image(blurred_img, caption=os.path.basename(file_path), use_column_width=True)
+            else:
+                if file_ext in [".jpg", ".png"]:
+                    st.image(Image.open(file_path), caption=os.path.basename(file_path), use_column_width=True)
+                col1, col2 = st.columns([4, 1])
+                with col1:
+                    if file_ext in [".jpg", ".png"]:
+                        st.image(Image.open(file_path), caption=os.path.basename(file_path), use_column_width=True)
+                with col2:
+                    if col2.button("削除", key=file_path):
+                        os.remove(file_path)
+                        # Update the displayed images after deletion
+                        st.experimental_rerun()
 
 # Main part of the app
 if uploaded_files:
@@ -142,13 +138,12 @@ with open("temp/index.csv", mode='r', newline='', encoding='utf-8') as csv_file:
 # Display images with predictions and options
 current_page_files = paginate_files([row[0] for row in rows], st.session_state.page, files_per_page)
 st.write("ファイル一覧:")
-st.write(current_page_files)
-display_images(current_page_files)
+display_images(current_page_files, rows)
 
 # Blurのトグルを更新
 st.session_state.blur_option = st.checkbox("NSFW画像にBlurをかける", value=st.session_state.blur_option)
 
-# CSVに保存# CSVに保存ボタン
+# CSVに保存ボタン
 if st.button("CSVに保存"):
     csv_filename = "gradio_predictions_and_timestamps.csv"
     with open(csv_filename, mode='w', newline='', encoding='utf-8') as csv_file: