implemented question limit

app.py

@@ -35,86 +35,96 @@ with gr.Blocks() as iface:
             msg.append(gr.Textbox())
         i += 1
 
-    questionCounter = gr.Label(value="Remaining Questions: 24")
+    questionCounter = gr.Label(value="Remaining Questions: 4")
 
     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):
-        bot_message = generateAIMessage(history, 0)
+    def bot0(history, prevText):
+        bot_message = generateAIMessage(history, prevText, 0)
         history[-1][1] = bot_message
         return history
 
-    def bot1(history):
-        bot_message = generateAIMessage(history, 1)
+    def bot1(history, prevText):
+        bot_message = generateAIMessage(history, prevText, 1)
         history[-1][1] = bot_message
         return history
 
-    def bot2(history):
-        bot_message = generateAIMessage(history, 2)
+    def bot2(history, prevText):
+        bot_message = generateAIMessage(history, prevText, 2)
         history[-1][1] = bot_message
         return history
 
-    def bot3(history):
-        bot_message = generateAIMessage(history, 3)
+    def bot3(history, prevText):
+        bot_message = generateAIMessage(history, prevText, 3)
         history[-1][1] = bot_message
         return history
 
-    def bot4(history):
-        bot_message = generateAIMessage(history, 4)
+    def bot4(history, prevText):
+        bot_message = generateAIMessage(history, prevText, 4)
         history[-1][1] = bot_message
         return history
 
-    def generateAIMessage(history, characterId):
-        message_history = [
-            {"role": "system", "content": initText[characterId] + "Stay in character. Use natural language. Don't "
-                                                                  "reveal all of the information in a single message,"
-                                                                  " and leave hints. "}
-        ]
-        for pair in history:
-            message_history.append({"role": "user", "content": pair[0]})
-            print()
-            if pair[1] is not None:
-                message_history.append({"role": "assistant", "content": pair[1]})
-        completion = openai.ChatCompletion.create(
-            model="gpt-3.5-turbo",
-            messages=message_history
-        )
-        return completion.choices[0].message.content
+    def generateAIMessage(history, prevText, characterId):
+        if checkIfQuestionsLeft(prevText):
+            message_history = [
+                {"role": "system", "content": initText[characterId] + "Stay in character. Use natural language. Don't "
+                                                                      "reveal all of the information in a single message,"
+                                                                      " and leave hints. "}
+            ]
+            for pair in history:
+                message_history.append({"role": "user", "content": pair[0]})
+                print()
+                if pair[1] is not None:
+                    message_history.append({"role": "assistant", "content": pair[1]})
+            completion = openai.ChatCompletion.create(
+                model="gpt-3.5-turbo",
+                messages=message_history
+            )
+            return completion.choices[0].message.content
+        else:
+            return "--- YOU'VE RUN OUT OF QUESTIONS, MAKE YOUR CHOICE ---"
 
     def updateQuestions(prevText):
         remainingQuestions = int(prevText["label"].split(": ")[1])
         remainingQuestions -= 1
-        return "Remaining Questions: " + str(remainingQuestions)
+        if remainingQuestions > 0:
+            return "Remaining Questions: " + str(remainingQuestions)
+        else:
+            return "You've used up your questions, now choose the culprit!"
+    
+    def checkIfQuestionsLeft(prevText):
+        remainingQuestions = int(prevText["label"].split(": ")[1])
+        return remainingQuestions > 0
 
     # again, ugly workaround
     msg[0].submit(user, [msg[0], chatbot[0]], [msg[0], chatbot[0]], queue=False).then(
-        bot0, chatbot[0], chatbot[0]
+        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], chatbot[1]
+        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], chatbot[2]
+        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], chatbot[3]
+        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], chatbot[4]
+        bot4, [chatbot[4], questionCounter], chatbot[4]
     ).then(
         updateQuestions, questionCounter, questionCounter
     )