Update app.py

app.py

@@ -221,6 +221,14 @@ homunculus_parts = {
 # Function to display the homunculus parts with expanders and chat buttons
 def display_homunculus_parts():
     st.title("Homunculus Model")
+    
+    with st.expander(f"Head ({homunculus_parts['Head']})", expanded=False):
+        head_parts = ["Left Eye", "Right Eye", "Left Eyebrow", "Right Eyebrow", "Nose", "Mouth"]
+        for part in head_parts:
+            prompt = f"Learn about the key features of {part}"
+            if st.button(f"Explore {part}", key=part):
+                response = chat_with_model(prompt, part)
+                st.write(response)
 
     with st.expander(f"Brain ({homunculus_parts['Brain']})", expanded=False):
         brain_parts = {
@@ -243,14 +251,6 @@ def display_homunculus_parts():
                 response = chat_with_model(prompt, part)
                 st.write(response)
 
-            
-        head_parts = ["Left Eye", "Right Eye", "Left Eyebrow", "Right Eyebrow", "Nose", "Mouth"]
-        for part in head_parts:
-            prompt = f"Learn about the key features of {part}"
-            if st.button(f"Explore {part}", key=part):
-                response = chat_with_model(prompt, part)
-                st.write(response)
-
     # Displaying symmetric body parts
     symmetric_parts = ["Shoulder", "Upper Arm", "Elbow", "Forearm", "Wrist", "Hand", "Hip", "Thigh", "Knee", "Shin"]
     for part in symmetric_parts: