added missing buttons, shortened function definition code

app.py

@@ -49,6 +49,7 @@ with gr.Blocks() as iface:
     with gr.Row() as chatRow:
         with gr.Column(scale=2):
             questionCounter = gr.Label(label="Remaining Questions", value="24")
+            mapButton = gr.Button("Show Floor Plan")
             finishButton = gr.Button("Try to Solve")
         with gr.Column(scale=10):
             i = 0
@@ -62,6 +63,10 @@ with gr.Blocks() as iface:
                             characterImage = gr.Label(label="")
                 i += 1
 
+    with gr.Row(visible=False) as mapRow:
+        with gr.Column():
+            closeMapButton = gr.Button("X")
+
     with gr.Row(visible=False) as solvingRow:
         i = 0
         while i < len(roles):
@@ -69,6 +74,8 @@ with gr.Blocks() as iface:
                 accusationButtons.append(gr.Button(roles[i]))
             i += 1
 
+        closeSolvingButton = gr.Button("X")
+
     with gr.Row(visible=False) as resultRow:
         with gr.Column():
             resultLabel = gr.Label(label="Result")
@@ -135,6 +142,12 @@ with gr.Blocks() as iface:
 
     def tryToSolve():
         return {solvingRow: gr.update(visible=True)}
+    def showMap():
+        return {mapRow: gr.update(visible=True)}
+    def hideSolveBox():
+        return {solvingRow: gr.update(visible=False)}
+    def hideMap():
+        return {mapRow: gr.update(visible=False)}
 
     def showResult():
         return {chatRow: gr.update(visible=False), solvingRow: gr.update(visible=False), resultRow: gr.update(visible=True)}
@@ -146,37 +159,20 @@ with gr.Blocks() as iface:
         return "Congratulations, you found the killer! "
 
     # again, ugly workaround
-    msg[0].submit(user, [msg[0], chatbot[0]], [msg[0], chatbot[0]], queue=False).then(
-        bot0, [chatbot[0], questionCounter], chatbot[0]
-    ).then(
-        updateQuestions, questionCounter, questionCounter
-    )
-
-    msg[1].submit(user, [msg[1], chatbot[1]], [msg[1], chatbot[1]], queue=False).then(
-        bot1, [chatbot[1], questionCounter], chatbot[1]
-    ).then(
-        updateQuestions, questionCounter, questionCounter
-    )
-
-    msg[2].submit(user, [msg[2], chatbot[2]], [msg[2], chatbot[2]], queue=False).then(
-        bot2, [chatbot[2], questionCounter], chatbot[2]
-    ).then(
-        updateQuestions, questionCounter, questionCounter
-    )
-
-    msg[3].submit(user, [msg[3], chatbot[3]], [msg[3], chatbot[3]], queue=False).then(
-        bot3, [chatbot[3], questionCounter], chatbot[3]
-    ).then(
-        updateQuestions, questionCounter, questionCounter
-    )
-
-    msg[4].submit(user, [msg[4], chatbot[4]], [msg[4], chatbot[4]], queue=False).then(
-        bot4, [chatbot[4], questionCounter], chatbot[4]
-    ).then(
-        updateQuestions, questionCounter, questionCounter
-    )
+    botfunctions = [bot0, bot1, bot2, bot3, bot4]
+    i = 0
+    while i < len(roles):
+        msg[i].submit(user, [msg[i], chatbot[i]], [msg[i], chatbot[i]], queue=False).then(
+            botfunctions[i], [chatbot[i], questionCounter], chatbot[i]
+        ).then(
+            updateQuestions, questionCounter, questionCounter
+        )
+        i += 1
 
+    mapButton.click(showMap, None, mapRow)
+    closeMapButton.click(hideMap, None, mapRow)
     finishButton.click(tryToSolve, None, solvingRow)
+    closeSolvingButton.click(hideSolveBox, None, solvingRow)
 
     i = 0
     while i < len(roles):