add solving box

app.py

@@ -54,43 +54,50 @@ with gr.Blocks() as iface:
                 with gr.Tab(roles[i]):
                     with gr.Row():
                         with gr.Column(scale=9):
-                            chatbot.append(gr.Chatbot())
-                            msg.append(gr.Textbox())
+                            chatbot.append(gr.Chatbot(label=roles[i]))
+                            msg.append(gr.Textbox(label=""))
                         with gr.Column(scale=3):
-                            characterImage = gr.Label()
+                            characterImage = gr.Label(label="")
                 i += 1
 
+    with gr.Row(visible="false") as solvingBox:
+        i = 0
+        while i < len(roles):
+            with gr.Column():
+                finishButton = gr.Button(roles[i])
+            i += 1
+
     def user(user_message, history):
         return "", history + [[user_message, None]]
 
     # this is an ugly workaround, you cannot pass regular integers to gradio functions, so i had to do it this way
-    def bot0(history, prevText):
-        bot_message = generateAIMessage(history, prevText, 0)
+    def bot0(history, questionCounterText):
+        bot_message = generateAIMessage(history, questionCounterText, 0)
         history[-1][1] = bot_message
         return history
 
-    def bot1(history, prevText):
-        bot_message = generateAIMessage(history, prevText, 1)
+    def bot1(history, questionCounterText):
+        bot_message = generateAIMessage(history, questionCounterText, 1)
         history[-1][1] = bot_message
         return history
 
-    def bot2(history, prevText):
-        bot_message = generateAIMessage(history, prevText, 2)
+    def bot2(history, questionCounterText):
+        bot_message = generateAIMessage(history, questionCounterText, 2)
         history[-1][1] = bot_message
         return history
 
-    def bot3(history, prevText):
-        bot_message = generateAIMessage(history, prevText, 3)
+    def bot3(history, questionCounterText):
+        bot_message = generateAIMessage(history, questionCounterText, 3)
         history[-1][1] = bot_message
         return history
 
-    def bot4(history, prevText):
-        bot_message = generateAIMessage(history, prevText, 4)
+    def bot4(history, questionCounterText):
+        bot_message = generateAIMessage(history, questionCounterText, 4)
         history[-1][1] = bot_message
         return history
 
-    def generateAIMessage(history, prevText, characterId):
-        if checkIfQuestionsLeft(prevText):
+    def generateAIMessage(history, questionCounterText, characterId):
+        if checkIfQuestionsLeft(questionCounterText):
             message_history = [
                 {"role": "system", "content": initText[characterId] + "Stay in character. Use natural language. Don't "
                                                                       "reveal all of the information in a single "
@@ -109,16 +116,19 @@ with gr.Blocks() as iface:
         else:
             return "--- YOU'VE RUN OUT OF QUESTIONS, MAKE YOUR CHOICE ---"
 
-    def updateQuestions(prevText):
-        remainingQuestions = int(prevText["label"].split(": ")[1])
+    def updateQuestions(questionCounterText):
+        remainingQuestions = int(questionCounterText["label"].split(": ")[1])
         if remainingQuestions > 0:
             remainingQuestions -= 1
         return "Remaining Questions: " + str(remainingQuestions)
     
-    def checkIfQuestionsLeft(prevText):
-        remainingQuestions = int(prevText["label"].split(": ")[1])
+    def checkIfQuestionsLeft(questionCounterText):
+        remainingQuestions = int(questionCounterText["label"].split(": ")[1])
         return remainingQuestions > 0
 
+    def tryToSolve():
+        return {solvingBox: gr.update(visible=True)}
+
     # again, ugly workaround
     msg[0].submit(user, [msg[0], chatbot[0]], [msg[0], chatbot[0]], queue=False).then(
         bot0, [chatbot[0], questionCounter], chatbot[0]
@@ -150,5 +160,7 @@ with gr.Blocks() as iface:
         updateQuestions, questionCounter, questionCounter
     )
 
+    finishButton.click(tryToSolve, None, solvingBox)
+
 
 iface.launch()