Update app.py

app.py

@@ -187,43 +187,42 @@ def display_homunculus_parts():
             st.write(f"Details about the {part}")
 
 
-# Homunculus parts 2 with emojis and corresponding healthcare prompts
+# Homunculus parts with emojis
 homunculus_parts2 = {
-    "Head (H)": "🧠", "Left Eye (LE)": "👁️", "Right Eye (RE)": "👁️",
-    "Left Eyebrow (LB)": "🤨", "Right Eyebrow (RB)": "🤨", "Nose (N)": "👃",
-    "Mouth (M)": "👄", "Neck (Ne)": "🧣", "Left Shoulder (LS)": "💪", 
-    "Right Shoulder (RS)": "💪", "Left Upper Arm (LUA)": "💪", 
-    "Right Upper Arm (RUA)": "💪", "Left Elbow (LEl)": "💪", "Right Elbow (REl)": "💪", 
-    "Left Forearm (LF)": "💪", "Right Forearm (RF)": "💪", "Left Wrist (LW)": "🖐️", 
-    "Right Wrist (RW)": "🖐️", "Left Hand (LH)": "🤚", "Right Hand (RH)": "🤚",
-    "Chest (C)": "🫁", "Abdomen (Ab)": "🧘", "Pelvis (P)": "🧍",
-    "Left Hip (LHip)": "🦵", "Right Hip (RHip)": "🦵", "Left Thigh (LT)": "🦵",
-    "Right Thigh (RT)": "🦵", "Left Knee (LK)": "🦵", "Right Knee (RK)": "🦵",
-    "Left Shin (LSh)": "🦵", "Right Shin (RSh)": "🦵"
+    "Head (H)": ["🧠", ["Left Eye (LE)", "Right Eye (RE)", "Left Eyebrow (LB)", "Right Eyebrow (RB)", "Nose (N)", "Mouth (M)"]],
+    "Neck (Ne)": "🧣",
+    "Shoulders": [["Left Shoulder (LS)", "Right Shoulder (RS)"], "💪"],
+    "Upper Arms": [["Left Upper Arm (LUA)", "Right Upper Arm (RUA)"], "💪"],
+    "Elbows": [["Left Elbow (LEl)", "Right Elbow (REl)"], "💪"],
+    "Forearms": [["Left Forearm (LF)", "Right Forearm (RF)"], "💪"],
+    "Wrists": [["Left Wrist (LW)", "Right Wrist (RW)"], "🖐️"],
+    "Hands": [["Left Hand (LH)", "Right Hand (RH)"], "🤚"],
+    "Chest (C)": "🫁",
+    "Abdomen (Ab)": "🧘",
+    "Pelvis (P)": "🧍",
+    "Hips": [["Left Hip (LHip)", "Right Hip (RHip)"], "🦵"],
+    "Thighs": [["Left Thigh (LT)", "Right Thigh (RT)"], "🦵"],
+    "Knees": [["Left Knee (LK)", "Right Knee (RK)"], "🦵"],
+    "Shins": [["Left Shin (LSh)", "Right Shin (RSh)"], "🦵"]
 }
 
 def display_homunculus_parts2():
     st.title("Homunculus Model")
 
-    # Display container parts with sub-parts
-    container_parts = ["Head (H)", "Chest (C)", "Abdomen (Ab)", "Pelvis (P)"]
-    for part in container_parts:
-        with st.expander(f"{homunculus_parts2[part]} {part}", expanded=False):
-            col1, col2 = st.columns(2)
-            for subpart in homunculus_parts2.keys():
-                if subpart.startswith(part.split(" ")[0]):
-                    col = col1 if "Left" in subpart or part in container_parts else col2
-                    col.write(f"{homunculus_parts2[subpart]} {subpart}")
-
-    # Display other parts
-    col1, col2 = st.columns(2)
-    for part in homunculus_parts2:
-        if part not in container_parts:
-            col = col1 if "Left" in part else col2
-            col.button(f"{homunculus_parts2[part]} {part}", key=part)
-
+    for part, content in homunculus_parts2.items():
+        if isinstance(content, list):
+            emoji, sub_parts = content[0], content[1]
+            with st.expander(f"{emoji} {part}", expanded=False):
+                if isinstance(sub_parts, list):
+                    col1, col2 = st.columns(2)
+                    for subpart in sub_parts:
+                        col = col1 if "Left" in subpart else col2
+                        col.write(f"{homunculus_parts2.get(subpart, '❓')} {subpart}")
+                else:
+                    st.write(f"{emoji} {sub_parts}")
+        else:
+            st.write(f"{content} {part}")
 
-    
 # Define function to add paper buttons and links
 def add_paper_buttons_and_links():