Update app.py

app.py

@@ -292,17 +292,17 @@ def display_python_parts():
             if st.button(f"Show Example for {part}", key=f"example_{part}"):
                 example = "Python script example with mock example inputs for " + part
                 example_placeholders[part] = example
-                st.code(example_placeholders[part], language="python")
                 response = chat_with_model('Create detailed advanced python script code examples for:' + example_placeholders[part], part)
+                st.code(response, language="python")
             if st.button(f"Take Quiz on {part}", key=f"quiz_{part}"):
                 quiz = "Python script quiz example with mock example inputs for " + part
                 response = chat_with_model(quiz, part)
+                st.code(response, language="python")
             prompt = f"Learn about advanced coding examples using mock example inputs for {content['details']}"
             if st.button(f"Explore {part}", key=part):
                 response = chat_with_model(prompt, part)
                 response_placeholders[part] = response
-            if part in response_placeholders:
-                st.markdown(f"**Response:** {response_placeholders[part]}")
+                st.code(response, language="python")
 
 def add_paper_buttons_and_links():
     page = st.sidebar.radio("Choose a page:", ["Python Pair Programmer"])