added user sessions

app.py

@@ -3,6 +3,7 @@ from dotenv import load_dotenv
 import os
 import anthropic
 import openai
+from uuid import uuid4
 from util import ClaudeCompletion, GPTCompletion
 
 gr.close_all()
@@ -40,20 +41,18 @@ def chatWithGPT(chatHistory):
     return chatHistory
 
 with gr.Blocks() as demo:
-    chatHistory = []
-    gptsTurn = True
+    chatState = gr.State([])
+    gptsTurnState = gr.State(True)
 
-    def startConversation(prompt):
-        global chatHistory, gptsTurn
+    def startConversation(prompt, gptsTurn):
         # nextBtn.interactive = False
         chatHistory = []
         startNewChat(prompt, chatHistory)
         # nextBtn.interactive = True
         gptsTurn = False
-        return chatHistory
+        return chatHistory, chatHistory, gptsTurn
 
-    def nextResponse():
-        global gptsTurn
+    def nextResponse(chatHistory, gptsTurn):
         # nextBtn.interactive = False
         if (gptsTurn):
             chatWithGPT(chatHistory)
@@ -61,7 +60,7 @@ with gr.Blocks() as demo:
             chatWithClaude(chatHistory)
         gptsTurn = not gptsTurn
         # nextBtn.interactive = True
-        return chatHistory
+        return chatHistory, chatHistory, gptsTurn
 
     context = gr.Textbox(label="Context",
                          placeholder="Set the context for two LLMs to chat with each other",
@@ -74,11 +73,11 @@ with gr.Blocks() as demo:
         ["A person is interested in Formula 1 and asks questions about the sport",
          "A little child wants to understand how planes fly",
          "You are a young indie game dev searching for ideas about their next strategy game",
-         "You are an old irish man is interested in medieval european history"]
+         "You are an old irish man interested in medieval european history"]
         , context)
     # nextBtn.interactive = False
-    start_button.click(startConversation, context, chatbot, scroll_to_output=True, show_progress=True)
-    nextBtn.click(nextResponse, None, chatbot, scroll_to_output=True, show_progress=True)
+    start_button.click(startConversation, [context, gptsTurnState], [chatbot, chatState, gptsTurnState], scroll_to_output=True, show_progress=True)
+    nextBtn.click(nextResponse, [chatState, gptsTurnState], [chatbot, chatState, gptsTurnState], scroll_to_output=True, show_progress=True)
 
 try:
     demo.launch()