Update app.py

app.py

@@ -4,10 +4,10 @@ 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 typing import List, Set, Optional
 from main import DirectoryStructureScanner, MarkdownGenerator
 
 class StreamlitFileWriter:
@@ -26,12 +26,14 @@ class StreamlitFileWriter:
             content.append("\n")
         return "\n".join(content)
 
+# ページ設定
 st.set_page_config(
    page_title="Repository Code Analysis",
    page_icon="🔍",
    layout="wide"
 )
 
+# スタイル設定
 st.markdown("""
 <style>
     .stApp {
@@ -88,31 +90,61 @@ if 'llm_service' not in st.session_state:
         st.error(str(e))
         st.stop()
 
+# メインのUIレイアウト
 st.title("🔍 リポジトリ解析・質問システム")
 
+# サイドバーの設定
 with st.sidebar:
     st.subheader("📌 使い方")
     st.markdown("""
     1. GitHubリポジトリのURLを入力
-    2. スキャンを実行
-    3. ディレクトリ構造と内容を確認
-    4. Markdownファイルをダウンロード
-    5. コードについて質問
+    2. スキャン対象の拡張子を選択
+    3. スキャンを実行
+    4. ディレクトリ構造と内容を確認
+    5. Markdownファイルをダウンロード
+    6. コードについて質問
     """)
+    
+    # スキャン対象の拡張子選択
+    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}
 
+# URLの入力
 repo_url = st.text_input(
    "GitHubリポジトリのURLを入力",
    placeholder="https://github.com/username/repository.git"
 )
 
+# スキャン実行ボタン
 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} or Settings.DEFAULT_EXTENSIONS
+                                for ext, selected in exts.items() if selected}
+            
+            # 選択がない場合はデフォルトを使用
+            if not selected_extensions:
+                selected_extensions = Settings.DEFAULT_EXTENSIONS
             
-            # 選択された拡張子を使用してMarkdownGeneratorを初期化
+            # MarkdownGeneratorを初期化
             generator = MarkdownGenerator(repo_url, selected_extensions)
             content_md, structure_md, files = generator.generate_markdowns()
             
@@ -130,6 +162,7 @@ if st.button("スキャン開始", disabled=not repo_url):
     except Exception as e:
         st.error(f"エラーが発生しました: {str(e)}")
 
+# スキャン結果の表示
 if st.session_state.structure_md and st.session_state.content_md:
     tab1, tab2 = st.tabs(["📁 ディレクトリ構造", "📄 ファイル内容"])
     
@@ -155,10 +188,12 @@ if st.session_state.structure_md and st.session_state.content_md:
             mime="text/markdown"
         )
 
+# 質問セクション
 if st.session_state.repo_content:
     st.divider()
     st.subheader("💭 コードについて質問する")
     
+    # 会話履歴の表示
     for message in st.session_state.llm_service.conversation_history:
         if message.role == "assistant":
             st.markdown(f'<div class="chat-message assistant-message">{message.content}</div>', 
@@ -186,4 +221,4 @@ if st.session_state.repo_content:
                 if error:
                     st.error(error)
                 else:
-                    st.rerun()
+                    st.rerun()