Update app.py

app.py

@@ -39,11 +39,25 @@ def generate_colors(n):
     return ['#%02x%02x%02x' % tuple(int(x*255) for x in colorsys.hsv_to_rgb(*hsv)) for hsv in HSV_tuples]
 
 # Streamlit 앱 설정
-st.title('한국어 토픽 모델링 앱')
+st.set_page_config(layout="wide")  # 페이지를 와이드 모드로 설정
 
-# CSS를 사용하여 테이블 스타일 정의
+# CSS를 사용하여 헤더와 테이블 스타일 정의
 st.markdown("""
 <style>
+.header {
+    position: fixed;
+    top: 0;
+    right: 0;
+    background-color: #f1f1f1;
+    color: black;
+    text-align: right;
+    padding: 10px 20px;
+    font-size: 16px;
+    z-index: 999;
+}
+.main-content {
+    margin-top: 60px;  # 헤더의 높이만큼 여백 추가
+}
 .stDataFrame {
     width: 100%;
 }
@@ -59,6 +73,19 @@ st.markdown("""
 </style>
 """, unsafe_allow_html=True)
 
+# 헤더 HTML
+header_html = """
+<div class="header">
+    mySUNI 행복 College 행복담당조직 Meet-Up
+</div>
+"""
+
+# 헤더 렌더링
+st.markdown(header_html, unsafe_allow_html=True)
+
+st.markdown('<div class="main-content">', unsafe_allow_html=True)
+st.title('한국어 토픽 모델링 앱')
+
 # 사이드바 설정
 st.sidebar.header('설정')
 
@@ -181,6 +208,7 @@ if uploaded_file is not None:
             anthropic = Anthropic(api_key=api_key)
             
             st.header("Claude의 토픽 해석")
+            topic_interpretations = []
             for idx, topic in enumerate(lda.components_):
                 with st.spinner(f"토픽 {idx + 1} 해석 중..."):
                     lda_top_words = [feature_names[i] for i in topic.argsort()[:-11:-1]]
@@ -208,33 +236,26 @@ if uploaded_file is not None:
                     )
                     st.subheader(f"토픽 {idx + 1} 해석:")
                     st.write(response.completion)
+                    
+                    # 토픽명과 분포를 저장
+                    topic_name = response.completion.split("토픽명 제안:")[1].split("\n")[0].strip()
+                    topic_interpretations.append((idx + 1, topic_name, df_weights.iloc[idx]['퍼센트']))
+
+            # 종합 해석
+            st.header("토픽 종합 해석")
+            summary_prompt = f"{HUMAN_PROMPT} 다음은 토픽 모델링 결과로 나온 각 토픽의 이름과 분포입니다. 이를 바탕으로 전체 문서의 주제를 종합적으로 해석해주세요:\n\n"
+            for topic_num, topic_name, topic_percent in topic_interpretations:
+                summary_prompt += f"토픽 {topic_num}: {topic_name} ({topic_percent:.1f}%)\n"
+            summary_prompt += "\n위 정보를 바탕으로 전체 문서의 주제를 3-4문장으로 종합해서 설명해주세요."
+
+            summary_response = anthropic.completions.create(
+                model="claude-2.1",
+                max_tokens_to_sample=1000,
+                prompt=f"{summary_prompt}\n\n{AI_PROMPT}",
+            )
+            st.write(summary_response.completion)
+
         else:
             st.warning("Claude API 키가 설정되지 않았습니다. https://console.anthropic.com/settings/keys 에 접속하여 API 키를 발급받으시면 토픽명과 해석을 제공받으실 수 있습니다.")
 
-# CSS를 사용하여 푸터 스타일 정의
-footer_style = """
-<style>
-.footer {
-    position: fixed;
-    left: 0;
-    bottom: 0;
-    width: 100%;
-    background-color: #f1f1f1;
-    color: black;
-    text-align: center;
-    padding: 30px;
-    font-size: 20px;
-    z-index: 999;
-}
-</style>
-"""
-
-# 푸터 HTML
-footer_html = footer_style + """
-<div class="footer">
-    mySUNI 행복 College 행복담당조직 Meet-Up
-</div>
-"""
-
-# 푸터 렌더링
-st.markdown(footer_html, unsafe_allow_html=True)
+st.markdown('</div>', unsafe_allow_html=True)