import streamlit as st
from PIL import Image

# Page Title
st.title("Talk with the Great Physicist")

# Load Images
image1 = Image.open("images/image01.png")  # Einstein
image2 = Image.open("images/image02.png")  # Dirac
image3 = Image.open("images/image03.png")  # Feynman
image4 = Image.open("images/image04.png")  # Maxwell
image5 = Image.open("images/image05.png")  # Newton
image6 = Image.open("images/image06.png")  # Chat with All

# Define image links and names
image_links = [
    ("Einstein", "https://chatgpt.com/g/g-ocQnGKXGx-einstein-learn-relativity", image1),
    ("Dirac", "https://chatgpt.com/g/g-JjKhIfBft-dirac-learn-quantum-mechanics", image2),
    ("Feynman", "https://chatgpt.com/g/g-stfxVmJcu-feynman-learn-physics", image3),
    ("Maxwell", "https://chatgpt.com/g/g-cxxsVKyoj-maxwell-learn-electromagnetism", image4),
    ("Newton", "https://chatgpt.com/g/g-yIo2Tvujk-newton-learn-classical-mechanics", image5),
    ("Chat with All", "https://chatgpt.com/g/g-5jFCsQ6x4-dialog-with-the-great-physicists", image6),
]

# Create three columns for the images
col1, col2, col3 = st.columns(3)

# First column (Einstein and Maxwell)
with col1:
    st.image(image1)
    st.markdown(f"[Einstein](https://chatgpt.com/g/g-ocQnGKXGx-einstein-learn-relativity)")
    
    st.image(image4)
    st.markdown(f"[Maxwell](https://chatgpt.com/g/g-cxxsVKyoj-maxwell-learn-electromagnetism)")

# Second column (Dirac and Newton)
with col2:
    st.image(image2)
    st.markdown(f"[Dirac](https://chatgpt.com/g/g-JjKhIfBft-dirac-learn-quantum-mechanics)")
    
    st.image(image5)
    st.markdown(f"[Newton](https://chatgpt.com/g/g-yIo2Tvujk-newton-learn-classical-mechanics)")

# Third column (Feynman and Chat with All)
with col3:
    st.image(image3)
    st.markdown(f"[Feynman](https://chatgpt.com/g/g-stfxVmJcu-feynman-learn-physics)")
    
    st.image(image6)
    st.markdown(f"[Chat with All](https://chatgpt.com/g/g-5jFCsQ6x4-dialog-with-the-great-physicists)")