further optimization

app.py

@@ -77,6 +77,9 @@ def load_answer(question):
             passage_and_research = chat(st.session_state.sessionMessages)
             st.session_state.sessionMessages.append(AIMessage(content=passage_and_research.content))
 
+            # Print passage
+            st.write(passage_and_research.content)
+
             # Based on the above generated passage and research topic, we will now generate 3 FRQs and their grading rubrics
             frqPrompt='Generate 3 questions based on the previously discussed common core standard, and the above generated passage. The answers should be available either in the above passage, or the student should be able to find them with easy web research.  Good questions have: introduction, context, and open-ended question. Remember, the student is in grade: '+chatResponseDict['grade-level']
             frqPrompt+='Generate a rubric for evaluating the student responses for the above questions. Return the result in JSON format. E.g. { "rubric":[{"question":"..text of the question", "rubric":".. text of the rubric"}] etc.'
@@ -86,6 +89,9 @@ def load_answer(question):
             question_and_rubric = chat(st.session_state.sessionMessages)
             st.session_state.sessionMessages.append(AIMessage(content=question_and_rubric.content))
 
+            # Print rubric & questions
+            st.write(question_and_rubric.content)
+
             ## Do the QC Step
             qcPrompt='For the JSON generated in the previous step, please evaluate for each question and rubric:\
                     - Does the FRQ have an introduction, context, and an open-ended question\
@@ -103,9 +109,7 @@ def load_answer(question):
             st.session_state.sessionMessages.append(HumanMessage(content=qcPrompt))
             qcResponseJson = json.loads(chat(st.session_state.sessionMessages).content)
 
-            ## Write the passage & questions, and QC results
-            st.write(passage_and_research.content)
-            st.write(question_and_rubric.content)
+            ## Write QC results
             st.write(qcResponseJson)
 
             ## Qc Response doesn't need to into the session state