Update app.py

app.py

@@ -1,177 +1,114 @@
+# app.py
+
 import streamlit as st
-import tempfile
-import git
-import os
 from pathlib import Path
-from datetime import datetime 
-from typing import List, Set, Optional
-from config.settings import Settings
 from core.file_scanner import FileScanner, FileInfo
-from services.llm_service import LLMService
-from main import DirectoryStructureScanner, MarkdownGenerator
 
-class StreamlitFileWriter:
-    def __init__(self, output_file: Path):
-        self.output_file = output_file
-    
-    def create_markdown(self, files: List[FileInfo]) -> str:
-        content = []
-        for file_info in files:
-            content.append(f"## {file_info.path}")
-            content.append("------------")
-            if file_info.content is not None:
-                content.append(file_info.content)
-            else:
-                content.append("# Failed to read content")
-            content.append("\n")
-        return "\n".join(content)
+# セッション状態を初期化
+if 'scanned_files' not in st.session_state:
+    st.session_state.scanned_files = []  # List[FileInfo]
+if 'selected_files' not in st.session_state:
+    st.session_state.selected_files = set()  # {Path} 相対パス or 絶対パス
 
-# ページ設定
-st.set_page_config(
-   page_title="Repository Code Analysis",
-   page_icon="🔍",
-   layout="wide"
-)
+st.title("指定拡張子のみディレクトリスキャン & 選択ダウンロード")
 
-# スタイル設定
-st.markdown("""
-<style>
-    .stApp {
-        background-color: #0e1117;
-        color: #ffffff;
-    }
-    .directory-structure {
-        font-family: monospace;
-        white-space: pre;
-        background-color: #1e2329;
-        padding: 1rem;
-        border-radius: 0.5rem;
-        margin: 1rem 0;
-    }
-    .download-section {
-        background-color: #1e2329;
-        padding: 1rem;
-        border-radius: 0.5rem;
-        margin: 1rem 0;
-    }
-</style>
-""", unsafe_allow_html=True)
+# --- 1) ユーザー入力: スキャンしたいローカルディレクトリパス ---
+dir_path_str = st.text_input("ローカルディレクトリパス", value=".")
 
-# セッション状態の初期化
-if 'repo_content' not in st.session_state:
-    st.session_state.repo_content = None
-if 'structure_md' not in st.session_state:
-    st.session_state.structure_md = None
-if 'content_md' not in st.session_state:
-    st.session_state.content_md = None
-if 'temp_dir' not in st.session_state:
-    st.session_state.temp_dir = None
-if 'llm_service' not in st.session_state:
-    try:
-        api_key = os.getenv("ANTHROPIC_API_KEY")
-        if not api_key:
-            st.error("ANTHROPIC_API_KEY環境変数が設定されていません")
-            st.stop()
-        st.session_state.llm_service = LLMService(api_key)
-    except Exception as e:
-        st.error(str(e))
-        st.stop()
+# --- 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"])):
+        chosen_exts.append(ext)
 
-# メインのUIレイアウト
-st.title("🔍 リポジトリ解析システム")
+# --- 3) 「スキャン実行」ボタン ---
+if st.button("スキャン開始"):
+    base_path = Path(dir_path_str).resolve()
+    if not base_path.is_dir():
+        st.error("有効なディレクトリを指定してください")
+    else:
+        scanner = FileScanner(base_path, set(chosen_exts))
+        found_files = scanner.scan_files()
 
-# サイドバーの設定
-with st.sidebar:
-    st.subheader("📌 使い方")
-    st.markdown("""
-    1. GitHubリポジトリのURLを入力
-    2. スキャン対象の拡張子を選択
-    3. スキャンを実行
-    4. ディレクトリ構造を確認
-    5. 解析結果をダウンロード
-    """)
-    
-    # スキャン対象の拡張子選択
-    st.subheader("🔍 スキャン対象の選択")
-    
-    # プログラミング言語
-    st.write("プログラミング言語:")
-    prog_exts = {'.py', '.js', '.ts', '.java', '.cpp', '.hpp', '.c', '.h', '.go', '.rs'}
-    selected_prog = {ext: st.checkbox(ext, value=True, key=f"prog_{ext}") 
-                    for ext in prog_exts}
-    
-    # 設定ファイル
-    st.write("設定ファイル:")
-    config_exts = {'.json', '.yml', '.yaml', '.toml'}
-    selected_config = {ext: st.checkbox(ext, value=True, key=f"config_{ext}") 
-                      for ext in config_exts}
-    
-    # ドキュメント
-    st.write("ドキュメント:")
-    doc_exts = {'.md', '.txt'}
-    selected_doc = {ext: st.checkbox(ext, value=True, key=f"doc_{ext}") 
-                   for ext in doc_exts}
+        st.session_state.scanned_files = found_files
+        st.session_state.selected_files = set()  # スキャン毎に選択リセット
 
