Update app.py

app.py

@@ -6,7 +6,7 @@ from text_gen import generate_text
 
 def get_answer(question, answer, options):
     '''File given answer'''
-    
+
     global questions
     global SEPARATOR
     answer = options.split(SEPARATOR)[answer]
@@ -15,7 +15,7 @@ def get_answer(question, answer, options):
 
 def find_score(column_name, questions):
     '''Count chosen answers in a given column'''
-    
+    #global questions
     actual = questions[column_name].to_list()
     chosen = questions.Choice.to_list()
     return sum(a == c for a, c in zip(actual, chosen))
@@ -23,7 +23,7 @@ def find_score(column_name, questions):
 
 def set_score():
     '''Return the score (for the human and for the models)'''
-    
+
     global NUM_QUESTIONS
     score = find_score(answers[0], questions)
     ai_score = {name:find_score(name, questions) for name in answers[1:]}
@@ -31,16 +31,33 @@ def set_score():
     start = '## <p style="text-align: center;">'
     end = '</p>'
 
-    if       score == NUM_QUESTIONS:         text = f'Perfect score!'
+    if score == NUM_QUESTIONS:
+        text = f'Perfect score!'
     else:
-        if   score == 0:                     text = 'Not a single right answer. Are you doing this on purpose?'
-        elif score <= NUM_QUESTIONS / 2 + 1: text = f'Only {score} right answers out of {NUM_QUESTIONS}. You ought to pay more attention.'
-        else:                                text = f'{score} right answers out of {NUM_QUESTIONS}. It\'s probably alright.'
+        if score == 0:
+            text = 'Not a single right answer. Are you doing this on purpose?'
+        elif score <= NUM_QUESTIONS / 2 + 1:
+            text = f'Only {score} right answers out of {NUM_QUESTIONS}. You ought to pay more attention.'
+        else:
+            text = f'{score} right answers out of {NUM_QUESTIONS}. It\'s probably alright.'
+
         for name in ai_score.keys():
             text += f'\n{name} got {ai_score[name]}.'
 
+        # Add a section to display correct/incorrect answers
+        text += "\n\nList of questions with correct answers:\n"
+        i = 0
+        for idx, row in questions.iterrows():
+            i += 1
+            question = row['Question']
+            answer = row['Answer']
+            chosen_answer = row['Choice']
+            status = "✅" if answer == chosen_answer else "❌"
+            text += f"\n\n{status} {i}. {question} (Correct: {answer})\n"
+
     return start + text + end
 
+
 if __name__ == "__main__":
 
     NUM_QUESTIONS = 10
@@ -48,8 +65,8 @@ if __name__ == "__main__":
 
 
     # Adding/replacing models may require adjustments to text_gen.py
-    model_names = ['microsoft/GODEL-v1_1-base-seq2seq',
-                   'facebook/blenderbot-400M-distill',
+    model_names = ['microsoft/GODEL-v1_1-large-seq2seq',
+                   'facebook/blenderbot-1B-distill',
                    'facebook/blenderbot_small-90M']
 
     tokenizers = [transformers.AutoTokenizer.from_pretrained(model_names[0]),