GPT + DALL-E

pip install streamlit openai

# https://docs.streamlit.io/library/api-reference
import streamlit as st
import openai

openai.api_key = "발급 받은 키 입력"

st.title("ChatGPT Plus DALL-E")

with st.form("form"):
    user_input = st.text_input("Prompt")
    submit = st.form_submit_button("Submit")

if submit:
    st.write(user_input)

    # ChatGPT
    gpt_prompt = []

    gpt_prompt.append({
        "role": "system",
        "content": "Imagine the detail appearance of the input. Response shortly. Translate in English"
    })

    gpt_prompt.append({
        "role": "user",
        "content": user_input
    })

    with st.spinner("Wating for ChatGPT..."):
        prompt = openai.ChatCompletion.create(
            model="gpt-3.5-turbo",
            messages=gpt_prompt)

    prompt = prompt["choices"][0]["message"]["content"]
    st.caption(prompt)

    # DALL-E
    with st.spinner("Wating for DALL-E..."):
        result = openai.Image.create(
            prompt=prompt,
            size="1024x1024"
        )

    st.image(result["data"][0]["url"])