app5.py

from dotenv import load_dotenv
import os
from openai import OpenAI
import streamlit as st

load_dotenv()

client = OpenAI(
    api_key=os.environ.get("OPENAI_API_KEY"),
)

st.title('홍보 포스터 만들기 🚀')

keyword = st.text_input("키워드를 입력하세요")

if st.button('생성하기'):
  with st.spinner('생성중...'):
      chat_completion = client.chat.completions.create(
        messages=[
            {
                "role": "user",
                "content": keyword,
            },
            {
                "role": "system",
                "content": "입력 받은 키워드에 대한 150자 이내에 솔깃한 제품 홍보 문구를 작성해줘.",
            }
        ],
        model="gpt-4o",
        )
      response = client.images.generate(
        model="dall-e-3",
        prompt=f'{keyword}를 수채화풍으로 그려줘',
        quality="standard",
        size="1024x1024",
        n=1,
    )

  result = chat_completion.choices[0].message.content
  image_url = response.data[0].url
  st.write(result)
  st.image(image_url)