Update messages for demo

app.py

@@ -35,13 +35,16 @@ def get_assistant_response(gpt_history):
     print("\n" + response_message)
     return response_message
 
-intro_message = "Hi, I'm an AI powered college counselor from Cledge! Do you want help brainstorming ideas for your personal statement?"
+hardcoded = {
+    1: "Hi, I'm an AI powered college counselor from Cledge! What prompt do you want help with?",
+    3: "Let me know if you want more suggestions."
+}
 
 instructions = {
-    2: '''Generate 5 questions to help them brainstorm.''',
-    3: "Based on these responses, ask follow up questions that help them narrow down the focus of the essay",
-    5: "Based on these responses, ask follow up questions that help them identify key themes in the essay",
-    7: "Based on these responses, think of 5 ideas for personal statement essays. Write a synopsis of each idea."
+    2: "Based on these responses, generate 5 questions to help them brainstorm.",
+    # 3: "Based on these responses, ask follow up questions that help them narrow down the focus of the essay",
+    # 4: "Based on these responses, ask follow up questions that help them identify key themes in the essay",
+    3: "Based on these responses, think of 5 ideas for personal statement essays. Write a synopsis of each idea.",
 }
 
 def grad_demo():
@@ -54,9 +57,14 @@ def grad_demo():
         def bot(history):
             step = len(history)
             print(f"STEP: {step}")
+            bot_message = ""
             if step in instructions:
+                if step > 3:
+                    step = (step - 2) % 2 + 2
                 gpt_history.append({"role": ChatRoles.SYSTEM, "content": instructions[step]})
-            bot_message = get_assistant_response(gpt_history) if step > 1 else intro_message
+                bot_message = get_assistant_response(gpt_history)
+            if step in hardcoded:
+                bot_message = f"{bot_message}\n\n {hardcoded[step]}"
             history[-1][1] = bot_message
             gpt_history.append({"role": ChatRoles.ASSISTANT, "content": bot_message})
             print(f"Length of gpt_history: {gpt_history}")