-# URLの入力
-repo_url = st.text_input(
-   "GitHubリポジトリのURLを入力",
-   placeholder="https://github.com/username/repository.git"
-)
+        st.success(f"スキャン完了: {len(found_files)}個のファイル")
 
-# スキャン実行ボタン
-if st.button("スキャン開始", disabled=not repo_url):
-    try:
-        with st.spinner('リポジトリを解析中...'):
-            # 選択された拡張子を集約
-            selected_extensions = {ext for exts in [selected_prog, selected_config, selected_doc]
-                                for ext, selected in exts.items() if selected}
-            
-            # 選択がない場合はデフォルトを使用
-            if not selected_extensions:
-                selected_extensions = Settings.DEFAULT_EXTENSIONS
-            
-            # MarkdownGeneratorを初期化
-            generator = MarkdownGenerator(repo_url, selected_extensions)
-            content_md, structure_md, files = generator.generate_markdowns()
-            
-            writer = StreamlitFileWriter(Path("dummy"))
-            formatted_content = writer.create_markdown(files)
-            
-            st.session_state.content_md = formatted_content
-            st.session_state.structure_md = structure_md
-            st.session_state.repo_content = LLMService.format_code_content(files)
-            
-        st.success(f"スキャン完了: {len(files)}個のファイルを検出")
-        
-    except Exception as e:
-        st.error(f"エラーが発生しました: {str(e)}")
+# --- 4) ファイルを表示するUI（ディレクトリ構造 or リスト） + 全選択 / 全解除 ---
 
-# スキャン結果の表示とダウンロード
-if st.session_state.structure_md and st.session_state.content_md:
-    # ディレクトリ構造の表示
-    st.subheader("📁 ディレクトリ構造")
-    st.markdown(f'<div class="directory-structure">{st.session_state.structure_md}</div>', 
-               unsafe_allow_html=True)
-    
-    # ダウンロードセクション
-    st.subheader("📥 解析結果のダウンロード")
-    st.markdown('<div class="download-section">', unsafe_allow_html=True)
-    
+if st.session_state.scanned_files:
+    # 全選択 / 全解除ボタン
     col1, col2 = st.columns(2)
-    with col1:
-        st.download_button(
-            label="📁 ディレクトリ構造をダウンロード",
-            data=st.session_state.structure_md,
-            file_name=f"directory_structure_{datetime.now().strftime('%Y%m%d_%H%M%S')}.md",
-            mime="text/markdown"
-        )
     
+    with col1:
+        if st.button("すべて選択"):
+            # すべてのファイルを選択
+            base_path = Path(dir_path_str).resolve()
+            # relative_to で相対パスを使うか、絶対パスを使うかは好みでOK
+            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()
+
+    st.write("### ファイル一覧 (指定拡張子のみ)")
+    base_path = Path(dir_path_str).resolve()
+    
+    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)
+
+# --- 5) 選択ファイルをまとめてMarkdown化 & ダウンロード ---
+
+def create_markdown_for_selected(files, selected_paths, base_dir: Path) -> str:
+    """
+    選択されたファイルだけをMarkdownテキストとして返す
+    """
+    output = []
+    for f in files:
+        rel_path = f.path.relative_to(base_dir)
+        if rel_path in selected_paths:
+            output.append(f"## {rel_path}")
+            output.append("------------")
+            if f.content is not None:
+                output.append(f.content)
+            else:
+                output.append("# Failed to read content")
+            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()
+        markdown_text = create_markdown_for_selected(
+            st.session_state.scanned_files,
+            st.session_state.selected_files,
+            base_path
+        )
+        
         st.download_button(
-            label="📄 ファイル内容をダウンロード",
-            data=st.session_state.content_md,
-            file_name=f"repository_content_{datetime.now().strftime('%Y%m%d_%H%M%S')}.md",
+            label="ダウンロード",
+            data=markdown_text,
+            file_name="selected_files.md",
             mime="text/markdown"
         )
-    
-    st.markdown('</div>', unsafe_allow_html=True)