Update app.py

app.py

@@ -1,53 +1,86 @@
 # app.py
 
 import streamlit as st
+import tempfile
+import shutil
 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 = []  # List[FileInfo]
+    st.session_state.scanned_files = []  # スキャンしたFileInfoリスト
 if 'selected_files' not in st.session_state:
-    st.session_state.selected_files = set()  # {Path} 相対パス or 絶対パス
+    st.session_state.selected_files = set()  # ユーザーが選択中のファイルパス
+if 'cloned_repo_dir' not in st.session_state:
+    st.session_state.cloned_repo_dir = None  # クローン先ディレクトリのパス文字列
 
-st.title("指定拡張子のみディレクトリスキャン & 選択ダウンロード")
+st.title("Gitリポジトリ スキャナー (削除ボタン付き)")
 
-# --- 1) ユーザー入力: スキャンしたいローカルディレクトリパス ---
-dir_path_str = st.text_input("ローカルディレクトリパス", value=".")
+# --- 1) ユーザー入力: GitリポジトリURL ---
+repo_url = st.text_input("GitリポジトリURLを入力 (例: https://github.com/username/repo.git)")
 
-# --- 2) ユーザー入力: スキャン対象拡張子選択 ---
+# --- 2) スキャン対象拡張子の選択 ---
 st.subheader("スキャン対象拡張子")
 available_exts = [".py", ".js", ".ts", ".md", ".txt", ".java", ".cpp"]
 chosen_exts = []
 for ext in available_exts:
-    if st.checkbox(ext, key=f"ext_{ext}", value=(ext in [".py", ".md"])):
+    default_checked = (ext in [".py", ".md"])  # デモ用に.pyと.mdを初期オン
+    if st.checkbox(ext, key=f"ext_{ext}", value=default_checked):
         chosen_exts.append(ext)
 
-# --- 3) 「スキャン実行」ボタン ---
+# --- 3) 「スキャン開始」ボタンで clone + scan ---
 if st.button("スキャン開始"):
-    base_path = Path(dir_path_str).resolve()
-    if not base_path.is_dir():
-        st.error("有効なディレクトリを指定してください")
+    if not repo_url.strip():
+        st.error("リポジトリURLを入力してください")
     else:
-        scanner = FileScanner(base_path, set(chosen_exts))
+        # 既にクローンしたフォルダがあれば削除
+        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:
+            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.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.session_state.selected_files = set()
 
-        st.success(f"スキャン完了: {len(found_files)}個のファイル")
+        st.success(f"スキャン完了: {len(found_files)}個のファイルを検出")
 
-# --- 4) ファイルを表示するUI（ディレクトリ構造 or リスト） + 全選択 / 全解除 ---
+# --- 4) データ削除ボタン ---
+if st.session_state.cloned_repo_dir:
+    # クローンに成功している場合のみ表示
+    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("クローンしたデータを削除しました")
 
+
+# --- 5) スキャン結果の表示 + チェックボックスで選択 ---
 if st.session_state.scanned_files:
-    # 全選択 / 全解除ボタン
+    st.write("## スキャン結果 (指定拡張子のみ)")
+
     col1, col2 = st.columns(2)
-    
     with col1:
         if st.button("すべて選択"):
-            # すべてのファイルを選択
-            base_path = Path(dir_path_str).resolve()
-            # relative_to で相対パスを使うか、絶対パスを使うかは好みでOK
+            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
             }
@@ -55,32 +88,27 @@ if st.session_state.scanned_files:
         if st.button("すべて解除"):
             st.session_state.selected_files = set()
 
-    st.write("### ファイル一覧 (指定拡張子のみ)")
-    base_path = Path(dir_path_str).resolve()
-    
+    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は重複しないように文字列にしておく
+            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)
 
-# --- 5) 選択ファイルをまとめてMarkdown化 & ダウンロード ---
 
+# --- 6) 選択したファイルをまとめてMarkdown化 & ダウンロード ---
 def create_markdown_for_selected(files, selected_paths, base_dir: Path) -> str:
     """
-    選択されたファイルだけをMarkdownテキストとして返す
+    選択されたファイルを1つのMarkdownとして連結。
     """
     output = []
     for f in files:
@@ -92,22 +120,22 @@ 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("選択ファイルをまとめてダウンロード"):
-        base_path = Path(dir_path_str).resolve()
+    st.write("## 選択ファイルをMarkdownとしてダウンロード")
+    if st.button("選択ファイルをダウンロード"):
+        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="ダウンロード",
+            label="Markdownダウンロード",
             data=markdown_text,
             file_name="selected_files.md",
             mime="text/markdown"