app.py

import gradio as gr
import os
import sys
import json
import requests

# 환경 변수에서 필요한 설정을 로드합니다.
API_URL = "https://api.openai.com/v1/completions"
OPENAI_API_KEY = "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"  # 이 값은 예시이며, 실제 키값을 사용해야 합니다.
NUM_THREADS = 5

def exception_handler(exception_type, exception, traceback):
    # 예외 발생 시 로그를 출력합니다.
    print("%s: %s" % (exception_type.__name__, exception))

sys.excepthook = exception_handler
sys.tracebacklimit = 0

def predict(inputs):
    # OpenAI API에 요청을 보낼 페이로드를 구성합니다.
    payload = {
        "model": "gpt-4-turbo-preview",
        "prompt": inputs,
        "temperature": 0.7,
        "max_tokens": 150
    }

    headers = {
        "Content-Type": "application/json",
        "Authorization": f"Bearer {OPENAI_API_KEY}"
    }

    try:
        response = requests.post(API_URL, headers=headers, json=payload)
        if response.status_code == 200:
            data = response.json()
            return data["choices"][0]["text"].strip()
        elif response.status_code == 429:
            return "요청 한도 초과: 잠시 후 다시 시도해주세요."
        else:
            return f"오류가 발생했습니다. 상태 코드: {response.status_code}"
    except Exception as e:
        return f"요청 처리 중 예외가 발생했습니다: {str(e)}"

# Gradio 인터페이스 구성
with gr.Interface(fn=predict, inputs="text", outputs="text") as interface:
    interface.launch()