Update app.py

app.py

@@ -857,22 +857,27 @@ async def boost_prompt(prompt: str) -> str:
     try:
         # Claude API 시도
         try:
+            messages = [{
+                "role": "user",
+                "content": f"다음 프롬프트를 분석하고 증강하시오: {prompt}"
+            }]
+            
             response = await claude_client.messages.create(
                 model="claude-3-5-sonnet-20241022",
                 max_tokens=2000,
                 system=boost_system_prompt,
-                messages=[{
-                    "role": "user",
-                    "content": f"다음 프롬프트를 분석하고 증강하시오: {prompt}"
-                }]
+                messages=messages
             )
-            return response.content[0].text
+            
+            if hasattr(response, 'content') and len(response.content) > 0:
+                return response.content[0].text
+            raise Exception("Claude API 응답 형식 오류")
             
         except Exception as claude_error:
             print(f"Claude API 에러, OpenAI로 전환: {str(claude_error)}")
             
             # OpenAI API 시도
-            response = await openai_client.chat.completions.create(
+            completion = await openai_client.chat.completions.create(
                 model="gpt-4",
                 messages=[
                     {"role": "system", "content": boost_system_prompt},
@@ -881,23 +886,26 @@ async def boost_prompt(prompt: str) -> str:
                 max_tokens=2000,
                 temperature=0.7
             )
-            return response.choices[0].message.content
+            
+            if completion.choices and len(completion.choices) > 0:
+                return completion.choices[0].message.content
+            raise Exception("OpenAI API 응답 형식 오류")
             
     except Exception as e:
         print(f"프롬프트 증강 중 오류 발생: {str(e)}")
         return prompt  # 오류 발생시 원본 프롬프트 반환
 
 # Boost 버튼 이벤트 핸들러 수정
-async def handle_boost(prompt: str):
+def handle_boost(prompt: str):
     try:
-        boosted_prompt = await boost_prompt(prompt)
+        # asyncio.run()을 사용하여 비동기 함수 실행
+        boosted_prompt = asyncio.run(boost_prompt(prompt))
         return boosted_prompt, gr.update(active_key="empty")
     except Exception as e:
         print(f"Boost 처리 중 오류: {str(e)}")
         return prompt, gr.update(active_key="empty")
 
 
-
         
 # Demo 인스턴스 생성
 demo_instance = Demo()
@@ -1094,7 +1102,9 @@ with gr.Blocks(css_paths="app.css",theme=theme) as demo:
                 fn=handle_boost,
                 inputs=[input],
                 outputs=[input, state_tab]
-            ) 
+            )
+
+
 
 # 이벤트 핸들러 수정
             deploy_btn.click(