Update app.py

app.py

@@ -50,6 +50,33 @@ if "blur_enabled" not in st.session_state:
 # blur_enabledの値を取得
 blur_enabled = st.session_state.blur_enabled
 
+# タグ検索用のテキスト入力
+tag_search_input = st.sidebar.text_input("タグ検索 (カンマでセパレート)", "")
+
+# タグ検索があればフィルタリング
+if tag_search_input:
+    tags_to_search = [tag.strip() for tag in tag_search_input.split(',')]
+    current_page = current_page[current_page["Tags"].apply(lambda x: all(tag in x.split(', ') for tag in tags_to_search))]
+
+# 並び替えのための選択肢
+selected_sort = st.sidebar.selectbox("写真の並び替え", ["TimeStamp 昇順", "TimeStamp 降順", "名前 昇順", "名前 降順"])
+
+# 並び替え
+if selected_sort == "TimeStamp 昇順":
+    current_page = current_page.sort_values(by="Timestamp", ascending=True)
+elif selected_sort == "TimeStamp 降順":
+    current_page = current_page.sort_values(by="Timestamp", ascending=False)
+elif selected_sort == "名前 昇順":
+    current_page = current_page.sort_values(by="File Name", ascending=True)
+elif selected_sort == "名前 降順":
+    current_page = current_page.sort_values(by="File Name", ascending=False)
+
+# タグ検索があれば結果を表示
+if tag_search_input:
+    st.sidebar.subheader("タグ検索結果")
+    st.sidebar.write(f"検索したタグ: {', '.join(tags_to_search)}")
+    st.sidebar.write(f"結果数: {len(current_page)}")
+
 # タグ付け関数
 def predict_tags(image: Image.Image, score_threshold: float) -> tuple[dict[str, float], dict[str, float], str]:
     _, height, width, _ = model.input_shape
@@ -208,6 +235,12 @@ def main():
     elif selected_sort == "名前 降順":
         current_page = current_page.sort_values(by="File Name", ascending=False)
 
+    # タグ検索があれば結果を表示
+    if tag_search_input:
+        st.sidebar.subheader("タグ検索結果")
+        st.sidebar.write(f"検索したタグ: {', '.join(tags_to_search)}")
+        st.sidebar.write(f"結果数: {len(current_page)}")
+
     # ページに写真を表示
     display_photos(current_page)