import gradio as gr
import openai
import os

# OpenAI API 키 설정
openai.api_key = os.getenv("OPENAI_API_KEY")

def generate_keywords_and_search_images(input_text):
    # 입력 텍스트를 줄별로 분리
    lines = input_text.strip().split('\n')
    all_image_urls = []

    for line in lines:
        # 각 줄에 대한 키워드 생성
        keyword = generate_keyword_from_text(line)
        if keyword.startswith("키워드 생성 중 에러 발생"):
            print(keyword)
            continue  # 키워드 생성 실패 시, 다음 줄로 넘어감

        # 생성된 키워드로 Pexels 검색
        image_urls = search_pexels(keyword)
        all_image_urls.extend(image_urls)  # 검색된 이미지 URL들을 전체 목록에 추가

    return all_image_urls

# Gradio 인터페이스 설정과 실행
iface = gr.Interface(
    fn=generate_keywords_and_search_images,
    inputs=gr.Textbox(lines=5, placeholder="여기에 Pexels 검색을 위한 텍스트를 입력하세요. 각 줄마다 별도의 검색 키워드가 생성됩니다."),
    outputs=gr.Gallery(),
    title="GPT 및 Pexels API를 이용한 다중 이미지 검색",
    description="제공된 텍스트의 각 줄을 바탕으로 Pexels 검색에 사용할 영문 키워드를 자동 생성하고, 해당 키워드로 Pexels에서 이미지를 검색합니다."
)

iface.launch()