Update app.py

app.py

@@ -8,6 +8,23 @@ from config.settings import Settings
 from core.file_scanner import FileScanner, FileInfo
 from services.llm_service import LLMService
 from typing import List, Set
+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)
 
 st.set_page_config(
    page_title="Repository Code Analysis",
@@ -15,7 +32,6 @@ st.set_page_config(
    layout="wide"
 )
 
-# ダークテーマの設定
 st.markdown("""
 <style>
     .stApp {
@@ -31,29 +47,34 @@ st.markdown("""
         background-color: #1e2329;
         color: #ffffff;
     }
+    .directory-structure {
+        font-family: monospace;
+        white-space: pre;
+        background-color: #1e2329;
+        padding: 1rem;
+        border-radius: 0.5rem;
+        margin: 1rem 0;
+    }
+    .markdown-preview {
+        background-color: #1e2329;
+        padding: 1rem;
+        border-radius: 0.5rem;
+        margin: 1rem 0;
+        max-height: 500px;
+        overflow-y: auto;
+    }
 </style>
 """, unsafe_allow_html=True)
 
-def create_download_content(files: List[FileInfo]) -> str:
-    content = "# スキャン結果\n\n"
-    for file in files:
-        content += f"## {file.path}\n"
-        content += f"サイズ: {file.formatted_size}\n"
-        content += f"エンコーディング: {file.encoding or '不明'}\n\n"
-        if file.content:
-            content += f"```{file.extension[1:] if file.extension else ''}\n"
-            content += file.content
-            content += "\n```\n\n"
-    return content
-
-def clone_repository(repo_url: str) -> Path:
-    temp_dir = Path(tempfile.mkdtemp())
-    git.Repo.clone_from(repo_url, temp_dir)
-    return temp_dir
-
 # セッション状態の初期化
 if 'repo_content' not in st.session_state:
     st.session_state.repo_content = None
+if 'directory_structure' not in st.session_state:
+    st.session_state.directory_structure = None
+if 'content_md' not in st.session_state:
+    st.session_state.content_md = None
+if 'structure_md' not in st.session_state:
+    st.session_state.structure_md = None
 if 'temp_dir' not in st.session_state:
     st.session_state.temp_dir = None
 if 'llm_service' not in st.session_state:
@@ -67,83 +88,72 @@ 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. スキャン対象の拡張子を選択
-    2. GitHubリポジトリのURLを入力
-    3. スキャンを実行
-    4. コードについて質問（最大5ターンの会話が可能）
+    1. GitHubリポジトリのURLを入力
+    2. スキャンを実行
+    3. ディレクトリ構造と内容を確認
+    4. Markdownファイルをダウンロード
+    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}
-    
-    # 選択された拡張子の集合を作成
-    selected_extensions = {ext for exts in [selected_prog, selected_config, selected_doc]
-                         for ext, selected in exts.items() if selected}
 
-# 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('リポジトリをクローン中...'):
-            temp_dir = clone_repository(repo_url)
-            st.session_state.temp_dir = temp_dir
-        
-        with st.spinner('ファイルをスキャン中...'):
-            scanner = FileScanner(temp_dir, selected_extensions)
-            files = scanner.scan_files()
+        with st.spinner('リポジトリを解析中...'):
+            generator = MarkdownGenerator(repo_url)
+            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)}個のファイルを検出")
         
-        # スキャン結果のダウンロードボタン
-        scan_result = create_download_content(files)
-        st.download_button(
-            label="スキャン結果をダウンロード",
-            data=scan_result,
-            file_name=f"scan_result_{datetime.now().strftime('%Y%m%d_%H%M%S')}.md",
-            mime="text/markdown"
-        )
-        
-        # 新しいスキャン時に会話履歴をクリア
         st.session_state.llm_service.clear_history()
         
     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(["📁 ディレクトリ構造", "📄 ファイル内容"])
+    
+    with tab1:
+        st.markdown(f'<div class="markdown-preview">{st.session_state.structure_md}</div>', 
+                   unsafe_allow_html=True)
+        
+        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 tab2:
+        st.markdown(f'<div class="markdown-preview">{st.session_state.content_md}</div>', 
+                   unsafe_allow_html=True)
+        
+        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",
+            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>', 
@@ -172,11 +182,3 @@ if st.session_state.repo_content:
                     st.error(error)
                 else:
                     st.rerun()
-
-# セッション終了時のクリーンアップ
-if st.session_state.temp_dir and Path(st.session_state.temp_dir).exists():
-    try:
-        import shutil
-        shutil.rmtree(st.session_state.temp_dir)
-    except:
-        pass