Update app.py

app.py

@@ -1,7 +1,7 @@
 import streamlit as st
 import pandas as pd
 import numpy as np
-import anthropic
+from anthropic import Anthropic, HUMAN_PROMPT, AI_PROMPT
 import json
 from sklearn.feature_extraction.text import CountVectorizer, TfidfVectorizer
 from sklearn.decomposition import LatentDirichletAllocation
@@ -87,7 +87,63 @@ def plot_network_graph(G):
     
     return img_bytes
 
-# 헤더 스타일 변경
+def perform_topic_modeling(df, text_column, num_topics, stop_words):
+    df['processed_text'] = df[text_column].apply(lambda x: preprocess_text(x, stop_words))
+    
+    vectorizer = CountVectorizer(max_df=0.95, min_df=2)
+    doc_term_matrix = vectorizer.fit_transform(df['processed_text'])
+
+    lda = LatentDirichletAllocation(n_components=num_topics, random_state=42)
+    lda_output = lda.fit_transform(doc_term_matrix)
+
+    tfidf_vectorizer = TfidfVectorizer(max_df=0.95, min_df=2)
+    tfidf_matrix = tfidf_vectorizer.fit_transform(df['processed_text'])
+
+    feature_names = vectorizer.get_feature_names_out()
+
+    topic_results = []
+
+    for idx, topic in enumerate(lda.components_):
+        # LDA 상위 단어를 점수 순으로 정렬
+        lda_top_words = sorted([(feature_names[i], topic[i]) for i in range(len(topic))], key=lambda x: x[1], reverse=True)[:10]
+        topic_docs = lda_output[:, idx].argsort()[::-1][:100]
+        topic_tfidf = tfidf_matrix[topic_docs].mean(axis=0).A1
+        # TF-IDF 상위 단어도 점수 순으로 정렬
+        tfidf_top_words = sorted([(feature_names[i], topic_tfidf[i]) for i in range(len(topic_tfidf))], key=lambda x: x[1], reverse=True)[:10]
+        weight = lda_output[:, idx].mean() * 100
+        
+        topic_name = ", ".join([word for word, _ in lda_top_words[:5]])
+        
+        topic_results.append({
+            'topic_num': idx + 1,
+            'topic_name': topic_name,
+            'lda_words': [word for word, _ in lda_top_words],
+            'tfidf_words': [word for word, _ in tfidf_top_words],
+            'weight': weight
+        })
+
+    return topic_results, lda, lda_output, tfidf_matrix, feature_names
+
+def perform_conditional_analysis(df, condition_column, text_column, num_topics, stop_words, is_numeric, condition):
+    if is_numeric:
+        if isinstance(condition, tuple) and condition[0] == "이상":
+            filtered_df = df[df[condition_column] >= condition[1]]
+        elif isinstance(condition, tuple) and condition[0] == "이하":
+            filtered_df = df[df[condition_column] <= condition[1]]
+        else:  # 범위 선택
+            filtered_df = df[(df[condition_column] >= condition[0]) & (df[condition_column] <= condition[1])]
+        topic_results, _, _, _, _ = perform_topic_modeling(filtered_df, text_column, num_topics, stop_words)
+        return {f"{condition_column} {condition}": topic_results}
+    else:
+        results = {}
+        for value in condition:
+            filtered_df = df[df[condition_column] == value]
+            if len(filtered_df) > 0:
+                topic_results, _, _, _, _ = perform_topic_modeling(filtered_df, text_column, num_topics, stop_words)
+                results[value] = topic_results
+        return results
+
+# 스타일 설정
 st.markdown("""
 <style>
     .css-1adrfps {
@@ -131,9 +187,14 @@ with st.sidebar:
     
     stop_words_input = st.text_area("불용어 목록 (쉼표로 구분)", ', '.join(default_stop_words))
     stop_words = [word.strip() for word in stop_words_input.split(',') if word.strip()]
+
+    st.caption("결과를 보고 업데이트해주세요.")
     
     uploaded_file = st.file_uploader("CSV 파일을 업로드하세요", type="csv")
 
+    st.caption("csv-UTF형식을 사용해주세요!")
+
+# 데이터 로드 및 초기 설정
 if uploaded_file is not None:
     try:
         df = pd.read_csv(uploaded_file)
@@ -147,10 +208,40 @@ if uploaded_file is not None:
             text_column = st.selectbox("텍스트 컬럼을 선택하세요", df.columns)
             num_topics = st.slider("토픽 수를 선택하세요", 2, 20, 5)
             
+            # 분석 방법 선택
+            analysis_type = st.radio("분석 방법 선택", ["전체 분석", "조건부 분석"])
+            
+            if analysis_type == "조건부 분석":
+                condition_column = st.selectbox("조건부 분석에 사용할 변수를 선택하세요", df.columns)
+                
+                if pd.api.types.is_numeric_dtype(df[condition_column]):
+                    min_val, max_val = df[condition_column].min(), df[condition_column].max()
+                    st.write(f"{condition_column}의 범위: {min_val:.2f} ~ {max_val:.2f}")
+                    
+                    analysis_method = st.radio("분석 방법 선택", ["범위 선택", "임계값 기준"])
+                    
+                    if analysis_method == "범위 선택":
+                        condition = st.slider(f"{condition_column} 범위 선택", float(min_val), float(max_val), (float(min_val), float(max_val)))
+                    else:  # 임계값 기준
+                        threshold = st.number_input(f"{condition_column} 임계값 설정", min_value=float(min_val), max_value=float(max_val), value=float((min_val + max_val) / 2))
+                        comparison = st.radio("비교 기준", ["이상", "이하"])
+                        condition = (comparison, threshold)
+                    
+                    is_numeric = True
+                else:
+                    unique_values = df[condition_column].unique()
+                    condition = st.multiselect(f"{condition_column} 값 선택", unique_values, default=unique_values)
+                    is_numeric = False
+            
             if st.button("토픽 모델링 실행"):
                 st.session_state.run_analysis = True
                 st.session_state.text_column = text_column
                 st.session_state.num_topics = num_topics
+                st.session_state.analysis_type = analysis_type
+                if analysis_type == "조건부 분석":
+                    st.session_state.condition_column = condition_column
+                    st.session_state.condition = condition
+                    st.session_state.is_numeric = is_numeric
             else:
                 st.session_state.run_analysis = False
     except pd.errors.EmptyDataError:
@@ -162,116 +253,74 @@ if uploaded_file is not None:
 else:
     st.info("CSV 파일을 업로드해주세요.")
 
+# 메인 분석 로직
 if 'run_analysis' in st.session_state and st.session_state.run_analysis:
-    if 'text_column' in st.session_state and 'num_topics' in st.session_state:
+    if 'text_column' in st.session_state and 'num_topics' in st.session_state and 'analysis_type' in st.session_state:
         try:
-            with st.spinner("텍스트 전처리 중..."):
-                df['processed_text'] = df[st.session_state.text_column].apply(lambda x: preprocess_text(x, stop_words))
+            if st.session_state.analysis_type == "전체 분석":
+                st.header("전체 데이터 분석 결과")
+                
+                with st.spinner("토픽 모델링 실행 중..."):
+                    topic_results, lda, lda_output, tfidf_matrix, feature_names = perform_topic_modeling(
+                        df, st.session_state.text_column, st.session_state.num_topics, stop_words
+                    )
 
-            with st.spinner("토픽 모델링 실행 중..."):
-                vectorizer = CountVectorizer(max_df=0.95, min_df=2)
-                doc_term_matrix = vectorizer.fit_transform(df['processed_text'])
+                # 토픽 요약 표시
+                topic_summary = ", ".join([f"토픽{info['topic_num']}({info['topic_name']}, {info['weight']:.1f}%)" for info in topic_results])
+                st.markdown(f'<div class="topic-summary">{topic_summary}</div>', unsafe_allow_html=True)
 
-                lda = LatentDirichletAllocation(n_components=st.session_state.num_topics, random_state=42)
-                lda_output = lda.fit_transform(doc_term_matrix)
+                # 토픽별 상세 정보 표시
+                for idx, topic_info in enumerate(topic_results):
+                    st.subheader(f"토픽 {idx + 1}")
+                    
+                    col1, col2 = st.columns(2)
+                    
+                    with col1:
+                        df_lda = pd.DataFrame(list(zip(topic_info['lda_words'], lda.components_[idx][np.argsort(lda.components_[idx])[::-1][:10]])), 
+                                              columns=['단어', 'LDA 점수'])
+                        st.subheader("LDA 상위 단어")
+                        st.table(df_lda.style.format({'LDA 점수': '{:.4f}'}))
+                    
+                    with col2:
+                        df_tfidf = pd.DataFrame(list(zip(topic_info['tfidf_words'], 
+                                                         tfidf_matrix[lda_output[:, idx].argsort()[::-1][:100]].mean(axis=0).A1[np.argsort(tfidf_matrix[lda_output[:, idx].argsort()[::-1][:100]].mean(axis=0).A1)[::-1][:10]])), 
+                                                columns=['단어', 'TF-IDF'])
+                        st.subheader("TF-IDF 상위 단어")
+                        st.table(df_tfidf.style.format({'TF-IDF': '{:.4f}'}))
+
+                # 토픽 비중 그래프
+                st.header("토픽 비중 그래프")
+                df_weights = pd.DataFrame({
+                    '토픽': [f'토픽 {i+1}' for i in range(st.session_state.num_topics)],
+                    '비중': [result['weight'] for result in topic_results]
+                })
 
-                tfidf_vectorizer = TfidfVectorizer(max_df=0.95, min_df=2)
-                tfidf_matrix = tfidf_vectorizer.fit_transform(df['processed_text'])
+                colors = generate_colors(st.session_state.num_topics)
 
-            st.header("토픽 모델링 결과")
-            feature_names = vectorizer.get_feature_names_out()
+                chart = alt.Chart(df_weights).mark_bar().encode(
+                    x=alt.X('토픽:N', axis=alt.Axis(labelAngle=0)),
+                    y=alt.Y('비중:Q', axis=alt.Axis(format=',.1f')),
+                    color=alt.Color('토픽:N', scale=alt.Scale(range=colors))
+                ).properties(
+                    width=600,
+                    height=400,
+                    title='문서 내 토픽 비중 (%)'
+                )
 
-            topic_results = []
+                text = chart.mark_text(
+                    align='center',
+                    baseline='bottom',
+                    dy=-5
+                ).encode(
+                    text=alt.Text('비중:Q', format='.1f')
+                )
 
-            for idx, topic in enumerate(lda.components_):
-                lda_top_words = [(feature_names[i], topic[i]) for i in topic.argsort()[:-11:-1]]
-                topic_docs = lda_output[:, idx].argsort()[::-1][:100]
-                topic_tfidf = tfidf_matrix[topic_docs].mean(axis=0).A1
-                tfidf_top_words = [(feature_names[i], topic_tfidf[i]) for i in topic_tfidf.argsort()[:-11:-1]]
-                weight = lda_output[:, idx].mean() * 100
-                
-                topic_name = ", ".join([word for word, _ in lda_top_words[:5]])
-                
-                topic_results.append({
-                    'topic_num': idx + 1,
-                    'topic_name': topic_name,
-                    'lda_words': [word for word, _ in lda_top_words],
-                    'tfidf_words': [word for word, _ in tfidf_top_words],
-                    'weight': weight
-                })
-            
-            topic_summary = ", ".join([f"토픽{info['topic_num']}({info['topic_name']}, {info['weight']:.1f}%)" for info in topic_results])
-            st.markdown(f'<div class="topic-summary">{topic_summary}</div>', unsafe_allow_html=True)
+                st.altair_chart(chart + text, use_container_width=True)
 
-            for idx, topic in enumerate(lda.components_):
-                st.subheader(f"토픽 {idx + 1}")
-                
-                col1, col2 = st.columns(2)
-                
-                with col1:
-                    lda_top_words = [(feature_names[i], topic[i]) for i in topic.argsort()[:-11:-1]]
-                    df_lda = pd.DataFrame(lda_top_words, columns=['단어', 'LDA 점수'])
-                    st.subheader("LDA 상위 단어")
-                    st.table(df_lda.style.format({'LDA 점수': '{:.4f}'}))
-                
-                with col2:
-                    topic_docs = lda_output[:, idx].argsort()[::-1][:100]
-                    topic_tfidf = tfidf_matrix[topic_docs].mean(axis=0).A1
-                    tfidf_top_words = [(feature_names[i], topic_tfidf[i]) for i in topic_tfidf.argsort()[:-11:-1]]
-                    df_tfidf = pd.DataFrame(tfidf_top_words, columns=['단어', 'TF-IDF'])
-                    st.subheader("TF-IDF 상위 단어")
-                    st.table(df_tfidf.style.format({'TF-IDF': '{:.4f}'}))
-
-            st.header("토픽 비중 그래프")
-            df_weights = pd.DataFrame({
-                '토픽': [f'토픽 {i+1}' for i in range(st.session_state.num_topics)],
-                '비중': [result['weight'] for result in topic_results]
-            })
-
-            colors = generate_colors(st.session_state.num_topics)
-
-            base = alt.Chart(df_weights).encode(
-                x=alt.X('토픽:N', axis=alt.Axis(labelAngle=0)),
-                y=alt.Y('비중:Q', axis=alt.Axis(format=',.1f'))
-            )
-
-            bars = base.mark_bar().encode(
-                color=alt.Color('토픽:N', scale=alt.Scale(range=colors))
-            )
-
-            text = base.mark_text(
-                align='center',
-                baseline='middle',
-                dy=-10  
-            ).encode(
-                text=alt.Text('비중:Q', format='.1f')
-            )
-
-            chart = (bars + text).properties(
-                width=600,
-                height=400,
-                title='문서 내 토픽 비중 (%)'
-            ).configure_axis(
-                labelFontSize=12,
-                titleFontSize=14
-            ).configure_title(
-                fontSize=16,
-                font='Arial',
-                anchor='middle',
-                color='gray'
-            )
-
-            st.altair_chart(chart, use_container_width=True)
-
-            st.header("토픽 단어 네트워크 그래프")
-            try:
-                if 'network_graph' not in st.session_state:
-                    G = create_network_graph(topic_results, num_words=20)
-                    img_bytes = plot_network_graph(G)
-                    st.session_state.network_graph = img_bytes
-                else:
-                    img_bytes = st.session_state.network_graph
-                
+                # 네트워크 그래프
+                st.header("토픽 단어 네트워크 그래프")
+                G = create_network_graph(topic_results, num_words=20)
+                img_bytes = plot_network_graph(G)
                 st.image(img_bytes, caption="토픽별 상위 20개 단어 네트워크", use_column_width=True)
                 
                 st.download_button(
@@ -281,94 +330,176 @@ if 'run_analysis' in st.session_state and st.session_state.run_analysis:
                     mime="image/png",
                     key="download_graph"
                 )
-            except Exception as e:
-                st.error(f"네트워크 그래프 생성 중 오류가 발생했습니다: {str(e)}")
-
-            # 토픽 요약 테이블 (항상 표시)
-            st.subheader("토픽 요약 테이블")
-            topic_summary_df = pd.DataFrame([
-                {
-                    '토픽 번호': f"토픽{info['topic_num']}",
-                    '비중': f"{info['weight']:.1f}%",
-                    'LDA 상위 단어': ", ".join(info['lda_words'][:10]),
-                    'TF-IDF 상위 단어': ", ".join(info['tfidf_words'][:10])
-                } for info in topic_results
-            ])
-            st.table(topic_summary_df)
-
-            def interpret_topics(api_key, topic_results):
-                client = anthropic.Anthropic(api_key=api_key)
-                
-                prompt = f"""다음은 LDA 토픽 모델링 결과로 나온 각 토픽의 정보입니다. 이를 바탕으로 전체 토픽을 종합적으로 해석해주세요:
+
+                # 토픽 요약 테이블
+                st.subheader("토픽 요약 테이블")
+                topic_summary_df = pd.DataFrame([
+                    {
+                        '토픽 번호': f"토픽{info['topic_num']}",
+                        '비중': f"{info['weight']:.1f}%",
+                        'LDA 상위 단어': ", ".join(info['lda_words'][:10]),
+                        'TF-IDF 상위 단어': ", ".join(info['tfidf_words'][:10])
+                    } for info in topic_results
+                ])
+                st.table(topic_summary_df)
+
+            elif st.session_state.analysis_type == "조건부 분석":
+                st.header("조건부 분석 결과")
                 
-                {", ".join([f"토픽 {{info['topic_num']}} (비중: {{info['weight']:.1f}}%)" for info in topic_results])}
-
-                각 토픽의 주요 단어:
-                """
-                for info in topic_results:
-                    prompt += f"""
-                토픽 {info['topic_num']} (비중: {info['weight']:.1f}%):
-                LDA 상위 단어: {', '.join(info['lda_words'][:10])}
-                TF-IDF 상위 단어: {', '.join(info['tfidf_words'][:10])}
-                """
+                with st.spinner("조건부 토픽 모델링 실행 중..."):
+                    conditional_results = perform_conditional_analysis(
+                        df, st.session_state.condition_column, st.session_state.text_column, 
+                        st.session_state.num_topics, stop_words, st.session_state.is_numeric, 
+                        st.session_state.condition
+                    )
                 
-                prompt += """
-                위 정보를 바탕으로 다음 형식에 맞춰 답변해주세요:
-
-                1. 전체 문서의 주제 요약 (3-4문장):
-                   [여기에 전체 문서의 주제를 종합적으로 설명해주세요. 각 토픽의 비중을 고려하여 중요도를 반영해주세요.]
-
-                2. 각 토픽 요약:
-                   [각 토픽에 대해 다음 형식으로 요약해주세요]
-                   토픽[번호] "[토픽명]" [비중]%
-                   • LDA 상위 단어 10개: [LDA 상위 단어 10개를 쉼표로 구분하여 나열]
-                   • TF-IDF 상위 단어 10개: [TF-IDF 상위 단어 10개를 쉼표로 구분하여 나열]
-                   • 토픽명 설명: [토픽명이 이렇게 지어진 이유를 1-2문장으로 설명해주세요. LDA와 TF-IDF 상위 단어들이 어떻게 이 토픽명과 연관되는지 설명하세요.]
-                   • 토픽 설명: [2-3문장으로 토픽의 전반적인 내용을 설명해주세요.]
-
-                주의사항:
-                1. 토픽명은 "[구체적인 토픽명]" 형식으로 작성해주세요. 반드시 8어절 이상으로 구체적이고 설명적으로 작성해야 합니다. 
-                   예시: "구성원의 전문성 향상을 위한 체계적인 학습과 역량 개발 방안 모색", "조직의 장기적 성과 향상을 위한 핵심 학습 역량 강화 전략", 
-                   "현재 컬리지 멤버들의 역할 고민과 향후 발전 방향에 대한 논의" 등
-                2. 토픽명은 단순히 단어를 나열하는 것이 아니라, 토픽의 핵심 주제나 의미를 잘 나타내는 구체적이고 설명적인 문구로 만들어주세요.
-                3. 토픽명 설명에서는 왜 그러한 토픽명이 선택되었는지, LDA와 TF-IDF 상위 단어들과의 연관성을 구체적으로 설명해주세요.
-                4. 각 토픽의 LDA 상위 단어와 TF-IDF 상위 단어 10개를 반드시 포함해주세요.
-
-                위 형식에 ��춰 답변해주세요. 사용자가 쉽게 복사하여 사용할 수 있도록 간결하고 명확하게 작성해주세요.
-                """
-
-                try:
-                    response = client.messages.create(
-                        model="claude-3-sonnet-20240229",
-                        max_tokens=4000,
-                        temperature=0,
-                        system="당신은 토픽 모델링과 텍스트 분석 전문가입니다. 토픽 모델링 결과에 대해 명확하고 간결하며 상세한 해석을 제공합니다. 모든 응답은 반드시 한국어로만 작성해야 합니다.",
-                        messages=[
-                            {"role": "user", "content": prompt}
-                        ]
+                for value, topic_results in conditional_results.items():
+                    st.subheader(f"{st.session_state.condition_column}: {value}")
+                    
+                    # 토픽 요약 표시
+                    topic_summary = ", ".join([f"토픽{info['topic_num']}({info['topic_name']}, {info['weight']:.1f}%)" for info in topic_results])
+                    st.markdown(f'<div class="topic-summary">{topic_summary}</div>', unsafe_allow_html=True)
+                    
+                    # 토픽 요약 테이블
+                    topic_summary_df = pd.DataFrame([
+                        {
+                            '토픽 번호': f"토픽{info['topic_num']}",
+                            '비중': f"{info['weight']:.1f}%",
+                            'LDA 상위 단어': ", ".join(info['lda_words'][:10]),
+                            'TF-IDF 상위 단어': ", ".join(info['tfidf_words'][:10])
+                        } for info in topic_results
+                    ])
+                    st.table(topic_summary_df)
+                    
+                    # 네트워크 그래프
+                    st.subheader(f"{value} - 토픽 단어 네트워크 그래프")
+                    G = create_network_graph(topic_results, num_words=20)
+                    img_bytes = plot_network_graph(G)
+                    st.image(img_bytes, caption=f"{value} - 토픽별 상위 20개 단어 네트워크", use_column_width=True)
+                    
+                    st.download_button(
+                        label=f"{value} - 네트워크 그래프 다운로드",
+                        data=img_bytes,
+                        file_name=f"topic_network_graph_{value}.png",
+                        mime="image/png",
+                        key=f"download_graph_{value}"
                     )
-                    return response.content[0].text
-                except Exception as e:
-                    return f"Claude API 호출 중 오류가 발생했습니다: {str(e)}"
-   
+                    
+                    st.markdown("---")  # 각 카테고리 결과 사이에 구분선 추가
+
             # Claude API를 사용한 토픽 해석 부분
             if api_key:
-                st.header("토픽 종합 해석/8월 28일 코드 일부 수정 중입니다:)")
+                st.header("토픽 종합 해석")
                 
-                col1, col2 = st.columns([3, 1])
-                with col2:
-                    if st.button("토픽 다시 해석하기", key="reinterpret"):
-                        # 해석 결과만 초기화
-                        st.session_state.topic_interpretation = None
-                
-                with col1:
-                    # 해석 결과가 없거나 다시 해석하기 버튼을 눌렀을 때만 새로 해석
-                    if 'topic_interpretation' not in st.session_state or st.session_state.topic_interpretation is None:
-                        with st.spinner("토픽 해석 중..."):
-                            st.session_state.topic_interpretation = interpret_topics(api_key, topic_results)
+                def interpret_topics_full(api_key, topic_results):
+                    anthropic = Anthropic(api_key=api_key)
+                    
+                    prompt = f"""다음은 LDA 토픽 모델링 결과로 나온 각 토픽의 정보입니다. 이를 바탕으로 전체 토픽을 종합적으로 해석해주세요:
+                    
+                    {", ".join([f"토픽 {{info['topic_num']}} (비중: {{info['weight']:.1f}}%)" for info in topic_results])}
+                    각 토픽의 주요 단어:
+                    """
+                    for info in topic_results:
+                        prompt += f"""
+                    토픽 {info['topic_num']} (비중: {info['weight']:.1f}%):
+                    LDA 상위 단어: {', '.join(info['lda_words'][:10])}
+                    TF-IDF 상위 단어: {', '.join(info['tfidf_words'][:10])}
+                    """
                     
-                    st.subheader("토픽 모델링 종합 결과")
-                    st.text_area("결과를 복사하여 사용하세요:", value=st.session_state.topic_interpretation, height=500)
+                    prompt += """
+                    �� 정보를 바탕으로 다음 형식에 맞춰 답변해주세요:
+                    1. 전체 문서의 주제 요약 (3-4문장):
+                       [여기에 전체 문서의 주제를 종합적으로 설명해주세요. 각 토픽의 비중을 고려하여 중요도를 반영해주세요.]
+                    2. 각 토픽 요약:
+                       [각 토픽에 대해 다음 형식으로 요약해주세요]
+                       토픽[번호] "[토픽명]" [비중]%
+                       • LDA 상위 단어 10개: [LDA 상위 단어 10개를 쉼표로 구분하여 나열]
+                       • TF-IDF 상위 단어 10개: [TF-IDF 상위 단어 10개를 쉼표로 구분하여 나열]
+                       • 토픽명 설명: [토픽명이 이렇게 지어진 이유를 1-2문장으로 설명해주세요. LDA와 TF-IDF 상위 단어들이 어떻게 이 토픽명과 연관되는지 설명하세요.]
+                       • 토픽 설명: [2-3문장으로 토픽의 전반적인 내용을 설명해주세요.]
+                    주의사항:
+                    1. 토픽명은 "[구체적인 토픽명]" 형식으로 작성해주세요. 반드시 8어절 이상으로 구체적이고 설명적으로 작성해야 합니다. 
+                    2. 예시: "구성원의 전문성 향상을 위한 체계적인 학습과 역량 개발 방안 모색", "조직의 장기적 성과 향상을 위한 핵심 학습 역량 강화 전략", 
+                       "현재 컬리지 멤버들의 역할 고민과 향후 발전 방향에 대한 논의" 등
+                    3. 토픽명은 단순히 단어를 나열하는 것이 아니라, 토픽의 핵심 주제나 의미를 잘 나타내는 구체적이고 설명적인 문구로 만들어주세요.
+                    4. 각 토픽의 LDA 상위 단어와 TF-IDF 상위 단어 10개를 반드시 포함해주세요.
+                    위 형식에 맞춰 답변해주세요. 사용자가 쉽게 복사하여 사용할 수 있도록 간결하고 명확하게 작성해주세요.
+                    """
+
+                    try:
+                        completion = anthropic.completions.create(
+                            model="claude-2.1",
+                            max_tokens_to_sample=4000,
+                            prompt=f"{HUMAN_PROMPT} {prompt} {AI_PROMPT}",
+                        )
+                        return completion.completion
+                    except Exception as e:
+                        return f"Claude API 호출 중 오류가 발생했습니다: {str(e)}"
+
+                def interpret_topics_conditional(api_key, topic_results, condition):
+                    anthropic = Anthropic(api_key=api_key)
+                    
+                    prompt = f"""다음은 LDA 토픽 모델링 결과로 나온 각 토픽의 정보입니다. 이를 바탕으로 각 토픽에 대해 간략히 요약해주세요:
+                    
+                    조건: {condition}
+                    {", ".join([f"토픽 {{info['topic_num']}} (비중: {{info['weight']:.1f}}%)" for info in topic_results])}
+                    각 토픽의 주요 단어:
+                    """
+                    for info in topic_results:
+                        prompt += f"""
+                    토픽 {info['topic_num']} (비중: {info['weight']:.1f}%):
+                    LDA 상위 단어: {', '.join(info['lda_words'][:10])}
+                    TF-IDF 상위 단어: {', '.join(info['tfidf_words'][:10])}
+                    """
+                    
+                    prompt += """
+                    위 정보를 바탕으로 다음 형식에 맞춰 답변해주세요:
+                    각 토픽 요약:
+                    토픽[번호] "[토픽명]" [비중]%
+
+                    주의사항:
+                    1. 토픽명은 "[구체적인 토픽명]" 형식으로 작성해주세요. 반드시 8어절 이상으로 구체적이고 설명적으로 작성해야 합니다. 
+                    2. 예시: "구성원의 전문성 향상을 위한 체계적인 학습과 역량 개발 방안 모색", "조직의 장기적 성과 향상을 위한 핵심 학습 역량 강화 전략", 
+                       "현재 컬리지 멤버들의 역할 고민과 향후 발전 방향에 대한 논의" 등
+                    3. 토픽명은 단순히 단어를 나열하는 것이 아니라, 토픽의 핵심 주제나 의미를 잘 나타내는 구체적이고 설명적인 문구로 만들어주세요.
+
+                    위 형식에 맞춰 각 토픽에 대해 간략히 요약해주세요. 사용자가 쉽게 복사하여 사용할 수 있도록 간결하고 명확하게 작성해주세요.
+                    """
+
+                    try:
+                        completion = anthropic.completions.create(
+                            model="claude-2.1",
+                            max_tokens_to_sample=2000,
+                            prompt=f"{HUMAN_PROMPT} {prompt} {AI_PROMPT}",
+                        )
+                        return completion.completion
+                    except Exception as e:
+                        return f"Claude API 호출 중 오류가 발생했습니다: {str(e)}"
+
+                if st.session_state.analysis_type == "전체 분석":
+                    col1, col2 = st.columns([3, 1])
+                    with col2:
+                        if st.button("토픽 다시 해석하기", key="reinterpret"):
+                            st.session_state.topic_interpretation = None
+
+                    with col1:
+                        if 'topic_interpretation' not in st.session_state or st.session_state.topic_interpretation is None:
+                            with st.spinner("토픽 해석 중..."):
+                                st.session_state.topic_interpretation = interpret_topics_full(api_key, topic_results)
+
+                        st.subheader("토픽 모델링 종합 결과")
+                        st.text_area("결과를 복사하여 사용하세요:", value=st.session_state.topic_interpretation, height=500)
+                else:
+                    for value, topic_results in conditional_results.items():
+                        st.subheader(f"{st.session_state.condition_column}: {value} - 토픽 해석")
+                        if st.button(f"{value} - 토픽 다시 해석하기", key=f"reinterpret_{value}"):
+                            st.session_state[f'topic_interpretation_{value}'] = None
+
+                        if f'topic_interpretation_{value}' not in st.session_state or st.session_state[f'topic_interpretation_{value}'] is None:
+                            with st.spinner(f"{value} 토픽 해석 중..."):
+                                st.session_state[f'topic_interpretation_{value}'] = interpret_topics_conditional(api_key, topic_results, condition=f"{st.session_state.condition_column}: {value}")
+
+                        st.text_area(f"{value} - 결과를 복사하여 사용하세요:", value=st.session_state[f'topic_interpretation_{value}'], height=300)
             else:
                 st.warning("Claude API 키가 설정되지 않았습니다. https://console.anthropic.com/settings/keys 에 접속하여 API 키를 발급받으시면 토픽명과 해석을 제공받으실 수 있습니다.")
 
@@ -377,7 +508,14 @@ if 'run_analysis' in st.session_state and st.session_state.run_analysis:
     else:
         st.error("유효한 데이터가 없습니다. CSV 파일을 다시 업로드해주세요.")
 
+# 분석 초기화 버튼
+if st.button("분석 초기화"):
+    for key in st.session_state.keys():
+        if key.startswith('topic_interpretation') or key in ['run_analysis', 'run_conditional_analysis', 'conditional_topic_results']:
+            del st.session_state[key]
+    st.experimental_rerun()
+
 st.markdown("""
 ---
 © 2024 SK mySUNI 행복 College. All rights reserved. 문의사항이 있으시면 연락주세요 (배수정RF, soojeong.bae@sk.com)
-""")
+""")