Update app.py

app.py

@@ -10,7 +10,11 @@ DIALOGUES = dataset["train"]["dialogues"]
 
 def generate_statement():
     """Return a random dialogue from the dataset."""
-    return random.choice(DIALOGUES)
+    # Pick a random sublist from the dataset
+    random_dialogue_list = random.choice(DIALOGUES)
+    # Pick a random dialogue from the sublist
+    return random.choice(random_dialogue_list)
+
 
 def main():
     # Define the UI using gr.Interface