openAi.py

import streamlit as st
import os
from openai import OpenAI
os.environ['OPENAI_API_KEY'] = st.secrets['api_key']

st.title("이미지 생성기입니다.")

# 입력받는 곳
with st.form("form"): 
    user_input = st.text_input("그리고싶은 그림에 대해 묘사해주세요.")
    size = st.selectbox("size",["1024x1024", '512x512','256x256'])
    submit = st.form_submit_button("Submit")

if submit and user_input:
    gpt_prompt = [
        {
        'role': "system",
        'content': "Based on the user's input, describe the detailed appearance of the image in about 15 words."
    }
    ]
    gpt_prompt.append(
        {"role": "user", 
        "content": user_input
        }
    )

    client = OpenAI()
    with st.spinner("waiting for chatgpt ..."):
        gpt_response = client.chat.completions.create(
            model = 'gpt-3.5-turbo',
            messages = gpt_prompt,
        )
        
    dalle_prompt = gpt_response.choices[0].message.content
    
    st.write("dall-e prompt:",dalle_prompt)
    
    with st.spinner("waiting for dalle ..."):
        dalle_response = client.images.generate(
            model = 'dall-e-2',
            prompt = dalle_prompt,
            size = size
        )

        st.image(dalle_response.data[0].url)