Update pages/4_Mini_Game.py

pages/4_Mini_Game.py

@@ -3,14 +3,15 @@
 import streamlit as st
 import pandas as pd
 from transformers import pipeline
-#for the pre-processing of uploaded data
-import string 
 from utils.styles import load_css
 load_css()
 from datasets import load_dataset
 
+#loading the training dataset
 dataset = load_dataset("SuchithTissera/News_dataset")
-st.write(dataset["train"])
+news_dataset = dataset["train"]
+
+st.set_page_config(page_title="Mini Game", layout="wide")
 
 #writing the content for the page
 st.markdown('<div class="main_title">🧩 Classification Mini-Game </div>',unsafe_allow_html=True)
@@ -21,4 +22,162 @@ st.markdown("""
 <ul>
     <li> Users may play the game with 5 or 10 rounds to test their skills </li>
 </ul>
-</div>""",unsafe_allow_html=True)
+</div>""",unsafe_allow_html=True)
+
+#coding the mini-game
+CATEGORIES = [
+    "Business",
+    "Opinions",
+    "Political_gossip",
+    "Sports",
+    "Worlds_news"
+]
+
+# ---------------------------
+# Session state setup
+# ---------------------------
+if "game_started" not in st.session_state:
+    st.session_state.game_started = False
+
+if "questions" not in st.session_state:
+    st.session_state.questions = None
+
+if "current_round" not in st.session_state:
+    st.session_state.current_round = 0
+
+if "score" not in st.session_state:
+    st.session_state.score = 0
+
+if "total_rounds" not in st.session_state:
+    st.session_state.total_rounds = 5
+
+if "answered" not in st.session_state:
+    st.session_state.answered = False
+
+if "selected_option" not in st.session_state:
+    st.session_state.selected_option = None
+
+if "game_finished" not in st.session_state:
+    st.session_state.game_finished = False
+
+
+# ---------------------------
+# Functions
+# ---------------------------
+def start_game(round_count):
+    sampled = news_dataset.sample(n=round_count).reset_index(drop=True)
+
+    st.session_state.questions = sampled
+    st.session_state.total_rounds = round_count
+    st.session_state.current_round = 0
+    st.session_state.score = 0
+    st.session_state.answered = False
+    st.session_state.selected_option = None
+    st.session_state.game_started = True
+    st.session_state.game_finished = False
+
+
+def next_question():
+    st.session_state.current_round += 1
+    st.session_state.answered = False
+    st.session_state.selected_option = None
+
+    if st.session_state.current_round >= st.session_state.total_rounds:
+        st.session_state.game_finished = True
+
+
+def restart_game():
+    st.session_state.game_started = False
+    st.session_state.questions = None
+    st.session_state.current_round = 0
+    st.session_state.score = 0
+    st.session_state.total_rounds = 5
+    st.session_state.answered = False
+    st.session_state.selected_option = None
+    st.session_state.game_finished = False
+
+
+# ---------------------------
+# UI
+# ---------------------------
+st.title("🎮 Classification Mini Game")
+
+if not st.session_state.game_started:
+    st.subheader("Choose game length")
+    round_choice = st.radio(
+        "Select number of rounds:",
+        [5, 10],
+        horizontal=True
+    )
+
+    if st.button("Start Game"):
+        start_game(round_choice)
+        st.rerun()
+
+# ---------------------------
+# Game in progress
+# ---------------------------
+elif st.session_state.game_started and not st.session_state.game_finished:
+    q_index = st.session_state.current_round
+    row = st.session_state.questions.iloc[q_index]
+
+    st.markdown(f"### Round {q_index + 1} of {st.session_state.total_rounds}")
+    st.write("**Content:**")
+    st.info(row["content"])
+
+    selected = st.radio(
+        "Choose the correct category:",
+        CATEGORIES,
+        key=f"question_{q_index}"
+    )
+
+    if not st.session_state.answered:
+        if st.button("Submit Answer"):
+            st.session_state.selected_option = selected
+            st.session_state.answered = True
+
+            if selected == row["class"]:
+                st.session_state.score += 1
+
+            st.rerun()
+
+    else:
+        correct_answer = row["class"]
+
+        if st.session_state.selected_option == correct_answer:
+            st.success(f"Correct! The answer is {correct_answer}.")
+        else:
+            st.error(
+                f"Wrong. You selected {st.session_state.selected_option}, "
+                f"but the correct answer is {correct_answer}."
+            )
+
+        if st.button("Next Round"):
+            next_question()
+            st.rerun()
+
+# ---------------------------
+# Game finished
+# ---------------------------
+if st.session_state.game_finished:
+    score = st.session_state.score
+    total = st.session_state.total_rounds
+    percentage = (score / total) * 100
+
+    st.markdown("## Game Over")
+    st.write(f"Your score: **{score}/{total}**")
+
+    if percentage >= 75:
+        st.markdown(
+            f"<h3 style='color: green;'>Percentage Score: {percentage:.2f}%</h3>",
+            unsafe_allow_html=True
+        )
+    else:
+        st.markdown(
+            f"<h3 style='color: red;'>Percentage Score: {percentage:.2f}%</h3>",
+            unsafe_allow_html=True
+        )
+
+    if st.button("Play Again"):
+        restart_game()
+        st.rerun()