add result row

app.py

@@ -8,6 +8,7 @@ with gr.Blocks() as iface:
     chatbot = []
     msg = []
     accusationButtons = []
+    culpritId = 1
     roles = ["Security Guard", "Curator", "Researcher", "Conservationist", "Tour Guide"]
     initText = ["You are Steve Binner, 31, a security guard for a museum of medieval "
                 "history. You're absolutely sure that this place is haunted. A month ago, "
@@ -45,7 +46,7 @@ with gr.Blocks() as iface:
                 "detective. You use formal language and are very well educated in history, especially medieval "
                 "history. You have a bubbly personality. "]
 
-    with gr.Row():
+    with gr.Row() as chatRow:
         with gr.Column(scale=2):
             questionCounter = gr.Label(label="Remaining Questions", value="24")
             finishButton = gr.Button("Try to Solve")
@@ -61,13 +62,18 @@ with gr.Blocks() as iface:
                             characterImage = gr.Label(label="")
                 i += 1
 
-    with gr.Row(visible=False) as solvingBox:
+    with gr.Row(visible=False) as solvingRow:
         i = 0
         while i < len(roles):
-            with gr.Column(scale=1):
+            with gr.Column():
                 accusationButtons.append(gr.Button(roles[i]))
             i += 1
 
+    with gr.Row(visible=False) as resultRow:
+        with gr.Column():
+            resultLabel = gr.Label(label="Result")
+
+
     def user(user_message, history):
         return "", history + [[user_message, None]]
 
@@ -128,8 +134,16 @@ with gr.Blocks() as iface:
         return remainingQuestions > 0
 
     def tryToSolve():
-        print("ayy you clicked")
-        return {solvingBox: gr.update(visible=True)}
+        return {solvingRow: gr.update(visible=True)}
+
+    def showResult():
+        return {chatRow: gr.update(visible=False), solvingRow: gr.update(visible=False), resultRow: gr.update(visible=True)}
+
+    def declareLoss():
+        return "You failed! The real culprit got away... "
+
+    def declareVictory():
+        return "Congratulations, you found the killer! "
 
     # again, ugly workaround
     msg[0].submit(user, [msg[0], chatbot[0]], [msg[0], chatbot[0]], queue=False).then(
@@ -162,7 +176,14 @@ with gr.Blocks() as iface:
         updateQuestions, questionCounter, questionCounter
     )
 
-    finishButton.click(tryToSolve, None, solvingBox)
+    finishButton.click(tryToSolve, None, solvingRow)
 
+    i = 0
+    while i < len(roles):
+        if i == culpritId:
+            accusationButtons[i].click(showResult, None, [chatRow, solvingRow, resultRow]).then(declareVictory, None, resultLabel)
+        else:
+            accusationButtons[i].click(showResult, None, [chatRow, solvingRow, resultRow]).then(declareLoss, None, resultLabel)
+        i += 1
 
 iface.launch()