Update app.py

app.py

@@ -835,7 +835,7 @@ def deploy_to_vercel(code: str):
 
 
 # 프롬프트 증강 함수 수정
-async def boost_prompt(prompt: str) -> str:
+def boost_prompt(prompt: str) -> str:
     if not prompt:
         return ""
     
@@ -857,16 +857,13 @@ async def boost_prompt(prompt: str) -> str:
     try:
         # Claude API 시도
         try:
-            messages = [{
-                "role": "user",
-                "content": f"다음 프롬프트를 분석하고 증강하시오: {prompt}"
-            }]
-            
-            response = await claude_client.messages.create(
+            response = claude_client.messages.create(
                 model="claude-3-5-sonnet-20241022",
                 max_tokens=2000,
-                system=boost_system_prompt,
-                messages=messages
+                messages=[{
+                    "role": "user",
+                    "content": f"다음 프롬프트를 분석하고 증강하시오: {prompt}"
+                }]
             )
             
             if hasattr(response, 'content') and len(response.content) > 0:
@@ -877,7 +874,7 @@ async def boost_prompt(prompt: str) -> str:
             print(f"Claude API 에러, OpenAI로 전환: {str(claude_error)}")
             
             # OpenAI API 시도
-            completion = await openai_client.chat.completions.create(
+            completion = openai_client.chat.completions.create(
                 model="gpt-4",
                 messages=[
                     {"role": "system", "content": boost_system_prompt},
@@ -895,17 +892,15 @@ async def boost_prompt(prompt: str) -> str:
         print(f"프롬프트 증강 중 오류 발생: {str(e)}")
         return prompt  # 오류 발생시 원본 프롬프트 반환
 
-# Boost 버튼 이벤트 핸들러 수정
+# Boost 버튼 이벤트 핸들러
 def handle_boost(prompt: str):
     try:
-        # asyncio.run()을 사용하여 비동기 함수 실행
-        boosted_prompt = asyncio.run(boost_prompt(prompt))
+        boosted_prompt = 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()