Update app.py

app.py

@@ -213,8 +213,8 @@ with gr.Blocks(title="Beat the AI", theme=gr.themes.Base(), css=css) as demo:
                 g = gr.Button(value="G", min_width=1)
                 h = gr.Button(value="H", min_width=1)
             with gr.Row():
-                model_prediction.render()
                 user_prediction.render()
+                model_prediction.render()
             solution  = gr.TextArea(
                 label="Solution",
                 visible=False,
@@ -225,7 +225,21 @@ with gr.Blocks(title="Beat the AI", theme=gr.themes.Base(), css=css) as demo:
             with gr.Row():
                 result.render()
 
-
+    def result_string(model_pred, user_pred, solution):
+        solution_letter = chr(ord('A') + int(z))
+        solution_string = f"The correct answer is f{solution_letter}."
+        
+        win_or_loose = "🥇" if user_pred == solution_letter else "🙈"
+        
+        if user_pred == solution_letter and model_pred == solution_letter:
+            comparison_string = "Both you and the AI got it correctly!"
+        elif user_pred == solution_letter and model_pred != solution_letter:
+            comparison_string = "You beat the AI!"
+        elif user_pred != solution_letter and model_pred != solution_letter:
+            comparison_string = "Both you and the AI got it wrong!"
+        
+        return f"{win_or_loose} {comparison_string}\n{solution_string}"
+    
     load_new_sample.click(
         fn=load_sample,
         inputs=[],
@@ -246,7 +260,7 @@ with gr.Blocks(title="Beat the AI", theme=gr.themes.Base(), css=css) as demo:
         inputs=[imagebox],
         outputs=[model_prediction],
     ).then(
-        fn=lambda x, y, z: "🥇" if z==y else f"💩 The solution is {chr(ord('A') + int(z))}",
+        fn=result_string,
         inputs=[model_prediction, user_prediction, solution],
         outputs=[result],
     )