Update app.py

app.py

@@ -13,7 +13,7 @@ if 'index' not in st.session_state:
 if 'game_over' not in st.session_state:
     st.session_state.game_over = False
 
-# Function to get next text from the dataset
+# Function to get the current text pair from the dataset
 def get_next_text():
     # If game is over, return None to stop
     if st.session_state.game_over:
@@ -27,10 +27,10 @@ def get_next_text():
     
     # Combine the text and make sure it doesn't exceed 10 sentences
     full_text = prompt_text + " _SEP_ " + gen_text
-    return full_text, text_data['model'], text_data['true_boundary_index']
+    return prompt_text, gen_text, text_data['model'], text_data['true_boundary_index']
 
 # Function to update the game state
-def update_game_state(user_answer, correct_answer):
+def update_game_state(user_answer, correct_answer, user_reason=None):
     # Check if answer is correct
     if user_answer == correct_answer:
         st.session_state.score += 5
@@ -41,26 +41,44 @@ def update_game_state(user_answer, correct_answer):
     else:
         st.session_state.game_over = True
         st.error("Game Over! Incorrect Answer.")
+    
+    # If user chose machine, display their reason
+    if user_answer == 'machine' and user_reason:
+        st.write(f"Reason selected: {user_reason}")
 
 # Display the score
 st.sidebar.text(f"Score: {st.session_state.score}")
 
 # Show the current text to classify
-text, model_used, true_boundary_index = get_next_text()
+prompt_text, gen_text, model_used, true_boundary_index = get_next_text()
 
-if text:
-    # Display the 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: {text}")
+    st.write(f"Text: {prompt_text}")
     
+    # Display the continuation text
+    st.header("Continuation Text")
+    st.write(f"{gen_text}")
+
     # User input (radio buttons for classification)
-    user_answer = st.radio("Classify the text as:", ["human", "machine"])
+    user_answer = st.radio("Classify the continuation as:", ["human", "machine"])
+
+    # 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
 
     # 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)
+        update_game_state(user_answer, correct_answer, user_reason)
 else:
     # Game over message
     st.write("Game Over!")