Update app.py

app.py

@@ -47,8 +47,10 @@ def predict_tags(image: Image.Image, score_threshold: float) -> tuple[dict[str,
 
 # ページングと並び替えのためのデータを取得する関数
 def load_data():
-    index_df = pd.read_csv(index_file_path)
-    return index_df
+    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):
@@ -110,13 +112,16 @@ def main():
     model = load_model()
     labels = load_labels()
 
+    # インデックスを初期化
+    global photo_df
+    photo_df = load_data()
+
     if uploaded_photos:
         for uploaded_photo in uploaded_photos:
             file_name = f"{datetime.now().strftime('%Y%m%d%H%M%S%f')}.png"
             save_uploaded_photo(uploaded_photo, file_name)
 
             # インデックスに追加
-            global photo_df  # globalキーワードを追加
             photo_df = pd.concat([photo_df, pd.DataFrame([[os.path.basename(file_name), datetime.now(), ""]], columns=["File Name", "Timestamp", "Tags"])], ignore_index=True)
 
         # インデックスを更新
@@ -148,6 +153,7 @@ def main():
     elif selected_sort == "名前 降順":
         current_page = current_page.sort_values(by="File Name", ascending=False)
 
+    # ページに写真を表示
     display_photos(current_page)
 
     # Next ボタン