Update app.py

app.py

@@ -2,7 +2,10 @@ import gradio as gr
 from huggingface_hub import InferenceClient
 import os
 from gradio_client import Client  # 이미지 생성 API 클라이언트
+import logging
 
+# 로깅 설정
+logging.basicConfig(level=logging.INFO)
 # 환경 변수에서 Hugging Face API 토큰을 가져옵니다.
 hf_client = InferenceClient("CohereForAI/c4ai-command-r-plus", token=os.getenv("HF_TOKEN"))
 
@@ -46,9 +49,18 @@ def respond(
             return result['url']
         else:
             return "이미지 생성에 실패했습니다."
+
+    try:
+        result = client.predict(...)
+        if 'url' in result:
+            return result['url']
+        else:
+            logging.error("이미지 생성 실패: %s", result.get('error', '알 수 없는 오류'))
+            return "이미지 생성에 실패했습니다."
     except Exception as e:
+        logging.error("API 요청 중 오류 발생: %s", str(e))
         return f"오류 발생: {str(e)}"
-
+        
 theme = "Nymbo/Nymbo_Theme"
 css = """
 footer {