Update app.py

app.py

@@ -203,9 +203,29 @@ def load_labels():
     with open(path) as f:
         labels = [line.strip() for line in f.readlines()]
     return labels
+def update_photo_detail_in_csv(df, photo_folder):
+    photo_files = os.listdir(photo_folder)
+    new_df_rows = []
 
+    for file in photo_files:
+        file_path = os.path.join(photo_folder, file)
+        if not df['File Name'].eq(file).any():
+            image = Image.open(file_path)
+
+            # タグを予測して表示
+            result_threshold, result_all, result_text = predict_tags(image, 0.7)
+
+            new_df_rows.append([file, datetime.now(), ", ".join(result_all.keys())])
+    
+    if new_df_rows:
+        new_df = pd.DataFrame(new_df_rows, columns=["File Name", "Timestamp", "Tags"])
+        df = pd.concat([df, new_df])
+        df.to_csv(index_file_path, index=False)
+
+    return df
 # Streamlit アプリケーションのメイン部分
 def main():
+    
     st.sidebar.title("アップロードオプション")
     uploaded_photos = st.sidebar.file_uploader("写真をアップロードしてください", type=["jpg", "jpeg", "png"], accept_multiple_files=True)
 
@@ -230,6 +250,7 @@ def main():
         photo_df.sort_values(by="Timestamp", inplace=True, ascending=False)
         photo_df.reset_index(drop=True, inplace=True)
         photo_df.to_csv(index_file_path, index=False)
+    photo_df = update_photo_detail_in_csv(photo_df, temp_folder)
 
     st.subheader("アップロードされた写真")