Update app.py

app.py

@@ -2,8 +2,8 @@ import streamlit as st
 import pandas as pd
 import random
 
-# Load the dataset (replace with your actual dataset path)
-df = pd.read_csv('roft.csv')  # Ensure your dataset path is correct
+# Load the dataset (ensure the file path is correct)
+df = pd.read_csv('roft.csv')  # Replace with the actual dataset path
 
 # Initialize session state variables
 if 'score' not in st.session_state:
@@ -21,17 +21,22 @@ def get_next_text():
     # Fetch the next text in sequence
     text_data = df.iloc[st.session_state.index]
     
-    # Combine the prompt body and generation body to form the text to classify
+    # Extract prompt and generation text
     prompt_text = text_data['prompt_body']
     gen_text = text_data['gen_body']
+    model_used = text_data['model']
     
-    # Combine the text and make sure it doesn't exceed 10 sentences
-    full_text = prompt_text + " _SEP_ " + gen_text
-    return prompt_text, gen_text, text_data['model'], text_data['true_boundary_index']
+    # Get the continuation text (sentence before "_SEP_")
+    continuation_text = gen_text.split("_SEP_")[0].strip()
+
+    # Get the true boundary index (where human text ends and machine text begins)
+    true_boundary_index = text_data['true_boundary_index']
+    
+    return prompt_text, continuation_text, gen_text, model_used, true_boundary_index
 
 # Function to update the game state
 def update_game_state(user_answer, correct_answer, user_reason=None):
-    # Check if answer is correct
+    # Check if the answer is correct
     if user_answer == correct_answer:
         st.session_state.score += 5
         st.session_state.index += 1
@@ -50,44 +55,40 @@ def update_game_state(user_answer, correct_answer, user_reason=None):
 st.sidebar.text(f"Score: {st.session_state.score}")
 
 # Show the current text to classify
-prompt_text, gen_text, model_used, true_boundary_index = get_next_text()
+if not st.session_state.game_over:
+    prompt_text, continuation_text, gen_text, model_used, true_boundary_index = get_next_text()
 
-if prompt_text:
-    # Display the human-generated text
-    st.header("Human-Generated Text")
-    st.write(f"Model used: {model_used}")
-    st.write(f"Text: {prompt_text}")
-    
-    # Display the continuation text
-    st.header("Continuation Text")
-    st.write(f"{gen_text}")
+    if prompt_text:
+        # Display the human-generated text (more readable format)
+        st.header("Human-Generated Text")
+        st.write(f"Model used: {model_used}")
+        st.write("Text: ")
+        st.markdown(f"**{prompt_text}**")
 
-    # User input (radio buttons for classification)
-    user_answer = st.radio("Classify the continuation as:", ["human", "machine"])
+        # Display the continuation text (sentence before _SEP_)
+        st.header("Continuation Text")
+        st.write(f"{continuation_text}")
 
-    # If user selects 'machine', ask for reasoning
-    if user_answer == 'machine':
-        user_reason = st.selectbox("Why do you think it's machine-generated?", [
-            "grammar", "repetition", "irrelevant", "contradicts_sentence", "contradicts_knowledge", 
-            "common_sense", "coreference", "generic", "other"
-        ])
-    else:
-        user_reason = None
+        # User input (radio buttons for classification)
+        user_answer = st.radio("Classify the continuation as:", ["human", "machine"])
 
-    # When user submits answer
-    if st.button("Submit"):
-        # Correct answer is determined based on boundary index
-        correct_answer = "human" if true_boundary_index == 0 else "machine"
-        update_game_state(user_answer, correct_answer, user_reason)
-else:
-    # Game over message
-    st.write("Game Over!")
-    st.write(f"Your final score is: {st.session_state.score}")
+        # If user selects 'machine', ask for reasoning
+        if user_answer == 'machine':
+            user_reason = st.selectbox("Why do you think it's machine-generated?", [
+                "grammar", "repetition", "irrelevant", "contradicts_sentence", "contradicts_knowledge", 
+                "common_sense", "coreference", "generic"
+            ])
+
+            # Button to submit answer
+            if st.button('Submit'):
+                correct_answer = 'machine' if true_boundary_index != 0 else 'human'
+                update_game_state(user_answer, correct_answer, user_reason)
 
-# Option to restart the game
-if st.session_state.game_over:
-    if st.button("Restart Game"):
-        st.session_state.score = 0
-        st.session_state.index = 0
-        st.session_state.game_over = False
-        st.experimental_rerun()
+        # Button to submit answer if 'human' is selected
+        elif st.button('Submit'):
+            correct_answer = 'human' if true_boundary_index == 0 else 'machine'
+            update_game_state(user_answer, correct_answer)
+
+else:
+    # If the game is over, display the final score
+    st.write(f"Final Score: {st.session_state.score}")