done

app.py

@@ -18,7 +18,9 @@ import uuid
 
 # radio button and button rendering: https://www.gradio.app/docs/radio
 
-with gr.Blocks(title="wrAIte") as demo:
+with gr.Blocks(title="wrAIte", theme='gradio/soft') as demo:
+    # gr.Image("/file=wraite.png", interactive=False)
+
     questionnaire_results = {
         'Demographic Details': {},
         'User Capabilities': {},
@@ -216,7 +218,7 @@ with gr.Blocks(title="wrAIte") as demo:
             )
         with gr.Column("Question 6", render=True, visible=False) as question6:
             next_q6_btn = gr.Button("Next Question", render=False, visible=False, variant="primary")
-            q6 = generate_unassisted_question(question_prompts[5], next_q6_btn, 6, question_answers)
+            q6 = generate_assisted_question(question_prompts[5], next_q6_btn, 6, question_answers)
             q6.render()
             next_q6_btn.render()
             next_q6_btn.click(

model_generate.py

@@ -29,7 +29,6 @@ def query_a_chat_completion(model, messages):
 
 
 def query_chatbot(model, messages):
-    print(messages)
     assert model in ["gpt-3.5-turbo", "gpt-4"]
     chat_completion = client.chat.completions.create(
         messages=messages,
@@ -38,7 +37,7 @@ def query_chatbot(model, messages):
     return chat_completion.choices[0].message.content
 
 
-def chatbot_generate(user_newest_input, history, model, current_answer, initial_txt):
+def chatbot_generate(user_newest_input, history, model, initial_txt):
     """
     Generate the next response from the chatbot
     :param user_newest_input: The newest input from the user
@@ -47,8 +46,6 @@ 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"
@@ -58,18 +55,16 @@ def chatbot_generate(user_newest_input, history, model, current_answer, initial_
     history.append(f"User: {user_newest_input.strip()}")
 
     # construct chat messages
-    print(initial_txt)
     chat_messages = [{"role": "system", "content": initial_txt}]
 
     # chat_messages = [{"role": "system", "content": initial_txt}]
-    current_txt = "My current answer to the instructions is as follows: " + current_answer + '. Now, assist me with the following: '
+    # 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": content
+                    "content": hist[5:].strip()
                 }
             )
         elif hist.startswith("Writing Assistant:"):
@@ -85,8 +80,8 @@ 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('generating chatbot')
+        # print(actual_model)
         print(chat_messages)
         ai_newest_output = query_chatbot(actual_model, chat_messages)
     else:

utils.py

@@ -326,7 +326,7 @@ def generate_assisted_question(question_prompt, next_q_btn, q_num, question_answ
 
         conversations_list = []
         with gr.Column() as chatbot_col:
-            chatbot = gr.Chatbot(conversations_list, height=300, label="Writing Helper")
+            chatbot = gr.Chatbot(conversations_list, height=350, label="Writing Helper")
         # Chat state
         state = gr.State(conversations_list)
         initial_usr_msg_state = gr.State(initial_user_message)
@@ -338,7 +338,7 @@ def generate_assisted_question(question_prompt, next_q_btn, q_num, question_answ
                     txt = gr.Textbox(
                         value="",
                         show_label=False,
-                        placeholder="Enter text and press the Interact button",
+                        placeholder="Enter text and press the Interact button. Your current answer will not be sent to the assistant. If you want the assistant to know your current answer, paste it into the chat window.",
                         lines=2,
                         container=False,
                         scale=4)
@@ -357,7 +357,7 @@ def generate_assisted_question(question_prompt, next_q_btn, q_num, question_answ
             autofocus=True,
             label="Write your response here:")
 
-        submit_button.click(chatbot_generate, [txt, state, model_state, tab_text2, initial_usr_msg_state], [chatbot, state, txt, submit_button])
+        submit_button.click(chatbot_generate, [txt, state, model_state, initial_usr_msg_state], [chatbot, state, txt, submit_button])
 
         text_button2 = gr.Button("Submit Response", variant="primary", interactive=False)
         tab_text2.input(count_words, tab_text2, [text_button2, word_count], show_progress="hidden")