Update app.py

app.py

@@ -3,47 +3,40 @@ import random
 import pandas as pd
 
 # Load the vocabulary data
-df = pd.read_csv("toeic_vocab_test_1.csv")
-vocab_dict = dict()
-for line in df.iterrows():
-    vocab_dict[line[1]['meaning']] = (line[1]['word'])
+df = pd.read_csv("toeic_vocal_1.csv")
+vocab_dict = {row['vocab']: row['meaning'] for _, row in df.iterrows()}
 
 # Function to generate a new question
 def generate_question():
-    thai_word, correct_answer = random.choice(vocab_dict.items())
-    
-    # Generate wrong answers
-    wrong_answers = random.sample([v[0] for v in vocab_dict.values()], 4)
-    while correct_answer in wrong_answers:
-        wrong_answers = random.sample([v[0] for v in vocab_dict.values()], 4)
-    
+    thai_word, correct_answer = random.choice(list(vocab_dict.items()))
+    wrong_answers = random.sample([v for v in vocab_dict.values() if v != correct_answer], 4)
     options = wrong_answers + [correct_answer]
     random.shuffle(options)
-    
-    return thai_word, correct_answer, correct_part_of_speech, options
+    return thai_word, correct_answer, options
 
-# Initial configuration
-if 'question_data' not in st.session_state:
-    st.session_state.question_data = generate_question()
-    st.session_state.answered = False
-
-thai_word, correct_answer, options = st.session_state.question_data
+# Initialize session state
+if 'question' not in st.session_state:
+    st.session_state['question'] = generate_question()
+    st.session_state['answered'] = False
 
 # Display question
-st.write(f"คำว่า '{thai_word}' (ชนิดคำ: {correct_part_of_speech}) คือคำว่าอะไร?")
-
-# Display answer options as radio buttons
-selected_answer = st.radio("เลือกคำตอบของคุณ:", options)
-
-# Check answer and show the result
-if not st.session_state.answered and st.button("ตรวจคำตอบ"):
-    st.session_state.answered = True
-    if selected_answer == correct_answer:
-        st.success("ถูกต้อง!")
-    else:
-        st.error(f"ผิด! คำตอบที่ถูกต้องคือ: {correct_answer}")
-
-# 'Next question' button to load a new question
-if st.session_state.answered and st.button("คำถามถัดไป"):
-    st.session_state.question_data = generate_question()
-    st.session_state.answered = False
+thai_word, correct_answer, options = st.session_state['question']
+st.header("ฝึกคำศัพท์ภาษาอังกฤษ")
+st.write(f"คำว่า '{thai_word}' คือคำว่าอะไร ?")
+
+# Display answer options as buttons
+if st.session_state['answered'] == False:
+    for option in options:
+        if st.button(option):
+            if option == correct_answer:
+                st.success("ถูกต้อง!")
+            else:
+                st.error("ผิด!")
+                st.info(f"คำตอบที่ถูกต้องคือ: {correct_answer}")
+            st.session_state['answered'] = True
+
+# Next button for a new question
+if st.session_state['answered']:
+    if st.button("Next"):
+        st.session_state['question'] = generate_question()
+        st.session_state['answered'] = False