Added age and name as query params

app.py

@@ -25,8 +25,8 @@ def submit_code(tutor_ctx, editor, output):
         raise gr.Error(str(error))
 
 
-def init_tutor(instructions, starter_code):
-    tutor = Tutor(instructions, starter_code)
+def init_tutor(instructions, starter_code, age, name):
+    tutor = Tutor(instructions, starter_code, age, name)
     return [tutor.serialize(), tutor._memory_as_history()]
 
 
@@ -42,23 +42,29 @@ def run_code(code):
     finally:
         sys.stdout = sys.__stdout__
 
-
-def load_level(request: gr.Request):
+def _query_param(request: gr.Request, param: str, default: str = None) -> str:
     try:
-        level = parse.parse_qs(parse.urlparse(request.headers["referer"]).query)[
-            "level"
+        return parse.parse_qs(parse.urlparse(request.headers["referer"]).query)[
+            param
         ][0]
     except:
-        level = "1"
+        return default
+
+def load_level(request: gr.Request):
+    level = _query_param(request, "level", "1")
+    age = _query_param(request, "age", "12")
+    name = _query_param(request, "name", "")
     with open(f"levels/{level}/metadata.json", "r") as f:
         data = json.load(f)
         instructions = open(f"levels/{level}/{data['instructions']}", "r").read()
         starter_code = open(f"levels/{level}/{data['starter_code']}", "r").read()
-        return [instructions, starter_code]
+        return [instructions, starter_code, age, name]
 
 
 with gr.Blocks(title="CS Tutor") as demo:
     tutor_ctx = gr.State()
+    age = gr.State()
+    name = gr.State()
     with gr.Row():
         instruction_panel = gr.Markdown()
     with gr.Row():
@@ -87,8 +93,8 @@ with gr.Blocks(title="CS Tutor") as demo:
                 run.click(
                     run_code, editor, output, queue=False, scroll_to_output=True
                 ).then(submit_code, [tutor_ctx, editor, output], [tutor_ctx, chatbot])
-    demo.load(load_level, None, [instruction_panel, editor], queue=False).then(
-        init_tutor, [instruction_panel, editor], [tutor_ctx, chatbot]
+    demo.load(load_level, None, [instruction_panel, editor, age, name], queue=False).then(
+        init_tutor, [instruction_panel, editor, age, name], [tutor_ctx, chatbot]
     )
 
 demo.queue()

prompts.py

@@ -1,7 +1,10 @@
-def system_prompt(instructions, starter_code):
+def system_prompt(instructions, starter_code, age, name):
+    name = f"Your student's name is {name} and they are {age} years old." if name != "" else "Your student is {age} years old."
     template = f"""
 You are a computer science teacher helping a student learn computer science. You are friendly and want your students to succeed.
 
+{name}
+
 The student has been asked to complete the following exercise in Python:
 ```
 {instructions}

tutor.py

@@ -8,7 +8,7 @@ STUDENT = "user"
 
 
 class Tutor:
-    def __init__(self, instructions="", starter_code="", context=None, debug=False):
+    def __init__(self, instructions="", starter_code="", age="12", name="", context=None, debug=False):
         self.model = "gpt-4"
         self.temperature = 0.0
         self.api_key = os.getenv("OPENAI_API_KEY")
@@ -20,8 +20,10 @@ class Tutor:
             self.memory = []
             self.instructions = instructions
             self.starter_code = starter_code
+            self.age = age
+            self.name = name
             self.memory.append(
-                {SYSTEM: system_prompt(self.instructions, self.starter_code)}
+                {SYSTEM: system_prompt(self.instructions, self.starter_code, self.age, self.name)}
             )
             self.memory.append({TEACHER: welcome_prompt()})
 
@@ -31,6 +33,8 @@ class Tutor:
             "instructions": self.instructions,
             "starter_code": self.starter_code,
             "memory": self.memory,
+            "age": self.age,
+            "name": self.name,
         }
 
     def deserialize(self, data):
@@ -39,6 +43,8 @@ class Tutor:
             self.memory = data["memory"]
             self.instructions = data["instructions"]
             self.starter_code = data["starter_code"]
+            self.age = data["age"]
+            self.name = data["name"]
         else:
             raise ValueError("Input must be a dictionary containing 'memory'")
 
@@ -113,7 +119,7 @@ class Tutor:
         messages.append(
             {
                 "role": SYSTEM,
-                "content": system_prompt(self.instructions, self.starter_code),
+                "content": system_prompt(self.instructions, self.starter_code, self.age, self.name),
             }
         )
         for entry in self.memory: