Update app.py

app.py

@@ -1,24 +1,26 @@
 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}!!"
 
-gradio_ui = gr.Interface(fn=greet, inputs="text", outputs="text")
-
+# FastAPI 엔드포인트
 @app.post("/predict")
 async def predict(request: Request):
-    data = await request.json()
-    input_text = data["data"][0]
-    result = greet(input_text)
-    return {"data": [result]}
-
-@app.get("/")
-async def root():
-    return {"message": "FastAPI is working!"}
+    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)
 
-# Mount Gradio UI
-gradio_ui.launch(share=False, inline=True, inbrowser=False)
+# 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)