Update app.py

app.py

@@ -12,6 +12,7 @@ hf_client = InferenceClient("CohereForAI/c4ai-command-r-plus", token=os.getenv("
 # 이미지 생성 API 클라이언트 설정
 client = Client("http://211.233.58.202:7960/")
 
+
 def respond(
     message,
     history: list[tuple[str, str]],
@@ -24,12 +25,10 @@ def respond(
     full_system_message = f"{system_prefix}{system_message}"
 
     messages = [{"role": "system", "content": f"{system_prefix} {system_message}"}]
-    # 이전 대화 내역을 추가합니다.
     for user_msg, assistant_msg in history:
         messages.append({"role": "user", "content": user_msg})
         if assistant_msg:
             messages.append({"role": "assistant", "content": assistant_msg})
-    # 현재 사용자 메시지를 추가합니다.
     messages.append({"role": "user", "content": message})
 
     # 이미지 생성 요청을 API에 전송하고 결과를 처리합니다.
@@ -44,14 +43,6 @@ def respond(
             num_inference_steps=28,
             api_name="/infer_t2i"
         )
-        # 결과 이미지 URL을 반환합니다.
-        if 'url' in result:
-            return result['url']
-        else:
-            return "이미지 생성에 실패했습니다."
-
-    try:
-        result = client.predict(...)
         if 'url' in result:
             return result['url']
         else:
@@ -60,6 +51,8 @@ def respond(
     except Exception as e:
         logging.error("API 요청 중 오류 발생: %s", str(e))
         return f"오류 발생: {str(e)}"
+    finally:
+        logging.info("요청 처리 완료")
         
 theme = "Nymbo/Nymbo_Theme"
 css = """