updated chatbot and saved the question number itself

app.py

@@ -14,76 +14,84 @@ from datetime import datetime
 from data_storage import store_user_data
 import uuid
 
-mongo_manager = MongoDBManager()
+
 
 # radio button and button rendering: https://www.gradio.app/docs/radio
-questionnaire_results = {
-    'Demographic Details': {},
-    'User Capabilities': {},
-    'Relationship to System': {},
-    'System Output Preferences': {}
-}
-question_answers = {
-    'q1': {
-        'Response': None,
-        'Assistance': None,
-        'CompletionTime': None,
-        'StartTime': None,
-        'EndTime': None
-    },
-    'q2': {
-        'Response': None,
-        'Assistance': None,
-        'CompletionTime': None,
-        'StartTime': None,
-        'EndTime': None
-    },
-    'q3': {
-        'Response': None,
-        'Assistance': None,
-        'CompletionTime': None,
-        'StartTime': None,
-        'EndTime': None
-    },
-    'q4': {
-        'Response': None,
-        'Assistance': None,
-        'CompletionTime': None,
-        'StartTime': None,
-        'EndTime': None
-    },
-    'q5': {
-        'Response': None,
-        'Assistance': None,
-        'CompletionTime': None,
-        'StartTime': None,
-        'EndTime': None
-    },
-    'q6': {
-        'Response': None,
-        'Assistance': None,
-        'CompletionTime': None,
-        'StartTime': None,
-        'EndTime': None
-    },
-    'q7': {
-        'Response': None,
-        'Assistance': None,
-        'CompletionTime': None,
-        'StartTime': None,
-        'EndTime': None
-    },
-    'q8': {
-        'Response': None,
-        'Assistance': None,
-        'CompletionTime': None,
-        'StartTime': None,
-        'EndTime': None
-    },
-}
 
 with gr.Blocks(title="wrAIte") as demo:
-    user_id = None
+    questionnaire_results = {
+        'Demographic Details': {},
+        'User Capabilities': {},
+        'Relationship to System': {},
+        'System Output Preferences': {}
+    }
+    question_answers = {
+        'q1': {
+            'Prompt': None,
+            'Response': None,
+            'Assistance': None,
+            'CompletionTime': None,
+            'StartTime': None,
+            'EndTime': None
+        },
+        'q2': {
+            'Prompt': None,
+            'Response': None,
+            'Assistance': None,
+            'CompletionTime': None,
+            'StartTime': None,
+            'EndTime': None
+        },
+        'q3': {
+            'Prompt': None,
+            'Response': None,
+            'Assistance': None,
+            'CompletionTime': None,
+            'StartTime': None,
+            'EndTime': None
+        },
+        'q4': {
+            'Prompt': None,
+            'Response': None,
+            'Assistance': None,
+            'CompletionTime': None,
+            'StartTime': None,
+            'EndTime': None
+        },
+        'q5': {
+            'Prompt': None,
+            'Response': None,
+            'Assistance': None,
+            'CompletionTime': None,
+            'StartTime': None,
+            'EndTime': None
+        },
+        'q6': {
+            'Prompt': None,
+            'Response': None,
+            'Assistance': None,
+            'CompletionTime': None,
+            'StartTime': None,
+            'EndTime': None
+        },
+        'q7': {
+            'Prompt': None,
+            'Response': None,
+            'Assistance': None,
+            'CompletionTime': None,
+            'StartTime': None,
+            'EndTime': None
+        },
+        'q8': {
+            'Prompt': None,
+            'Response': None,
+            'Assistance': None,
+            'CompletionTime': None,
+            'StartTime': None,
+            'EndTime': None
+        },
+    }
+    mongo_manager = MongoDBManager()
     # mongo_manager.create_user_record(user_id)
 
     with gr.Column("Survey Complete", render=True, visible=False) as completed_screen:

data/instructions/instr_email.md

@@ -1 +1 @@
-## Instructions: Write an email for the following scenario in under 300 words.
+## Instructions: Write an email for the following scenario in under 300 words:

data/instructions/instr_shortstory.md

@@ -1 +1 @@
-## Instructions: Write a short story (300 words or less) which only uses very simple words.
+## Instructions: Write a short story (300 words or less) which only uses very simple words. 

data/instructions/instr_summary.md

@@ -1 +1 @@
-## Instructions: Summarize the following article in 100 words or less.
+## Instructions: Summarize the following article in 100 words or less: 

data/instructions/instr_title.md

@@ -1 +1 @@
-## Instructions: Write a creative title for the following article using 10 words or less.
+## Instructions: Write a creative title for the following article using 10 words or less: 

model_generate.py

@@ -33,7 +33,9 @@ def query_chatbot(model, messages):
     chat_completion = client.chat.completions.create(
         messages=messages,
         model="gpt-3.5-turbo",
+
     )
+
     return chat_completion.choices[0].message.content
 
 
