Update app.py

app.py

@@ -1,32 +1,67 @@
 import streamlit as st
 import pandas as pd
 
-# 교사용 설문 설정 및 결과 확인 페이지
-st.title("강사용 설문 관리")
-
-# 설문 설정
-st.subheader("설문 설정")
-question = st.text_input("설문 질문을 입력하세요:")
-options = st.text_area("선택형 답변 항목을 입력하세요 (각 항목을 새 줄에 입력):").split('\n')
-open_ended = st.checkbox("주관식 답변도 허용하기")
-
-if st.button("설문 시작"):
-    if question and options:
-        st.session_state['question'] = question
-        st.session_state['options'] = options
-        st.session_state['open_ended'] = open_ended
-        st.session_state['responses'] = []
-        st.success(f"설문이 설정되었습니다. 학생들에게 이 URL을 공유하세요: https://huggingface.co/spaces/jonghhhh/inclass_poll_students")
+# 초기 페이지 설정 - 기본적으로 학생 응답 페이지가 먼저 보이도록 설정
+if 'page' not in st.session_state:
+    st.session_state['page'] = "학생 응답"
+
+# 사이드바에서 페이지 선택
+user_type = st.sidebar.selectbox("사용자 유형을 선택하세요:", ["학생 응답", "설문 설정", "결과 확인"], index=["학생 응답", "설문 설정", "결과 확인"].index(st.session_state['page']))
+
+# 학생 응답 페이지
+if user_type == "학생 응답":
+    st.session_state['page'] = "학생 응답"
+    st.title("학생용 설문 응답")
+
+    if 'question' in st.session_state and 'options' in st.session_state:
+        st.write(f"### {st.session_state['question']}")
+        selected_option = st.radio("답변을 선택하세요:", st.session_state['options'])
+
+        open_answer = None
+        if st.session_state['open_ended']:
+            open_answer = st.text_input("기타 의견을 입력하세요:")
+
+        if st.button("답변 제출"):
+            st.session_state['responses'].append({
+                '선택': selected_option, 
+                '주관식': open_answer if st.session_state['open_ended'] else None
+            })
+            st.success("답변이 제출되었습니다!")
     else:
-        st.error("모든 필드를 입력하세요.")
-
-# 결과 확인
-st.subheader("설문 결과 확인")
-if 'responses' in st.session_state and st.session_state['responses']:
-    df = pd.DataFrame(st.session_state['responses'])
-    st.bar_chart(df['선택'].value_counts())
-    if '주관식' in df.columns:
-        st.write("주관식 답변:")
-        st.table(df['주관식'].dropna())
-else:
-    st.write("아직 제출된 답변이 없습니다.")
+        st.warning("설문이 아직 설정되지 않았습니다. 교사에게 문의하세요.")
+
+# 설문 설정 페이지 (교사용)
+elif user_type == "설문 설정":
+    st.session_state['page'] = "설문 설정"
+    st.title("설문 설정")
+    question = st.text_input("설문 질문을 입력하세요:")
+    options = st.text_area("선택형 답변 항목을 입력하세요 (각 항목을 새 줄에 입력):").split('\n')
+    open_ended = st.checkbox("주관식 답변도 허용하기")
+
+    if st.button("설문 시작"):
+        if question and options:
+            st.session_state['question'] = question
+            st.session_state['options'] = options
+            st.session_state['open_ended'] = open_ended
+            st.session_state['responses'] = []
+            st.success("설문이 설정되었습니다.")
+        else:
+            st.error("모든 필드를 입력하세요.")
+
+# 결과 확인 페이지 (교사용)
+elif user_type == "결과 확인":
+    st.session_state['page'] = "결과 확인"
+    st.title("설문 결과 확인")
+
+    if 'responses' in st.session_state and st.session_state['responses']:
+        df = pd.DataFrame(st.session_state['responses'])
+
+        st.write("### 선택형 답변 분포")
+        st.bar_chart(df['선택'].value_counts())
+
+        if '주관식' in df.columns:
+            st.write("### 주관식 답변")
+            st.table(df['주관식'].dropna())
+    else:
+        st.warning("아직 제출된 답변이 없습니다.")
+