app.py

# gradio 라이브러리를 gr 별칭으로 임포트합니다.
import gradio as gr
from gradio_client import Client
import httpx

# Client 객체 생성 시 timeout 매개변수 제거
client = Client("seawolf2357/sajuice")




# 타임아웃 값을 초 단위로 설정합니다. 예: 읽기 타임아웃을 30초로 설정
timeout_config = httpx.Timeout(5.0, read=30.0)

try:
    # 수정된 타임아웃 설정을 적용하여 클라이언트를 생성합니다.
    client = Client("arxivgpt/sajutest", timeout=timeout_config)
    print("클라이언트가 성공적으로 생성되었습니다.")
except httpx.ReadTimeout:
    print("요청 시간 초과: 서버가 지정된 시간 내에 응답하지 않았습니다.")
except Exception as e:
    print(f"예상치 못한 오류가 발생했습니다: {e}")


def call_api(생년, 생월, 생일, 생시):
    # API를 호출하고 응답값을 받습니다.
    result = client.predict(
        생년,    # '생년(예: 1990)'에 대한 입력값
        생월,    # '생월(예: 01)'에 대한 입력값
        생일,    # '생일(예: 31)'에 대한 입력값
        생시,    # '생시(예시: 1030)'에 대한 입력값
        api_name="/get_lunar_info"
    )
    return result

# Gradio 인터페이스를 정의합니다.
inputs = [
    gr.Textbox(label="생년(예: 1990)"),
    gr.Textbox(label="생월(예: 01)"),
    gr.Textbox(label="생일(예: 31)"),
    gr.Textbox(label="생시(예시: 1030)")
]
outputs = gr.Textbox(label="결과")

# 인터페이스를 생성하고 실행합니다.
gr.Interface(fn=call_api, inputs=inputs, outputs=outputs, title="Get Lunar Info").launch()