Update app.py

app.py

@@ -7,109 +7,181 @@ from pathlib import Path
 import git  # GitPython
 from core.file_scanner import FileScanner, FileInfo
 
-
+# =====================================
 # セッション状態の初期化
+# =====================================
 if 'scanned_files' not in st.session_state:
     st.session_state.scanned_files = []  # スキャンしたFileInfoリスト
 if 'selected_files' not in st.session_state:
-    st.session_state.selected_files = set()  # ユーザーが選択中のファイルパス
+    st.session_state.selected_files = set()  # ユーザーが選択中のファイルパス (相対パス)
 if 'cloned_repo_dir' not in st.session_state:
-    st.session_state.cloned_repo_dir = None  # クローン先ディレクトリのパス文字列
+    st.session_state.cloned_repo_dir = None  # クローン先ディレクトリの絶対パス文字列
+
+# =====================================
+# タイトル等
+# =====================================
+st.title("Gitリポジトリ スキャナー")
+st.markdown("**ディレクトリ構造をツリー表示し、ファイルを選んでMarkdownダウンロードできます**")
+
+# =====================================
+# ツリー構造を生成する関数
+# =====================================
+def build_tree(paths):
+    """
+    相対パス(Pathオブジェクト)のリストからツリー状のネスト構造を構築する。
+    戻り値は {要素名 -> 子要素のdict or None} という入れ子の辞書。
+    """
+    tree = {}
+    for p in paths:
+        parts = p.parts
+        current = tree
+        for i, part in enumerate(parts):
+            if i == len(parts) - 1:
+                # ファイルやフォルダの末端
+                current[part] = None
+            else:
+                if part not in current:
+                    current[part] = {}
+                if isinstance(current[part], dict):
+                    current = current[part]
+                else:
+                    # もしNoneだった場合(同名のファイル/フォルダがあるなど) → 無理やりdictに
+                    current[part] = {}
+                    current = current[part]
+    return tree
+
+def format_tree(tree_dict, prefix=""):
+    """
+    build_tree()で作ったネスト構造をASCIIアートのツリー文字列にする。
+    """
+    lines = []
+    entries = sorted(tree_dict.keys())
+    for i, entry in enumerate(entries):
+        is_last = (i == len(entries) - 1)
+        marker = "└── " if is_last else "├── "
+        # 子要素がある(=dict)ならフォルダ、Noneならファイル
+        if isinstance(tree_dict[entry], dict):
+            # フォルダとして表示
+            lines.append(prefix + marker + entry + "/")
+            # 次の階層のプレフィックスを用意
+            extension = "    " if is_last else "│   "
+            sub_prefix = prefix + extension
+            # 再帰的に生成
+            lines.extend(format_tree(tree_dict[entry], sub_prefix))
+        else:
+            # ファイルとして表示
+            lines.append(prefix + marker + entry)
+    return lines
 
-st.title("Gitリポジトリ スキャナー (削除ボタン付き)")
 
-# --- 1) ユーザー入力: GitリポジトリURL ---
-repo_url = st.text_input("GitリポジトリURLを入力 (例: https://github.com/username/repo.git)")
+# =====================================
+# ユーザー入力
+# =====================================
+repo_url = st.text_input("GitリポジトリURL (例: https://github.com/username/repo.git)")
 
-# --- 2) スキャン対象拡張子の選択 ---
 st.subheader("スキャン対象拡張子")
-available_exts = [".py", ".js", ".ts", ".md", ".txt", ".java", ".cpp",".sh"]
+available_exts = [".py", ".js", ".ts", ".md", ".txt", ".java", ".cpp"]
 chosen_exts = []
 for ext in available_exts:
-    default_checked = (ext in [".py", ".md"])  # デモ用に.pyと.mdを初期オン
+    default_checked = (ext in [".py", ".md"])  # デモ用に .py と .md を初期ON
     if st.checkbox(ext, key=f"ext_{ext}", value=default_checked):
         chosen_exts.append(ext)
 
-# --- 3) 「スキャン開始」ボタンで clone + scan ---
+# =====================================
+# スキャン開始ボタン
+# =====================================
 if st.button("スキャン開始"):
     if not repo_url.strip():
-        st.error("リポジトリURLを入力してください")
+        st.error("リポジトリURLを入力してください。")
     else:
-        # 既にクローンしたフォルダがあれば削除
+        # 既にクローン済フォルダがあれば削除
         if st.session_state.cloned_repo_dir and Path(st.session_state.cloned_repo_dir).exists():
             shutil.rmtree(st.session_state.cloned_repo_dir, ignore_errors=True)
         
-        # 一時フォルダを作成
+        # 一時フォルダを作成してクローン
         tmp_dir = tempfile.mkdtemp()
         clone_path = Path(tmp_dir) / "cloned_repo"
 
-        st.write(f"リポジトリを一時ディレクトリへクローン中: {clone_path}")
         try:
+            st.write(f"リポジトリをクローン中: {clone_path}")
             git.Repo.clone_from(repo_url, clone_path)
             st.session_state.cloned_repo_dir = str(clone_path)
         except Exception as e:
             st.error(f"クローン失敗: {e}")
+            st.session_state.cloned_repo_dir = None
+            st.session_state.scanned_files = []
             st.stop()
