Update app.py

app.py

@@ -40,15 +40,6 @@ st.markdown("""
         background-color: #0e1117;
         color: #ffffff;
     }
-    .chat-message {
-        padding: 1rem;
-        margin: 1rem 0;
-        border-radius: 0.5rem;
-    }
-    .assistant-message {
-        background-color: #1e2329;
-        color: #ffffff;
-    }
     .directory-structure {
         font-family: monospace;
         white-space: pre;
@@ -57,13 +48,11 @@ st.markdown("""
         border-radius: 0.5rem;
         margin: 1rem 0;
     }
-    .markdown-preview {
+    .download-section {
         background-color: #1e2329;
         padding: 1rem;
         border-radius: 0.5rem;
         margin: 1rem 0;
-        max-height: 500px;
-        overflow-y: auto;
     }
 </style>
 """, unsafe_allow_html=True)
@@ -71,12 +60,10 @@ st.markdown("""
 # セッション状態の初期化
 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 '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:
@@ -91,7 +78,7 @@ if 'llm_service' not in st.session_state:
         st.stop()
 
 # メインのUIレイアウト
-st.title("🔍 リポジトリ解析・質問システム")
+st.title("🔍 リポジトリ解析システム")
 
 # サイドバーの設定
 with st.sidebar:
@@ -100,9 +87,8 @@ with st.sidebar:
     1. GitHubリポジトリのURLを入力
     2. スキャン対象の拡張子を選択
     3. スキャンを実行
-    4. ディレクトリ構造と内容を確認
-    5. Markdownファイルをダウンロード
-    6. コードについて質問
+    4. ディレクトリ構造を確認
+    5. 解析結果をダウンロード
     """)
     
     # スキャン対象の拡張子選択
@@ -157,68 +143,35 @@ if st.button("スキャン開始", disabled=not repo_url):
             
         st.success(f"スキャン完了: {len(files)}個のファイルを検出")
         
-        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(["📁 ディレクトリ構造", "📄 ファイル内容"])
+    # ディレクトリ構造の表示
+    st.subheader("📁 ディレクトリ構造")
+    st.markdown(f'<div class="directory-structure">{st.session_state.structure_md}</div>', 
+               unsafe_allow_html=True)
     
-    with tab1:
-        st.markdown(f'<div class="markdown-preview">{st.session_state.structure_md}</div>', 
-                   unsafe_allow_html=True)
-        
+    # ダウンロードセクション
+    st.subheader("📥 解析結果のダウンロード")
+    st.markdown('<div class="download-section">', unsafe_allow_html=True)
+    
+    col1, col2 = st.columns(2)
+    with col1:
         st.download_button(
-            label="ディレクトリ構造をダウンロード",
+            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)
-        
+    with col2:
         st.download_button(
-            label="ファイル内容をダウンロード",
+            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>', 
-                       unsafe_allow_html=True)
-    
-    query = st.text_area(
-        "質問を入力してください",
-        placeholder="例: このコードの主な機能は何ですか？"
-    )
-    
-    col1, col2 = st.columns([1, 5])
-    with col1:
-        if st.button("履歴クリア"):
-            st.session_state.llm_service.clear_history()
-            st.rerun()
-    
-    with col2:
-        if st.button("質問する", disabled=not query):
-            with st.spinner('回答を生成中...'):
-                response, error = st.session_state.llm_service.get_response(
-                    st.session_state.repo_content,
-                    query
-                )
-                
-                if error:
-                    st.error(error)
-                else:
-                    st.rerun()
+    st.markdown('</div>', unsafe_allow_html=True)