app.py

from fastapi import FastAPI, Request
from fastapi.responses import JSONResponse
import gradio as gr

app = FastAPI()

# 실제 처리 함수
def greet(name: str) -> str:
    return f"Hello {name}!!"

# FastAPI 엔드포인트
@app.post("/predict")
async def predict(request: Request):
    try:
        data = await request.json()
        name = data["data"][0]
        result = greet(name)
        return JSONResponse(content={"data": [result]})
    except Exception as e:
        return JSONResponse(content={"error": str(e)}, status_code=400)

# Gradio 인터페이스 (선택적)
demo = gr.Interface(fn=greet, inputs="text", outputs="text")

# Space 실행 시 Gradio UI와 FastAPI 동시에 작동
demo.launch(server_name="0.0.0.0", server_port=7860, inline=False)