Update app.py

app.py

@@ -1,73 +1,55 @@
+import time
 import streamlit as st
-from random import random
-import pandas as pd
-from transformers import pipeline
+from Toeic_manager import ToeicVocabularyApp
 
-#-----------------------FUNCTIONS-----------------------
-# Đọc và xử lý dữ liệu
-@st.cache_data
-def load_data(file_path):
-    df = pd.read_csv(file_path)
-    return df
+# Khởi tạo đối tượng app từ class ToeicVocabularyApp
+app = ToeicVocabularyApp('toeic_vocabulary.csv')
 
-# Đề xuất chủ đề sử dụng Zero-shot classification
-def suggest_topics(text, classifier, topics):
-    prediction = classifier(text, topics)
-    return prediction["labels"][0]
-
-# Thiết lập pipeline cho zero-shot classification
-classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
-
-# Các chủ đề có thể
-possible_topics = ["Kinh tế", "Du lịch", "Giao tiếp", "Công nghệ", "Giáo dục"]
-
-# Load data
-data_path = "Toeic.csv"
-df = load_data(data_path)
-
-
-#----------------------------------Frontend--------------------------------
-# Giao diện người dùng
+# Thiết lập tiêu đề chính và phụ
 st.markdown("<h1 style='text-align: center;'>Chương trình ôn tập từ vựng Toeic</h1>", unsafe_allow_html=True)
 st.markdown("<h3 style='text-align: center;'>Dành cho bé iu của anh❤️</h3>", unsafe_allow_html=True)
 
-# Thêm thanh bên để chọn chế độ ôn tập
-mode = st.sidebar.selectbox(
-    "Chọn chế độ ôn tập:",
-    ["Flashcard", "Trắc nghiệm", "Thử thách tốc độ", "Ôn tập theo chủ đề", "Kiểm tra"]
+# Sidebar để chọn cài đặt ứng dụng hoặc chế độ ôn tập
+sidebar_choice = st.sidebar.selectbox(
+    "Menu",
+    ["Cài đặt ứng dụng", "Chế độ ôn tập"]
 )
 
-# Hiển thị flashcards
-if mode == "Flashcard":
-    st.subheader("Flashcard Từ vựng")
-    word = st.selectbox("Chọn từ:", df["Từ vựng"])
-    if st.button("Hiện nghĩa"):
-        definition = df[df["Từ vựng"] == word]["Nghĩa"].values[0]
-        st.write("Nghĩa của từ:", definition)
-
-# Trắc nghiệm
-elif mode == "Trắc nghiệm":
-    st.subheader("Trắc nghiệm")
-    questions = df.sample(5)
-    correct_answers = 0
-    for i, row in questions.iterrows():
-        options = df["Nghĩa"].sample(3).tolist() + [row["Nghĩa"]]
-        shuffled_options = random.sample(options, len(options))
-        user_answer = st.radio(f"{row['Từ vựng']} có nghĩa là gì?", options=shuffled_options, key=i)
-        if st.button("Kiểm tra", key=i):
-            if user_answer == row["Nghĩa"]:
-                st.success("Đúng!")
-                correct_answers += 1
-            else:
-                st.error("Sai!")
-    st.write(f"Điểm của bạn: {correct_answers}/5")
-
-# Cập nhật chủ đề nếu cần
-if st.button("Cập nhật Chủ đề cho Từ Vựng"):
-    for i, row in df.iterrows():
-        if pd.isna(row["Chủ đề"]):
-            df.at[i, "Chủ đề"] = suggest_topics(row["Từ vựng"], classifier, possible_topics)
-    df.to_csv(data_path, index=False)
-    st.success("Đã cập nhật chủ đề cho từ vựng!")
-
-st.markdown("<h3 style='text-align: center;'>Chúc bé iu của anh thi đậu nhé😘</h3>", unsafe_allow_html=True)
+# Cài đặt ứng dụng
+if sidebar_choice == "Cài đặt ứng dụng":
+    st.subheader("Cài đặt Ứng Dụng")
+    # Cho phép người dùng điều chỉnh các cài đặt
+    time_limit = st.slider("Thiết lập thời gian cho Thử thách tốc độ (giây)", min_value=10, max_value=120, value=30)
+    app.update_settings("time_limit", time_limit)
+    if st.button("Lưu Cài Đặt"):
+        st.success("Cài đặt đã được lưu thành công!")
+
+# Chế độ ôn tập
+elif sidebar_choice == "Chế độ ôn tập":
+    st.subheader("Chọn Chế độ Ôn tập")
+    study_mode = st.selectbox(
+        "Chọn chế độ",
+        app.settings["study_mode"]
+    )
+
+    if study_mode == "Thử thách tốc độ":
+        # Đếm ngược thời gian cho Thử thách tốc độ
+        st.subheader("Thử thách tốc độ")
+        st.write("Thời gian: ", app.settings["time_limit"], " giây")
+        # Đặt một nút để bắt đầu thử thách
+        if st.button("Bắt đầu"):
+            st.write("Thử thách bắt đầu!")
+            # Thêm mã cho tính năng đếm ngược ở đây
+            # Đây là một giả lập, bạn sẽ cần sử dụng thư viện khác để xử lý đếm ngược thực tế
+            countdown = app.settings["time_limit"]
+            for i in range(countdown, 0, -1):
+                st.write(i)
+                time.sleep(1)
+            st.write("Hết giờ!")
+
+    # Đối với các chế độ ôn tập khác, hiển thị thông tin tương ứng
+    else:
+        st.write(f"Chế độ {study_mode} được chọn")
+
+# Tiêu đề chân trang
+st.markdown("<h3 style='text-align: center;'>Chúc bé iu của anh thi đậu nhé😘</h3>", unsafe_allow_html=True)