import streamlit as st


def setup():
    submit_button = st.button("Generate a response")
    # "Enter your OpenAPI key above (make sure there is money in it). Learn more [here](https://platform.openai.com/api-keys)",
    openai_models: str = st.selectbox(
        label="Choose your OpenAI model:",
        options=[
            "gpt-3.5-turbo",
            "gpt-4-turbo"
        ]
    )
    st.subheader("LinuxGPT")
    st.write(
        "LinuxGPT is currently trained on the Linux Fundamentals textbook, and course materials will be added soon."
    )
    st.subheader("Examples for Demoing")
    st.write("")
    return (
        submit_button,
        openai_models,
    )