-        
+
         # スキャン
         scanner = FileScanner(base_dir=clone_path, target_extensions=set(chosen_exts))
         found_files = scanner.scan_files()
 
-        # スキャン結果をセッションに保存
         st.session_state.scanned_files = found_files
         st.session_state.selected_files = set()
 
         st.success(f"スキャン完了: {len(found_files)}個のファイルを検出")
 
-# --- 4) データ削除ボタン ---
+# =====================================
+# クローン削除ボタン
+# =====================================
 if st.session_state.cloned_repo_dir:
-    # クローンに成功している場合のみ表示
-    if st.button("クローン済みリポジトリを削除"):
-        # クローン先フォルダを削除
+    if st.button("クローン済みデータを削除"):
         shutil.rmtree(st.session_state.cloned_repo_dir, ignore_errors=True)
         st.session_state.cloned_repo_dir = None
         st.session_state.scanned_files = []
         st.session_state.selected_files = set()
-        st.success("クローンしたデータを削除しました")
-
+        st.success("クローンしたディレクトリを削除しました")
 
-# --- 5) スキャン結果の表示 + チェックボックスで選択 ---
+# =====================================
+# スキャン結果がある場合 → ツリー表示 + ファイル選択
+# =====================================
 if st.session_state.scanned_files:
-    st.write("## スキャン結果 (指定拡張子のみ)")
-
-    col1, col2 = st.columns(2)
-    with col1:
-        if st.button("すべて選択"):
-            base_path = Path(st.session_state.cloned_repo_dir)
-            st.session_state.selected_files = {
-                f.path.relative_to(base_path) for f in st.session_state.scanned_files
-            }
-    with col2:
-        if st.button("すべて解除"):
-            st.session_state.selected_files = set()
-
     base_path = Path(st.session_state.cloned_repo_dir)
-    for file_info in st.session_state.scanned_files:
-        rel_path = file_info.path.relative_to(base_path)
-        checked = (rel_path in st.session_state.selected_files)
-        
-        # ファイルのチェックボックスを表示
-        new_checked = st.checkbox(
-            f"{rel_path} ({file_info.formatted_size})",
-            value=checked,
-            key=str(rel_path)  # key 重複回避のため文字列化
-        )
-        if new_checked:
-            st.session_state.selected_files.add(rel_path)
-        else:
-            st.session_state.selected_files.discard(rel_path)
 
+    # --- ツリーを作る ---
+    # scanned_files は「指定拡張子」だけ取得されているので、そのファイルパスのみでツリーを構築
+    rel_paths = [f.path.relative_to(base_path) for f in st.session_state.scanned_files]
+    tree_dict = build_tree(rel_paths)
+    tree_lines = format_tree(tree_dict)
+    ascii_tree = "\n".join(tree_lines)
+
+    st.write("## スキャン結果")
+    col_tree, col_files = st.columns([1, 2])  # 左:ツリー, 右:ファイル一覧
+
+    with col_tree:
+        st.markdown("**ディレクトリ構造 (指定拡張子のみ)**")
+        st.markdown(f"```\n{ascii_tree}\n```")
+
+    with col_files:
+        st.markdown("**ファイル一覧 (チェックボックス)**")
+
+        col_btn1, col_btn2 = st.columns(2)
+        with col_btn1:
+            if st.button("すべて選択"):
+                st.session_state.selected_files = set(rel_paths)
+        with col_btn2:
+            if st.button("すべて解除"):
+                st.session_state.selected_files = set()
+
+        for file_info in st.session_state.scanned_files:
+            rel_path = file_info.path.relative_to(base_path)
+            checked = rel_path in st.session_state.selected_files
+
+            new_checked = st.checkbox(
+                f"{rel_path} ({file_info.formatted_size})",
+                value=checked,
+                key=str(rel_path)  # keyの重複回避
+            )
+            if new_checked:
+                st.session_state.selected_files.add(rel_path)
+            else:
+                st.session_state.selected_files.discard(rel_path)
 
-# --- 6) 選択したファイルをまとめてMarkdown化 & ダウンロード ---
+
+# =====================================
+# 選択ファイルをまとめてMarkdown化 & ダウンロード
+# =====================================
 def create_markdown_for_selected(files, selected_paths, base_dir: Path) -> str:
-    """
-    選択さ���たファイルを1つのMarkdownとして連結。
-    """
     output = []
     for f in files:
         rel_path = f.path.relative_to(base_dir)
@@ -120,20 +192,18 @@ def create_markdown_for_selected(files, selected_paths, base_dir: Path) -> str:
                 output.append(f.content)
             else:
                 output.append("# Failed to read content")
-            output.append("")  # ファイルごとに空行
-
+            output.append("")  # 空行
     return "\n".join(output)
 
 if st.session_state.scanned_files:
-    st.write("## 選択ファイルをMarkdownとしてダウンロード")
-    if st.button("選択ファイルをダウンロード"):
+    st.write("## 選択ファイルをダウンロード")
+    if st.button("選択ファイルをMarkdownとしてダウンロード"):
         base_path = Path(st.session_state.cloned_repo_dir)
         markdown_text = create_markdown_for_selected(
             st.session_state.scanned_files,
             st.session_state.selected_files,
             base_path
         )
-        # ダウンロードボタン
         st.download_button(
             label="Markdownダウンロード",
             data=markdown_text,