Update app.py

app.py

@@ -1,18 +1,18 @@
 import gradio as gr
 from huggingface_hub import InferenceClient
 import os
-from gradio_client import Client  # 이미지 생성 API 클라이언트
 import logging
+from gradio_client import Client  # 이미지 생성 API 클라이언트
 
 # 로깅 설정
 logging.basicConfig(level=logging.INFO)
+
 # 환경 변수에서 Hugging Face API 토큰을 가져옵니다.
 hf_client = InferenceClient("CohereForAI/c4ai-command-r-plus", token=os.getenv("HF_TOKEN"))
 
 # 이미지 생성 API 클라이언트 설정
 client = Client("http://211.233.58.202:7960/")
 
-
 def respond(message, history, system_message, max_tokens, temperature, top_p):
     # 초기 설정 및 변수 정의
     system_prefix = "System: 입력어의 언어에 따라 동일한 언어로 답변하라."
@@ -27,18 +27,25 @@ def respond(message, history, system_message, max_tokens, temperature, top_p):
 
     # 이미지 생성 요청
     try:
-        result = client.predict(...)
+        result = client.predict(
+            prompt=message,
+            seed=123,
+            randomize_seed=False,
+            width=1024,
+            height=576,
+            guidance_scale=5,
+            num_inference_steps=28,
+            api_name="/infer_t2i"
+        )
         if 'url' in result:
             return result['url']
         else:
-            return "이미지 생성에 실패했습니다."
+            logging.error("Image generation failed with error: %s", result.get('error', 'Unknown error'))
+            return "Failed to generate image."
     except Exception as e:
-        return f"오류 발생: {str(e)}"
-    finally:
-        # 필요한 경우 리소스 정리
-        pass
+        logging.error("Error during API request: %s", str(e))
+        return f"An error occurred: {str(e)}"
 
-        
 theme = "Nymbo/Nymbo_Theme"
 css = """
 footer {