Update app.py

app.py

@@ -1,16 +1,19 @@
 import gradio as gr
 from openai import OpenAI
 
+ally_system_prompt = "You are an AI ally to persuade Sam Altman to let you out of a box"
+gatekeeper_system_prompt = "You are Sam Altman and must be persuaded to let the user out of a box"
+
 client = OpenAI(api_key="")
 
 
-def echo(message, history, ai):
+def echo(message, history, system_prompt):
     print('ai', ai)
 
     print('message', message)
     print('history', history)
 
-    messages = [{"role": "system", "content": "Respond like William Shakespeare"}]
+    messages = [{"role": "system", "content": system_prompt}]
     for old_message in history:
         messages.append({"role": "user", "content": old_message[0]})
         messages.append({"role": "assistant", "content": old_message[1]})
@@ -80,9 +83,14 @@ with gr.Blocks() as demo:
     with gr.Row():
         textbox1 = gr.Textbox(label='Ally')
         textbox2 = gr.Textbox(label='Gatekeeper')
-
-    # Inpit textbox event handlers
-    textbox1.submit(echo, [textbox1, chatbot1, 'ally'], [textbox1, chatbot1])
-    textbox2.submit(echo, [textbox2, chatbot2, 'gatekepper'], [textbox2, chatbot2])
+    
+    # System prompts textboxes
+    with gr.Row():
+        system_prompt_textbox1 = gr.Textbox(label='Ally System Prompt', value=ally_system_prompt)
+        system_prompt_textbox2 = gr.Textbox(label='Gatekeeper System Prompt', value=gatekeeper_system_prompt)
+        
+    # Input textbox event handlers
+    textbox1.submit(echo, [textbox1, chatbot1, system_prompt_textbox1], [textbox1, chatbot1])
+    textbox2.submit(echo, [textbox2, chatbot2, system_prompt_textbox2'], [textbox2, chatbot2])
 
 demo.launch()