app.py

import gradio as gr
import openai

# OpenAI API 키와 Assistant ID 설정
OPENAI_API_KEY = "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"
ASSISTANT_ID = "asst_GLnbQ7HdAoRtQL3IKiODpRFB"

# OpenAI API 키 설정
openai.api_key = OPENAI_API_KEY


def ask_openai(prompt):
    response = openai.ChatCompletion.create(
        model="gpt-4-turbo-preview",  # 또는 사용 가능한 다른 모델
        messages=[
            {"role": "system", "content": "You are a helpful assistant."},
            {"role": "user", "content": prompt}
        ]
    )
    return response['choices'][0]['message']['content']

# Gradio UI 구성
interface = gr.Interface(
    fn=ask_openai,
    inputs=gr.Textbox(lines=2, placeholder="여기에 질문을 입력하세요..."),
    outputs="text",
    title="OpenAI GPT Chat",
    description="OpenAI GPT 모델을 사용하여 답변을 제공하는 챗봇입니다."
)

# Gradio 앱 실행, 공개 링크 생성을 원할 경우 share=True로 설정
interface.launch(share=True)