Update app.py

app.py

@@ -163,13 +163,20 @@ homunculus_parts = {
     "Left Shin": "🦵", "Right Shin": "🦵"
 }
 
-# Function to display the homunculus parts with expanders
+
+# Function to display the homunculus parts with expanders and chat buttons
 def display_homunculus_parts():
     st.title("Homunculus Model")
 
     # Container parts
     with st.expander("Head (🧠)", expanded=False):
         st.write("Contains: Left Eye 👁️, Right Eye 👁️, Left Eyebrow 🤨, Right Eyebrow 🤨, Nose 👃, Mouth 👄")
+        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)
 
     # Symmetric body parts (Left & Right)
     for part in ["Shoulder", "Upper Arm", "Elbow", "Forearm", "Wrist", "Hand", "Hip", "Thigh", "Knee", "Shin"]:
@@ -177,51 +184,27 @@ def display_homunculus_parts():
         with col1:
             with st.expander(f"Left {part} (💪)", expanded=False):
                 st.write(f"Details about the Left {part}")
+                prompt = f"Learn about the key features of Left {part}"
+                if st.button(f"Explore Left {part}", key=f"Left {part}"):
+                    response = chat_with_model(prompt, f"Left {part}")
+                    st.write(response)
         with col2:
             with st.expander(f"Right {part} (💪)", expanded=False):
                 st.write(f"Details about the Right {part}")
+                prompt = f"Learn about the key features of Right {part}"
+                if st.button(f"Explore Right {part}", key=f"Right {part}"):
+                    response = chat_with_model(prompt, f"Right {part}")
+                    st.write(response)
 
     # Central body parts
-    for part in ["Neck", "Chest", "Abdomen", "Pelvis"]:
-        with st.expander(f"{part} ({homunculus_parts[part]})", expanded=False):
+    central_parts = ["Neck", "Chest", "Abdomen", "Pelvis"]
+    for part in central_parts:
+        with st.expander(f"{part} ({homunculus_parts2[part][0]})", expanded=False):
             st.write(f"Details about the {part}")
-
-
-# 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)": "🧣",
-    "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")
-
-    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}")
+            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)
 
 # Define function to add paper buttons and links
 def add_paper_buttons_and_links():
@@ -231,7 +214,7 @@ def add_paper_buttons_and_links():
     if page == "Detailed Homunculus Model":
         display_homunculus_parts()
     elif page == "Homunculus Emoji Representation":
-        display_homunculus_parts2()
+        display_homunculus_parts()
 
     col1, col2, col3, col4 = st.columns(4)