Update src/streamlit_app.py

src/streamlit_app.py

@@ -2,70 +2,17 @@
 import re
 import streamlit as st
 
-
-## assitant 아바타 이미지 경로
+# assitant 아바타 이미지 경로
 assistant_icon_path = "https://huggingface.co/spaces/randmimc/aitom/resolve/main/chat_icon/assistant.ico"
 
-## 세션 상태에서 approval_state 및 messages 초기화
+# 세션 상태에서 approval_state 및 messages 초기화
 if "approval_state" not in st.session_state:
     st.session_state.approval_state = "require"
 if "messages" not in st.session_state:
     st.session_state.messages = []
 
-st.markdown("""
-    <style>
-    .chat-container { /* 채팅 메시지를 감싸는 컨테이너를 설명합니다. */
-        display: flex; /* 채팅 요소(아바타, 말풍선 등)을 가로 방향으로 정렬합니다. */
-        margin-bottom: 10px; /*메세지간 간격을 10px로 합니다. */
-    }
-    .user-container {
-        justify-content: flex-end; /* 사용자의 메시지를 우측정렬합니다.*/
-    }
-    .assistant-container {
-        justify-content: flex-start; /* 어시스턴스 메시지를 좌측 정렬합니다.*/
-        flex-direction: row; /* Avatar와 Bubble을 수평으로 정렬합니다. */
-    }
-    .chat-bubble { /* 말풍선 스타일을 정의합니다. */
-        padding: 10px; /* 말풍선 안에 메시지간 간격을 10px로 합니다. */
-        border-radius: 10px; /*말풍선의 모서리를 동글게 만듭니다.*/
-        max-width: 70%; /* 한줄이 너무 길지 않게 제한합니다. */
-        word-wrap: break-word; /* 텍스트를 단어 단위로 줄내림합니다. */
-        white-space: pre-wrap; /* 텍스트를 단어 단위로 줄내림합니다. */
-    }
-    .user-bubble { /* 사용자 메시지 스타일을 정의합니다. */
-        font-size: 14px; /* 폰트 사이즈를 지정합니다. */
-        color: #333;  /* 폰트 색상을 지정합니다. */
-        background-color: #e5e5e5; /* 말풍선 배경색을 지정합니다. */
-    }
-    .assistant-bubble {
-        font-size: 14px; /* 폰트 사이즈를 지정합니다. */
-        color: #333;  /* 폰트 색상을 지정합니다. */
-        text-align: left; /* 메세지를 왼쪽 정렬 합니다. */
-        background-color: #feeddf; /* 말풍선 배경색을 지정합니다. */
-    }
-    .avatar { /* 프로필 이미지(아바타) 스타일을 지정합니다. */
-        width: 30px; /* 아바타 폭을 지정합니다. */
-        height: 30px; /* 아바타 너을 지정합니다. */
-        border-radius: 50%; /* 아바타를 원경으로 만듭니다. */
-        margin-right: 10px; /*메세지간 간격을 10px로 합니다. */
-    }
-    </style>
-""", unsafe_allow_html=True)
-
-def user_message_style(question):
-    return f"""<div class="chat-container user-container">
-        <div class="chat-bubble user-bubble">{question}</div>
-    </div>"""
-
-def assistant_message_style(assistant_icon_path, answer):
-    return f"""<div class="chat-container assistant-container">
-        <img src="{assistant_icon_path}" class="avatar">
-        <div class="chat-bubble assistant-bubble">{answer}</div>
-    </div>"""
-
 # 드롭다운 메뉴
-st.markdown("---")
-st.subheader("요약할 기사 선택")
+st.subheader("데이터 선택")
 
 options = [
     "",
@@ -74,4 +21,21 @@ options = [
     "오늘 열린 G7 정상회의에서는 글로벌 경제 회복과 기후 변화 대응을 위한 논의가 이어졌다.",
     "한국은행은 기준금리를 동결하면서도 물가 상승에 대한 우려를 나타냈다."
 ]
-selected_text = st.selectbox("기사 문장을 선택하세요", options)
+selected_text = st.selectbox("데이터를 선택하세요", options)
+
+# 데이터 상태 및 조치 상황 표시 함수
+def display_status_and_action(text):
+    if text == "":
+        st.info("먼저 데이터를 선택해주세요.")
+    elif text == "yes":
+        st.success("✅ 데이터 상태: 승인됨")
+        st.markdown("**조치 사항:** 추가 조치는 필요하지 않습니다.")
+    else:
+        st.warning("⚠️ 데이터 상태: 승인 대기 중")
+        st.markdown("**조치 사항:** 아래 버튼을 눌러 승인 요청을 진행하세요.")
+        if st.button("승인 요청"):
+            st.session_state.approval_state = "requested"
+            st.success("승인 요청이 제출되었습니다.")
+
+# 상태 및 조치 출력
+display_status_and_action(selected_text)