Update src/streamlit_app.py

src/streamlit_app.py

@@ -67,38 +67,15 @@ def assistant_message_style(assistant_icon_path, answer):
         <div class="chat-bubble assistant-bubble">{answer}</div>
     </div>"""
 
-for message in st.session_state.messages:
-    if message["role"] == "user":
-        st.markdown(user_message_style(message["content"]), unsafe_allow_html=True)
-    else:
-        st.markdown(assistant_message_style(assistant_icon_path, message["content"]), unsafe_allow_html=True)
+# 드롭다운 메뉴
+st.markdown("---")
+st.subheader("요약할 기사 선택")
 
-if not st.session_state.messages:
-    greeting = '신문기사 요약 모델 입니다. 사용을 위해서는 "yes"를 입력해 주세요'
-    st.session_state.messages.append({"role": "assistant", "content": greeting})
-    st.markdown(assistant_message_style(assistant_icon_path, greeting), unsafe_allow_html=True)
-
-if prompt := st.chat_input():
-    # 사용자 질문 세션에 저장
-    st.session_state.messages.append({"role": "user", "content": prompt})
-    
-    # 사용자 질문 화면에 표시
-    st.markdown(user_message_style(prompt), unsafe_allow_html=True)
-    
-    # 인증 
-    if st.session_state.approval_state == "require":
-        if prompt.lower() == "yes":
-            st.session_state.approval_state = "approved"
-            response = "이제 사용이 가능합니다. 단, 모델 크기가 작은 모델이여서 성능이 기대에 못미칠 수 있습니다."
-        else:
-            response = "인증에 실패하였습니다. 다시 yes를 입력해 주세요"
-    # 응답 생성
-    else:
-        result = re.sub(r'\s+', ' ', prompt)
-        summarized = pipe(result)
-        response = summarized[0]["summary_text"]
-    
-    # 어시스턴스의 메세지를 세션에 저장
-    st.session_state.messages.append({"role": "assistant", "content": response})
-    # 어시스턴스의 메세지를 화면에 표시
-    st.markdown(assistant_message_style(assistant_icon_path, response), unsafe_allow_html=True)
+options = [
+    "",
+    "yes",
+    "대한민국은 최근 AI 기술에 대한 관심이 급증하고 있다. 많은 기업이 AI 도입을 가속화하며 디지털 전환에 박차를 가하고 있다.",
+    "오늘 열린 G7 정상회의에서는 글로벌 경제 회복과 기후 변화 대응을 위한 논의가 이어졌다.",
+    "한국은행은 기준금리를 동결하면서도 물가 상승에 대한 우려를 나타냈다."
+]
+selected_text = st.selectbox("기사 문장을 선택하세요", options)