Update app.py

app.py

@@ -245,11 +245,12 @@ def display_python_parts():
                 example = generate_example(part)
                 example_placeholders[part] = example
                 st.code(example_placeholders[part], language="python")
+                response = chat_with_model('Create a STEM related 3 to 5 line python code example with output for:' + example_placeholders[part], part)
 
             # Quizzes
             if st.button(f"Take Quiz on {part}", key=f"quiz_{part}"):
                 quiz = generate_quiz(part)
-                st.write(quiz)
+                response = chat_with_model(quiz, part)
 
             # Chat responses
             prompt = f"Learn about {python_parts_extended[part]}"