@@ -46,6 +48,8 @@ def chatbot_generate(user_newest_input, history, model, current_answer, initial_
     :return: The chatbot state, the history, the text, the submit button
     """
     # convert to openai model format
+    print('generating chatbot')
+    print()
     actual_model = {
         "chatgpt4": "gpt-4",
         "chatgpt": "gpt-3.5-turbo"
@@ -55,24 +59,26 @@ def chatbot_generate(user_newest_input, history, model, current_answer, initial_
     history.append(f"User: {user_newest_input.strip()}")
 
     # construct chat messages
-    chat_messages = [{"role": "system", "content": "You are a helpful assistant to a writer."}]
+    print(initial_txt)
+    chat_messages = [{"role": "system", "content": initial_txt}]
 
     # chat_messages = [{"role": "system", "content": initial_txt}]
-    current_txt = "My current answer to the prompt is as follows: " + current_answer + '. Now help me answer my question: '
+    current_txt = "My current answer to the instructions is as follows: " + current_answer + '. Now, assist me with the following: '
     for hist in history:
         if hist.startswith("User:"):
+            content = current_txt + hist[5:].strip()
             chat_messages.append(
                 {
                     "role": "user",
-                    "content": hist[5:].strip()
+                    "content": content
                 }
             )
         elif hist.startswith("Writing Assistant:"):
-            content = current_txt + hist[18:].strip()
+
             chat_messages.append(
                 {
                     "role": "assistant",
-                    "content": content
+                    "content": hist[18:].strip()
                 }
             )
         else:
@@ -80,6 +86,9 @@ def chatbot_generate(user_newest_input, history, model, current_answer, initial_
 
     # Get the generation from OpenAI
     if actual_model in ["gpt-3.5-turbo", "gpt-4"]:
+        print('generating chatbot')
+        print(actual_model)
+        print(chat_messages)
         ai_newest_output = query_chatbot(actual_model, chat_messages)
     else:
         raise NotImplementedError

user_data/06dfcf09-173f-4c6c-a29a-202da5c036f4/questionnaire_06dfcf09-173f-4c6c-a29a-202da5c036f4.json

@@ -0,0 +1,41 @@
+{
+    "Demographic Details": {
+        "Which of the following best describes your race or ethnicity?": "Caucasian",
+        "How would you describe your socioeconomic status?": "Low",
+        "What is your gender?": "Male",
+        "What is your age range?": "18-24",
+        "What is your primary language?": "English",
+        "What is the highest level of education you have completed?": "Some high school",
+        "Which field best describes your current profession?": "Arts and Entertainment",
+        "How comfortable are you using LLMs (Large Language Models, such as ChatGPT, Claude, Gemini, etc.)?": "Not comfortable at all",
+        "How much experience do you have using LLMs (Large Language Models, such as ChatGPT, Claude, Gemini, etc.)?": "No experience",
+        "How often do you use LLMs (Large Language Models, such as ChatGPT, Claude, Gemini, etc.)?": "Never"
+    },
+    "User Capabilities": {
+        "How would you rate your expertise in writing?": "Novice",
+        "How efficient do you consider yourself in writing tasks?": "Very inefficient",
+        "How comfortable are you with using writing-related technology?": "Not comfortable at all",
+        "How confident do you feel about your writing?": "Not confident",
+        "How often do you engage in creative or thoughtful writing?": "Never",
+        "What is your typical emotional state when writing?": "Very stressed",
+        "How often do you feel that you can empathize with characters or subjects in your writing?": "Never",
+        "How often do you experience focus issues or writer\u2019s block when writing?": "Always",
+        "Do you identify as neurotypical or neurodivergent?": "Neurotypical"
+    },
+    "Relationship to System": {
+        "How much control do you feel you have when using a writing assistant?": "None",
+        "How much ownership do you feel over your writings created with the help of a writing assistant?": "None",
+        "How concerned are you about plagiarism when using a writing assistant?": "Not concerned",
+        "How much do you trust the writing assistant to help you effectively?": "Not at all",
+        "How important is it that the writing assistant is available whenever you need it?": "Not important",
+        "How concerned are you about your privacy when using a writing assistant?": "Not concerned",
+        "How well do you understand how the writing assistant works, including its capabilities and limitations?": "Not at all"
+    },
+    "System Output Preferences": {
+        "How important is it for the outputs from the writing assistant to be grammatically and contextually coherent?": "Not important",
+        "How much do you value novelty and diversity in the outputs from the writing assistant?": "Not at all",
+        "How important is it for the writing assistant to provide explanations for its outputs?": "Not important",
+        "How concerned are you about potential biases in the outputs from the writing assistant?": "Not concerned",
+        "How important is personalization of the outputs from the writing assistant according to your specific needs and preferences?": "Not important"
+    }
+}

user_data/06dfcf09-173f-4c6c-a29a-202da5c036f4/questions_06dfcf09-173f-4c6c-a29a-202da5c036f4.json

@@ -0,0 +1,82 @@
+{
+    "q1": {
+        "Prompt": "p_shortstory1.md",
+        "Response": "\"hiiiiii\"",
+        "Assistance": null,
+        "CompletionTime": "0:00:04.327353",
+        "StartTime": "2024-05-04T09:44:47.860339",
+        "EndTime": "2024-05-04T09:44:52.187692",
+        "Assist": false,
+        "AssistanceHistory": null
+    },
+    "q2": {
+        "Prompt": "p_email4.md",
+        "Response": "\"hiiiiii\"",
+        "Assistance": null,
+        "CompletionTime": "0:00:03.771818",
+        "StartTime": "2024-05-04T09:44:52.187715",
+        "EndTime": "2024-05-04T09:44:55.959533",
+        "Assist": true,
+        "AssistanceHistory": []
+    },
+    "q3": {
+        "Prompt": "p_email7.md",
+        "Response": "\"hiiiiii\"",
+        "Assistance": null,
+        "CompletionTime": "0:00:02.195703",
+        "StartTime": "2024-05-04T09:44:55.959557",
+        "EndTime": "2024-05-04T09:44:58.155260",
+        "Assist": false,
+        "AssistanceHistory": null
+    },
+    "q4": {
+        "Prompt": "p_shortstory3.md",
+        "Response": "\"hiiiiii\"",
+        "Assistance": null,
+        "CompletionTime": "0:00:02.495594",
+        "StartTime": "2024-05-04T09:44:58.155287",
+        "EndTime": "2024-05-04T09:45:00.650881",
+        "Assist": true,
+        "AssistanceHistory": []
+    },
+    "q5": {
+        "Prompt": "p_title4.md",
+        "Response": "\"hiiiiii\"",
+        "Assistance": null,
+        "CompletionTime": "0:00:03.161611",
+        "StartTime": "2024-05-04T09:45:00.650915",
+        "EndTime": "2024-05-04T09:45:03.812526",
+        "Assist": false,
+        "AssistanceHistory": null
+    },
+    "q6": {
+        "Prompt": "p_summary8.md",
+        "Response": "\"hiiiiii\"",
+        "Assistance": null,
+        "CompletionTime": "0:00:02.965479",
+        "StartTime": "2024-05-04T09:45:03.812549",
+        "EndTime": "2024-05-04T09:45:06.778028",
+        "Assist": false,
+        "AssistanceHistory": null
+    },
+    "q7": {
+        "Prompt": "p_summary3.md",
+        "Response": "\"hiiiiii\"",
+        "Assistance": null,
+        "CompletionTime": "0:00:08.803938",
+        "StartTime": "2024-05-04T09:45:06.778056",
+        "EndTime": "2024-05-04T09:45:15.581994",
+        "Assist": false,
+        "AssistanceHistory": null
+    },
+    "q8": {
+        "Prompt": "p_title10.md",
+        "Response": "\"hiiiiii\"",
+        "Assistance": null,
+        "CompletionTime": "0:00:03.268502",
+        "StartTime": "2024-05-04T09:45:15.582013",
+        "EndTime": "2024-05-04T09:45:18.850515",
+        "Assist": true,
+        "AssistanceHistory": []
+    }
+}

user_data/2936e053-6bdf-456a-8529-d59b0aa59ae3/questionnaire_2936e053-6bdf-456a-8529-d59b0aa59ae3.json

@@ -0,0 +1,41 @@
+{
+    "Demographic Details": {
+        "What is your gender?": "Female",
+        "Which of the following best describes your race or ethnicity?": "Hispanic or Latino",
+        "How would you describe your socioeconomic status?": "Lower-middle",
+        "What is your primary language?": "Spanish",
+        "What is your age range?": "25-34",
+        "What is the highest level of education you have completed?": "High school graduate",
+        "Which field best describes your current profession?": "Business and Finance",
+        "How comfortable are you using LLMs (Large Language Models, such as ChatGPT, Claude, Gemini, etc.)?": "Slightly comfortable",
+        "How much experience do you have using LLMs (Large Language Models, such as ChatGPT, Claude, Gemini, etc.)?": "Less than 6 months",
+        "How often do you use LLMs (Large Language Models, such as ChatGPT, Claude, Gemini, etc.)?": "Rarely (less than once a month)"
+    },
+    "User Capabilities": {
+        "How would you rate your expertise in writing?": "Expert",
+        "How efficient do you consider yourself in writing tasks?": "Somewhat inefficient",
+        "How comfortable are you with using writing-related technology?": "Slightly comfortable",
+        "How confident do you feel about your writing?": "Slightly confident",
+        "How often do you engage in creative or thoughtful writing?": "Sometimes",
+        "What is your typical emotional state when writing?": "Somewhat stressed",
+        "How often do you feel that you can empathize with characters or subjects in your writing?": "Sometimes",
+        "How often do you experience focus issues or writer\u2019s block when writing?": "Sometimes",
+        "Do you identify as neurotypical or neurodivergent?": "Neurodivergent"
+    },
+    "Relationship to System": {
+        "How much control do you feel you have when using a writing assistant?": "A lot",
+        "How much ownership do you feel over your writings created with the help of a writing assistant?": "Most",
+        "How concerned are you about plagiarism when using a writing assistant?": "Moderately concerned",
+        "How much do you trust the writing assistant to help you effectively?": "A lot",
+        "How important is it that the writing assistant is available whenever you need it?": "Moderately important",
+        "How concerned are you about your privacy when using a writing assistant?": "Moderately concerned",
+        "How well do you understand how the writing assistant works, including its capabilities and limitations?": "Very well"
+    },
+    "System Output Preferences": {
+        "How important is it for the outputs from the writing assistant to be grammatically and contextually coherent?": "Moderately important",
+        "How much do you value novelty and diversity in the outputs from the writing assistant?": "Moderately",
+        "How important is it for the writing assistant to provide explanations for its outputs?": "Moderately important",
+        "How concerned are you about potential biases in the outputs from the writing assistant?": "Moderately concerned",
+        "How important is personalization of the outputs from the writing assistant according to your specific needs and preferences?": "Moderately important"
+    }
+}

user_data/2936e053-6bdf-456a-8529-d59b0aa59ae3/questions_2936e053-6bdf-456a-8529-d59b0aa59ae3.json

@@ -0,0 +1,82 @@
+{
+    "q1": {
+        "Response": "\"hiiiiii\"",
+        "Assistance": null,
+        "CompletionTime": "0:00:09.581804",
+        "StartTime": "2024-05-04T11:12:40.387481",
+        "EndTime": "2024-05-04T11:12:49.969285",
+        "Prompt": "p_email3.md",
+        "Assist": false,
+        "AssistanceHistory": null
+    },
+    "q2": {
+        "Response": "\"hiiiiii\"",
+        "Assistance": null,
+        "CompletionTime": "0:00:02.799734",
+        "StartTime": "2024-05-04T11:12:49.969311",
+        "EndTime": "2024-05-04T11:12:52.769045",
+        "Prompt": "p_email2.md",
+        "Assist": true,
+        "AssistanceHistory": []
+    },
+    "q3": {
+        "Response": "\"hiiiiii\"",
+        "Assistance": null,
+        "CompletionTime": "0:00:02.041394",
+        "StartTime": "2024-05-04T11:12:52.769134",
+        "EndTime": "2024-05-04T11:12:54.810528",
+        "Prompt": "p_summary10.md",
+        "Assist": false,
+        "AssistanceHistory": null
+    },
+    "q4": {
+        "Response": "\"hiiiiii\"",
+        "Assistance": null,
+        "CompletionTime": "0:00:02.954487",
+        "StartTime": "2024-05-04T11:12:54.810558",
+        "EndTime": "2024-05-04T11:12:57.765045",
+        "Prompt": "p_title10.md",
+        "Assist": true,
+        "AssistanceHistory": []
+    },
+    "q5": {
+        "Response": "\"hiiiiii\"",
+        "Assistance": null,
+        "CompletionTime": "0:00:02.705187",
+        "StartTime": "2024-05-04T11:12:57.765077",
+        "EndTime": "2024-05-04T11:13:00.470264",
+        "Prompt": "p_title1.md",
+        "Assist": false,
+        "AssistanceHistory": null
+    },
+    "q6": {
+        "Response": "\"hiiiiii\"",
+        "Assistance": null,
+        "CompletionTime": "0:00:04.280624",
+        "StartTime": "2024-05-04T11:13:00.470284",
+        "EndTime": "2024-05-04T11:13:04.750908",
+        "Prompt": "p_summary8.md",
+        "Assist": false,
+        "AssistanceHistory": null
+    },
+    "q7": {
+        "Response": "\"hiiiiii\"",
+        "Assistance": null,
+        "CompletionTime": "0:00:02.475190",
+        "StartTime": "2024-05-04T11:13:04.750932",
+        "EndTime": "2024-05-04T11:13:07.226122",
+        "Prompt": "p_shortstory10.md",
+        "Assist": false,
+        "AssistanceHistory": null
+    },
+    "q8": {
+        "Response": "\"hiiiiii\"",
+        "Assistance": null,
+        "CompletionTime": "0:00:03.265426",
+        "StartTime": "2024-05-04T11:13:07.226159",
+        "EndTime": "2024-05-04T11:13:10.491585",
+        "Prompt": "p_shortstory3.md",
+        "Assist": true,
+        "AssistanceHistory": []
+    }
+}

user_data/66d61164-46eb-4935-aa1e-0515957bb8a5/questionnaire_66d61164-46eb-4935-aa1e-0515957bb8a5.json

@@ -0,0 +1,41 @@
+{
+    "Demographic Details": {
+        "What is your gender?": "Male",
+        "Which of the following best describes your race or ethnicity?": "Asian",
+        "How would you describe your socioeconomic status?": "High",
+        "What is your primary language?": "Other",
+        "What is your age range?": "55-64",
+        "What is the highest level of education you have completed?": "Some college",
+        "Which field best describes your current profession?": "Engineering and Technology",
+        "How comfortable are you using LLMs (Large Language Models, such as ChatGPT, Claude, Gemini, etc.)?": "Very comfortable",
+        "How much experience do you have using LLMs (Large Language Models, such as ChatGPT, Claude, Gemini, etc.)?": "1 to 2 years",
+        "How often do you use LLMs (Large Language Models, such as ChatGPT, Claude, Gemini, etc.)?": "Occasionally (once a month to once a week)"
+    },
+    "User Capabilities": {
+        "How would you rate your expertise in writing?": "Expert",
+        "How efficient do you consider yourself in writing tasks?": "Somewhat efficient",
+        "How comfortable are you with using writing-related technology?": "Moderately comfortable",
+        "How confident do you feel about your writing?": "Very confident",
+        "How often do you engage in creative or thoughtful writing?": "Always",
+        "What is your typical emotional state when writing?": "Somewhat relaxed",
+        "How often do you feel that you can empathize with characters or subjects in your writing?": "Always",
+        "How often do you experience focus issues or writer\u2019s block when writing?": "Never",
+        "Do you identify as neurotypical or neurodivergent?": "Prefer not to say"
+    },
+    "Relationship to System": {
+        "How much control do you feel you have when using a writing assistant?": "Some",
+        "How much ownership do you feel over your writings created with the help of a writing assistant?": "Some",
+        "How concerned are you about plagiarism when using a writing assistant?": "Very concerned",
+        "How much do you trust the writing assistant to help you effectively?": "Completely",
+        "How important is it that the writing assistant is available whenever you need it?": "Moderately important",
+        "How concerned are you about your privacy when using a writing assistant?": "Moderately concerned",
+        "How well do you understand how the writing assistant works, including its capabilities and limitations?": "Completely"
+    },
+    "System Output Preferences": {
+        "How important is it for the outputs from the writing assistant to be grammatically and contextually coherent?": "Slightly important",
+        "How much do you value novelty and diversity in the outputs from the writing assistant?": "A lot",
+        "How important is it for the writing assistant to provide explanations for its outputs?": "Moderately important",
+        "How concerned are you about potential biases in the outputs from the writing assistant?": "Moderately concerned",
+        "How important is personalization of the outputs from the writing assistant according to your specific needs and preferences?": "Moderately important"
+    }
+}

user_data/66d61164-46eb-4935-aa1e-0515957bb8a5/questions_66d61164-46eb-4935-aa1e-0515957bb8a5.json

@@ -0,0 +1,82 @@
+{
+    "q1": {
+        "Prompt": "p_email2.md",
+        "Response": "\"hjkhjhjk\"",
+        "Assistance": null,
+        "CompletionTime": "0:03:32.809494",
+        "StartTime": "2024-05-04T10:42:25.040424",
+        "EndTime": "2024-05-04T10:45:57.849918",
+        "Assist": false,
+        "AssistanceHistory": null
+    },
+    "q2": {
+        "Prompt": "p_title6.md",
+        "Response": "\"jkjkjkl\"",
+        "Assistance": null,
+        "CompletionTime": "0:00:09.948107",
+        "StartTime": "2024-05-04T10:45:57.849943",
+        "EndTime": "2024-05-04T10:46:07.798050",
+        "Assist": true,
+        "AssistanceHistory": []
+    },
+    "q3": {
+        "Prompt": "p_title10.md",
+        "Response": "\"a\"",
+        "Assistance": null,
+        "CompletionTime": "0:00:02.492447",
+        "StartTime": "2024-05-04T10:46:07.798087",
+        "EndTime": "2024-05-04T10:46:10.290534",
+        "Assist": false,
+        "AssistanceHistory": null
+    },
+    "q4": {
+        "Prompt": "p_shortstory9.md",
+        "Response": "\"a\"",
+        "Assistance": null,
+        "CompletionTime": "0:00:02.755677",
+        "StartTime": "2024-05-04T10:46:10.290559",
+        "EndTime": "2024-05-04T10:46:13.046236",
+        "Assist": true,
+        "AssistanceHistory": []
+    },
+    "q5": {
+        "Prompt": "p_shortstory3.md",
+        "Response": "\"a\"",
+        "Assistance": null,
+        "CompletionTime": "0:00:02.038245",
+        "StartTime": "2024-05-04T10:46:13.046257",
+        "EndTime": "2024-05-04T10:46:15.084502",
+        "Assist": false,
+        "AssistanceHistory": null
+    },
+    "q6": {
+        "Prompt": "p_summary5.md",
+        "Response": "\"a\"",
+        "Assistance": null,
+        "CompletionTime": "0:00:02.554129",
+        "StartTime": "2024-05-04T10:46:15.084540",
+        "EndTime": "2024-05-04T10:46:17.638669",
+        "Assist": false,
+        "AssistanceHistory": null
+    },
+    "q7": {
+        "Prompt": "p_summary4.md",
+        "Response": "\"a\"",
+        "Assistance": null,
+        "CompletionTime": "0:00:02.591096",
+        "StartTime": "2024-05-04T10:46:17.638698",
+        "EndTime": "2024-05-04T10:46:20.229794",
+        "Assist": false,
+        "AssistanceHistory": null
+    },
+    "q8": {
+        "Prompt": "p_email3.md",
+        "Response": "\"a\"",
+        "Assistance": null,
+        "CompletionTime": "0:00:02.499396",
+        "StartTime": "2024-05-04T10:46:20.229819",
+        "EndTime": "2024-05-04T10:46:22.729215",
+        "Assist": true,
+        "AssistanceHistory": []
+    }
+}

user_data/7d4117f5-2f82-45fe-af2d-a1b301b22009/questions_7d4117f5-2f82-45fe-af2d-a1b301b22009.json

@@ -0,0 +1,82 @@
+{
+    "q1": {
+        "Prompt": "p_shortstory8.md",
+        "Response": "\"hey hey hey\"",
+        "Assistance": null,
+        "CompletionTime": "0:00:06.299049",
+        "StartTime": "2024-05-04T09:25:56.233446",
+        "EndTime": "2024-05-04T09:26:02.532495",
+        "Assist": false,
+        "AssistanceHistory": null
+    },
+    "q2": {
+        "Prompt": "p_shortstory5.md",
+        "Response": "\"hey hey hey\"",
+        "Assistance": null,
+        "CompletionTime": "0:00:03.164310",
+        "StartTime": "2024-05-04T09:26:02.532516",
+        "EndTime": "2024-05-04T09:26:05.696826",
+        "Assist": true,
+        "AssistanceHistory": []
+    },
+    "q3": {
+        "Prompt": "p_email6.md",
+        "Response": "\"hey hey hey\"",
+        "Assistance": null,
+        "CompletionTime": "0:00:02.391042",
+        "StartTime": "2024-05-04T09:26:05.696851",
+        "EndTime": "2024-05-04T09:26:08.087893",
+        "Assist": false,
+        "AssistanceHistory": null
+    },
+    "q4": {
+        "Prompt": "p_email9.md",
+        "Response": "\"hey hey hey\"",
+        "Assistance": null,
+        "CompletionTime": "0:00:02.806623",
+        "StartTime": "2024-05-04T09:26:08.087922",
+        "EndTime": "2024-05-04T09:26:10.894545",
+        "Assist": true,
+        "AssistanceHistory": []
+    },
+    "q5": {
+        "Prompt": "p_summary8.md",
+        "Response": "\"hey hey hey\"",
+        "Assistance": null,
+        "CompletionTime": "0:00:03.158417",
+        "StartTime": "2024-05-04T09:26:10.894562",
+        "EndTime": "2024-05-04T09:26:14.052979",
+        "Assist": false,
+        "AssistanceHistory": null
+    },
+    "q6": {
+        "Prompt": "p_title4.md",
+        "Response": "\"hey hey hey\"",
+        "Assistance": null,
+        "CompletionTime": "0:00:02.961982",
+        "StartTime": "2024-05-04T09:26:14.053015",
+        "EndTime": "2024-05-04T09:26:17.014997",
+        "Assist": false,
+        "AssistanceHistory": null
+    },
+    "q7": {
+        "Prompt": "p_title7.md",
+        "Response": "\"hey hey hey\"",
+        "Assistance": null,
+        "CompletionTime": "0:00:03.619259",
+        "StartTime": "2024-05-04T09:26:17.015020",
+        "EndTime": "2024-05-04T09:26:20.634279",
+        "Assist": false,
+        "AssistanceHistory": null
+    },
+    "q8": {
+        "Prompt": "p_summary9.md",
+        "Response": "\"hey hey hey\"",
+        "Assistance": null,
+        "CompletionTime": "0:00:03.717442",
+        "StartTime": "2024-05-04T09:26:20.634306",
+        "EndTime": "2024-05-04T09:26:24.351748",
+        "Assist": true,
+        "AssistanceHistory": []
+    }
+}

user_data/87617e37-4645-49e3-a970-b4ab8c5fb965/questionnaire_87617e37-4645-49e3-a970-b4ab8c5fb965.json

@@ -0,0 +1,41 @@
+{
+    "Demographic Details": {
+        "What is your gender?": "Prefer not to say",
+        "Which of the following best describes your race or ethnicity?": "African American",
+        "How would you describe your socioeconomic status?": "Upper-middle",
+        "What is your primary language?": "Other",
+        "What is your age range?": "55-64",
+        "What is the highest level of education you have completed?": "Some college",
+        "Which field best describes your current profession?": "Education",
+        "How comfortable are you using LLMs (Large Language Models, such as ChatGPT, Claude, Gemini, etc.)?": "Moderately comfortable",
+        "How much experience do you have using LLMs (Large Language Models, such as ChatGPT, Claude, Gemini, etc.)?": "6 months to 1 year",
+        "How often do you use LLMs (Large Language Models, such as ChatGPT, Claude, Gemini, etc.)?": "Occasionally (once a month to once a week)"
+    },
+    "User Capabilities": {
+        "How would you rate your expertise in writing?": "Professional writer",
+        "How efficient do you consider yourself in writing tasks?": "Neutral",
+        "How comfortable are you with using writing-related technology?": "Moderately comfortable",
+        "How confident do you feel about your writing?": "Very confident",
+        "How often do you engage in creative or thoughtful writing?": "Always",
+        "What is your typical emotional state when writing?": "Somewhat relaxed",
+        "How often do you feel that you can empathize with characters or subjects in your writing?": "Always",
+        "How often do you experience focus issues or writer\u2019s block when writing?": "Never",
+        "Do you identify as neurotypical or neurodivergent?": "Prefer not to say"
+    },
+    "Relationship to System": {
+        "How much control do you feel you have when using a writing assistant?": "A lot",
+        "How much ownership do you feel over your writings created with the help of a writing assistant?": "Most",
+        "How concerned are you about plagiarism when using a writing assistant?": "Moderately concerned",
+        "How much do you trust the writing assistant to help you effectively?": "A lot",
+        "How important is it that the writing assistant is available whenever you need it?": "Moderately important",
+        "How concerned are you about your privacy when using a writing assistant?": "Moderately concerned",
+        "How well do you understand how the writing assistant works, including its capabilities and limitations?": "Very well"
+    },
+    "System Output Preferences": {
+        "How important is it for the outputs from the writing assistant to be grammatically and contextually coherent?": "Moderately important",
+        "How much do you value novelty and diversity in the outputs from the writing assistant?": "A lot",
+        "How important is it for the writing assistant to provide explanations for its outputs?": "Moderately important",
+        "How concerned are you about potential biases in the outputs from the writing assistant?": "Moderately concerned",
+        "How important is personalization of the outputs from the writing assistant according to your specific needs and preferences?": "Moderately important"
+    }
+}

user_data/87617e37-4645-49e3-a970-b4ab8c5fb965/questions_87617e37-4645-49e3-a970-b4ab8c5fb965.json

@@ -0,0 +1,82 @@
+{
+    "q1": {
+        "Prompt": "p_shortstory8.md",
+        "Response": "\"then then then\"",
+        "Assistance": null,
+        "CompletionTime": "0:00:11.568208",
+        "StartTime": "2024-05-04T09:26:42.698525",
+        "EndTime": "2024-05-04T09:26:54.266733",
+        "Assist": false,
+        "AssistanceHistory": null
+    },
+    "q2": {
+        "Prompt": "p_shortstory5.md",
+        "Response": "\"then then then\"",
+        "Assistance": null,
+        "CompletionTime": "0:00:02.794306",
+        "StartTime": "2024-05-04T09:26:54.266781",
+        "EndTime": "2024-05-04T09:26:57.061087",
+        "Assist": true,
+        "AssistanceHistory": []
+    },
+    "q3": {
+        "Prompt": "p_email6.md",
+        "Response": "\"then then then\"",
+        "Assistance": null,
+        "CompletionTime": "0:00:02.196984",
+        "StartTime": "2024-05-04T09:26:57.061102",
+        "EndTime": "2024-05-04T09:26:59.258086",
+        "Assist": false,
+        "AssistanceHistory": null
+    },
+    "q4": {
+        "Prompt": "p_email9.md",
+        "Response": "\"then then then\"",
+        "Assistance": null,
+        "CompletionTime": "0:00:03.102160",
+        "StartTime": "2024-05-04T09:26:59.258120",
+        "EndTime": "2024-05-04T09:27:02.360280",
+        "Assist": true,
+        "AssistanceHistory": []
+    },
+    "q5": {
+        "Prompt": "p_summary8.md",
+        "Response": "\"then then then\"",
+        "Assistance": null,
+        "CompletionTime": "0:00:02.862572",
+        "StartTime": "2024-05-04T09:27:02.360305",
+        "EndTime": "2024-05-04T09:27:05.222877",
+        "Assist": false,
+        "AssistanceHistory": null
+    },
+    "q6": {
+        "Prompt": "p_title4.md",
+        "Response": "\"then then then\"",
+        "Assistance": null,
+        "CompletionTime": "0:00:07.938450",
+        "StartTime": "2024-05-04T09:27:05.222893",
+        "EndTime": "2024-05-04T09:27:13.161343",
+        "Assist": false,
+        "AssistanceHistory": null
+    },
+    "q7": {
+        "Prompt": "p_title7.md",
+        "Response": "\"then then then\"",
+        "Assistance": null,
+        "CompletionTime": "0:00:03.164349",
+        "StartTime": "2024-05-04T09:27:13.161382",
+        "EndTime": "2024-05-04T09:27:16.325731",
+        "Assist": false,
+        "AssistanceHistory": null
+    },
+    "q8": {
+        "Prompt": "p_summary9.md",
+        "Response": "\"then then then\"",
+        "Assistance": null,
+        "CompletionTime": "0:00:03.512689",
+        "StartTime": "2024-05-04T09:27:16.325769",
+        "EndTime": "2024-05-04T09:27:19.838458",
+        "Assist": true,
+        "AssistanceHistory": []
+    }
+}

user_data/c77ecadb-3257-49ef-acc1-137fe9e61d4c/questionnaire_c77ecadb-3257-49ef-acc1-137fe9e61d4c.json

@@ -0,0 +1,41 @@
+{
+    "Demographic Details": {
+        "What is your gender?": "Male",
+        "Which of the following best describes your race or ethnicity?": "Caucasian",
+        "How would you describe your socioeconomic status?": "Low",
+        "What is your primary language?": "English",
+        "What is your age range?": "18-24",
+        "What is the highest level of education you have completed?": "Some high school",
+        "Which field best describes your current profession?": "Arts and Entertainment",
+        "How comfortable are you using LLMs (Large Language Models, such as ChatGPT, Claude, Gemini, etc.)?": "Not comfortable at all",
+        "How much experience do you have using LLMs (Large Language Models, such as ChatGPT, Claude, Gemini, etc.)?": "No experience",
+        "How often do you use LLMs (Large Language Models, such as ChatGPT, Claude, Gemini, etc.)?": "Never"
+    },
+    "User Capabilities": {
+        "How would you rate your expertise in writing?": "Novice",
+        "How efficient do you consider yourself in writing tasks?": "Very inefficient",
+        "How comfortable are you with using writing-related technology?": "Not comfortable at all",
+        "How confident do you feel about your writing?": "Not confident",
+        "How often do you engage in creative or thoughtful writing?": "Never",
+        "What is your typical emotional state when writing?": "Very stressed",
+        "How often do you feel that you can empathize with characters or subjects in your writing?": "Never",
+        "How often do you experience focus issues or writer\u2019s block when writing?": "Always",
+        "Do you identify as neurotypical or neurodivergent?": "Neurotypical"
+    },
+    "Relationship to System": {
+        "How much control do you feel you have when using a writing assistant?": "Little",
+        "How much ownership do you feel over your writings created with the help of a writing assistant?": "Little",
+        "How concerned are you about plagiarism when using a writing assistant?": "Not concerned",
+        "How much do you trust the writing assistant to help you effectively?": "A little",
+        "How important is it that the writing assistant is available whenever you need it?": "Slightly important",
+        "How concerned are you about your privacy when using a writing assistant?": "Slightly concerned",
+        "How well do you understand how the writing assistant works, including its capabilities and limitations?": "Slightly"
+    },
+    "System Output Preferences": {
+        "How important is it for the outputs from the writing assistant to be grammatically and contextually coherent?": "Slightly important",
+        "How much do you value novelty and diversity in the outputs from the writing assistant?": "A little",
+        "How important is it for the writing assistant to provide explanations for its outputs?": "Slightly important",
+        "How concerned are you about potential biases in the outputs from the writing assistant?": "Slightly concerned",
+        "How important is personalization of the outputs from the writing assistant according to your specific needs and preferences?": "Slightly important"
+    }
+}

user_data/c77ecadb-3257-49ef-acc1-137fe9e61d4c/questions_c77ecadb-3257-49ef-acc1-137fe9e61d4c.json

@@ -0,0 +1,82 @@
+{
+    "q1": {
+        "Response": "\"hey hey hey \"",
+        "Assistance": null,
+        "CompletionTime": "0:00:11.007788",
+        "StartTime": "2024-05-04T10:59:20.784250",
+        "EndTime": "2024-05-04T10:59:31.792038",
+        "Prompt": "p_title3.md",
+        "Assist": false,
+        "AssistanceHistory": null
+    },
+    "q2": {
+        "Response": "\"hey hey hey \"",
+        "Assistance": null,
+        "CompletionTime": "0:00:03.008481",
+        "StartTime": "2024-05-04T10:59:31.792061",
+        "EndTime": "2024-05-04T10:59:34.800542",
+        "Prompt": "p_title5.md",
+        "Assist": true,
+        "AssistanceHistory": []
+    },
+    "q3": {
+        "Response": "\"hey hey hey \"",
+        "Assistance": null,
+        "CompletionTime": "0:00:02.040568",
+        "StartTime": "2024-05-04T10:59:34.800572",
+        "EndTime": "2024-05-04T10:59:36.841140",
+        "Prompt": "p_shortstory6.md",
+        "Assist": false,
+        "AssistanceHistory": null
+    },
+    "q4": {
+        "Response": "\"hey hey hey \"",
+        "Assistance": null,
+        "CompletionTime": "0:00:03.469423",
+        "StartTime": "2024-05-04T10:59:36.841162",
+        "EndTime": "2024-05-04T10:59:40.310585",
+        "Prompt": "p_shortstory1.md",
+        "Assist": true,
+        "AssistanceHistory": []
+    },
+    "q5": {
+        "Response": "\"hey hey hey \"",
+        "Assistance": null,
+        "CompletionTime": "0:00:01.888858",
+        "StartTime": "2024-05-04T10:59:40.310618",
+        "EndTime": "2024-05-04T10:59:42.199476",
+        "Prompt": "p_email6.md",
+        "Assist": false,
+        "AssistanceHistory": null
+    },
+    "q6": {
+        "Response": "\"hey hey hey \"",
+        "Assistance": null,
+        "CompletionTime": "0:00:03.580783",
+        "StartTime": "2024-05-04T10:59:42.199512",
+        "EndTime": "2024-05-04T10:59:45.780295",
+        "Prompt": "p_summary3.md",
+        "Assist": false,
+        "AssistanceHistory": null
+    },
+    "q7": {
+        "Response": "\"hey hey hey \"",
+        "Assistance": null,
+        "CompletionTime": "0:00:02.848358",
+        "StartTime": "2024-05-04T10:59:45.780331",
+        "EndTime": "2024-05-04T10:59:48.628689",
+        "Prompt": "p_summary8.md",
+        "Assist": false,
+        "AssistanceHistory": null
+    },
+    "q8": {
+        "Response": "\"hey hey hey \"",
+        "Assistance": null,
+        "CompletionTime": "0:00:02.990456",
+        "StartTime": "2024-05-04T10:59:48.628719",
+        "EndTime": "2024-05-04T10:59:51.619175",
+        "Prompt": "p_email1.md",
+        "Assist": true,
+        "AssistanceHistory": []
+    }
+}

user_data/da0733fe-078f-443a-9299-1413ff2f3a18/questionnaire_da0733fe-078f-443a-9299-1413ff2f3a18.json

@@ -0,0 +1,41 @@
+{
+    "Demographic Details": {
+        "What is your gender?": "Non-binary",
+        "Which of the following best describes your race or ethnicity?": "Hispanic or Latino",
+        "How would you describe your socioeconomic status?": "Middle",
+        "What is your primary language?": "Other",
+        "What is your age range?": "45-54",
+        "What is the highest level of education you have completed?": "High school graduate",
+        "Which field best describes your current profession?": "Business and Finance",
+        "How comfortable are you using LLMs (Large Language Models, such as ChatGPT, Claude, Gemini, etc.)?": "Moderately comfortable",
+        "How much experience do you have using LLMs (Large Language Models, such as ChatGPT, Claude, Gemini, etc.)?": "6 months to 1 year",
+        "How often do you use LLMs (Large Language Models, such as ChatGPT, Claude, Gemini, etc.)?": "Occasionally (once a month to once a week)"
+    },
+    "User Capabilities": {
+        "How would you rate your expertise in writing?": "Expert",
+        "How efficient do you consider yourself in writing tasks?": "Somewhat inefficient",
+        "How comfortable are you with using writing-related technology?": "Slightly comfortable",
+        "How confident do you feel about your writing?": "Slightly confident",
+        "How often do you engage in creative or thoughtful writing?": "Sometimes",
+        "What is your typical emotional state when writing?": "Somewhat stressed",
+        "How often do you feel that you can empathize with characters or subjects in your writing?": "Often",
+        "How often do you experience focus issues or writer\u2019s block when writing?": "Rarely",
+        "Do you identify as neurotypical or neurodivergent?": "Prefer not to say"
+    },
+    "Relationship to System": {
+        "How much control do you feel you have when using a writing assistant?": "A lot",
+        "How much ownership do you feel over your writings created with the help of a writing assistant?": "Some",
+        "How concerned are you about plagiarism when using a writing assistant?": "Moderately concerned",
+        "How much do you trust the writing assistant to help you effectively?": "A lot",
+        "How important is it that the writing assistant is available whenever you need it?": "Moderately important",
+        "How concerned are you about your privacy when using a writing assistant?": "Moderately concerned",
+        "How well do you understand how the writing assistant works, including its capabilities and limitations?": "Very well"
+    },
+    "System Output Preferences": {
+        "How important is it for the outputs from the writing assistant to be grammatically and contextually coherent?": "Slightly important",
+        "How much do you value novelty and diversity in the outputs from the writing assistant?": "Moderately",
+        "How important is it for the writing assistant to provide explanations for its outputs?": "Slightly important",
+        "How concerned are you about potential biases in the outputs from the writing assistant?": "Slightly concerned",
+        "How important is personalization of the outputs from the writing assistant according to your specific needs and preferences?": "Slightly important"
+    }
+}

user_data/da0733fe-078f-443a-9299-1413ff2f3a18/questions_da0733fe-078f-443a-9299-1413ff2f3a18.json

@@ -0,0 +1,82 @@
+{
+    "q1": {
+        "Response": "\"hiiiiii\"",
+        "Assistance": null,
+        "CompletionTime": "0:00:05.244582",
+        "StartTime": "2024-05-04T11:02:02.649323",
+        "EndTime": "2024-05-04T11:02:07.893905",
+        "Prompt": "p_email10.md",
+        "Assist": false,
+        "AssistanceHistory": null
+    },
+    "q2": {
+        "Response": "\"hiiiiii\"",
+        "Assistance": null,
+        "CompletionTime": "0:00:03.409426",
+        "StartTime": "2024-05-04T11:02:07.893934",
+        "EndTime": "2024-05-04T11:02:11.303360",
+        "Prompt": "p_summary4.md",
+        "Assist": true,
+        "AssistanceHistory": []
+    },
+    "q3": {
+        "Response": "\"hiiiiii\"",
+        "Assistance": null,
+        "CompletionTime": "0:00:06.539088",
+        "StartTime": "2024-05-04T11:02:11.303381",
+        "EndTime": "2024-05-04T11:02:17.842469",
+        "Prompt": "p_title1.md",
+        "Assist": false,
+        "AssistanceHistory": null
+    },
+    "q4": {
+        "Response": "\"hiiiiii\"",
+        "Assistance": null,
+        "CompletionTime": "0:00:03.311494",
+        "StartTime": "2024-05-04T11:02:17.842493",
+        "EndTime": "2024-05-04T11:02:21.153987",
+        "Prompt": "p_email6.md",
+        "Assist": true,
+        "AssistanceHistory": []
+    },
+    "q5": {
+        "Response": "\"hiiiiii\"",
+        "Assistance": null,
+        "CompletionTime": "0:00:03.410036",
+        "StartTime": "2024-05-04T11:02:21.154022",
+        "EndTime": "2024-05-04T11:02:24.564058",
+        "Prompt": "p_summary8.md",
+        "Assist": false,
+        "AssistanceHistory": null
+    },
+    "q6": {
+        "Response": "\"hiiiiii\"",
+        "Assistance": null,
+        "CompletionTime": "0:00:03.214901",
+        "StartTime": "2024-05-04T11:02:24.564082",
+        "EndTime": "2024-05-04T11:02:27.778983",
+        "Prompt": "p_title9.md",
+        "Assist": false,
+        "AssistanceHistory": null
+    },
+    "q7": {
+        "Response": "\"hiiiiii\"",
+        "Assistance": null,
+        "CompletionTime": "0:00:03.409163",
+        "StartTime": "2024-05-04T11:02:27.779014",
+        "EndTime": "2024-05-04T11:02:31.188177",
+        "Prompt": "p_shortstory4.md",
+        "Assist": false,
+        "AssistanceHistory": null
+    },
+    "q8": {
+        "Response": "\"hiiiiii\"",
+        "Assistance": null,
+        "CompletionTime": "0:00:03.369873",
+        "StartTime": "2024-05-04T11:02:31.188196",
+        "EndTime": "2024-05-04T11:02:34.558069",
+        "Prompt": "p_shortstory9.md",
+        "Assist": true,
+        "AssistanceHistory": []
+    }
+}

utils.py

@@ -18,12 +18,10 @@ def process_markdown(prompt, question):
         markdown_content = md_file.read()
         return markdown_content
 
-
 def strip_markdown(text):
     """
-    Strips common markdown formatting from a string.
+    Strips common markdown formatting and `<span>` tags from a string.
     """
-    # Patterns to remove from the Markdown text
     patterns = [
         r'\!\[[^\]]*\]\([^\)]+\)',  # Images
         r'\[[^\]]*\]\([^\)]+\)',  # Links
@@ -35,11 +33,13 @@ def strip_markdown(text):
         r'\>(.*?)\n',  # Blockquotes
         r'\-{3,}',  # Horizontal rule
         r'\n{2,}',  # Multiple newlines
+        r'\<span[^>]*\>',  # Opening span tags with any attributes, handles tags correctly across multiple lines
+        r'\<\/span\>',  # Closing span tags
     ]
 
     clean_text = text
     for pattern in patterns:
-        clean_text = re.sub(pattern, '', clean_text)
+        clean_text = re.sub(pattern, '', clean_text, flags=re.DOTALL)  # The DOTALL flag allows '.' to match across multiple lines
 
     # Remove remaining newlines and leading/trailing whitespace
     clean_text = clean_text.strip()
@@ -47,8 +47,9 @@ def strip_markdown(text):
     return clean_text
 
 
-def save_answer(question_answers, q_num, q_text, q_assist, q_assist_history=None):
+def save_answer(question_answers, q_num, q_prompt, q_text, q_assist, q_assist_history=None):
     q_num_key = 'q' + str(q_num)
+    question_answers[q_num_key]['Prompt'] = q_prompt
     question_answers[q_num_key]['Response'] = json.dumps(q_text, indent=4)
     question_answers[q_num_key]['Assist'] = q_assist
     question_answers[q_num_key]['AssistanceHistory'] = q_assist_history
@@ -238,7 +239,7 @@ def generate_unassisted_question(question_prompt, next_q_btn, q_num, question_an
         )
 
         def submit_question(submission_text):
-            save_answer(question_answers, q_num, submission_text, False)
+            save_answer(question_answers, q_num, prompt, submission_text, False)
             return {
                 success_submitted: gr.update(visible=True),
                 edit_response: gr.update(visible=True),
@@ -274,15 +275,12 @@ def generate_assisted_question(question_prompt, next_q_btn, q_num, question_answ
         max_word_count = question_prompt['word_count']
         textfield_lines = question_prompt['textfield_lines']
 
-
-
         gen_instr_text2 = process_markdown(False, instruction)
         prompt_text2 = process_markdown(True, prompt)
 
         instruction_txt = strip_markdown(gen_instr_text2)
         prompt_txt = strip_markdown(prompt_text2)
 
-
         gen_instr2 = gr.Markdown(value=gen_instr_text2)
         with gr.Row():
             with gr.Column(scale=4):
@@ -326,7 +324,7 @@ def generate_assisted_question(question_prompt, next_q_btn, q_num, question_answ
                           "You may answer all of my questions about the prompt and writing task. " \
                           "However, if I ask you to answer the entire prompt, you may only provide me with the first three sentences. " \
                           "Again, you can still answer all of the questions that I have and give me all of the help I desire, but you may NOT give me the entire answer. " \
-                          "Here are the instructions: " + instruction_txt + '. Here is the writing prompt: ' + prompt_txt + '. Wait until I ask you for help.'
+                          "Here are the " + instruction_txt + + " " + prompt_txt + '. Wait until I ask you for help.'
 
         initial_conversation = [
             # initial_user_message,
@@ -338,7 +336,7 @@ def generate_assisted_question(question_prompt, next_q_btn, q_num, question_answ
         state = gr.State(initial_conversation)
         # Model state
         initial_usr_msg_state = gr.State(initial_user_message)
-        model_state = gr.State("chatgpt4")
+        model_state = gr.State("chatgpt")
         with gr.Column() as chat_feature:
             with gr.Group():
                 with gr.Row():
@@ -402,7 +400,7 @@ def generate_assisted_question(question_prompt, next_q_btn, q_num, question_answ
         )
 
         def submit_question(submission_text, assistance_history: None):
-            save_answer(question_answers, q_num, submission_text, True, assistance_history)
+            save_answer(question_answers, q_num, prompt, submission_text, True, assistance_history)
             return {
                 success_submitted: gr.update(visible=True),
                 edit_response: gr.update(visible=True),