Update app.py

app.py

@@ -755,17 +755,16 @@ def generate_space_name():
     letters = string.ascii_lowercase
     return ''.join(random.choice(letters) for i in range(6))
 
-
 def deploy_to_vercel(code: str):
-    """Vercel에 코드 배포"""
     try:
         token = "A8IFZmgW2cqA4yUNlLPnci0N"
         if not token:
             return "Vercel 토큰이 설정되지 않았습니다."
 
-        # 프로젝트 이름 생성 
+        # 6자리 영문 프로젝트 이름 생성
         project_name = ''.join(random.choice(string.ascii_lowercase) for i in range(6))
 
+
         # Vercel API 엔드포인트
         deploy_url = "https://api.vercel.com/v13/deployments"
 
@@ -818,23 +817,87 @@ def deploy_to_vercel(code: str):
             "projectSettings": project_settings
         }
 
+
         deploy_response = requests.post(deploy_url, headers=headers, json=deploy_data)
         
         if deploy_response.status_code != 200:
             return f"배포 실패: {deploy_response.text}"
 
-        # 응답에서 실제 URL 가져오기
-        response_data = deploy_response.json()
-        deployment_url = response_data.get('url', f"{project_name}.vercel.app")
+        # URL 형식 수정 - 6자리.vercel.app 형태로 반환
+        deployment_url = f"{project_name}.vercel.app"
         
-        # 배포 상태 확인을 위한 대기
-        time.sleep(5)  # 배포가 완료될 때까지 잠시 대기
+        time.sleep(5)
 
-        return f"""배포 완료! <a href="https://{deployment_url}" target="_blank" style="color: #1890ff; text-decoration: underline; cursor: pointer;">여기를 클릭하여 열기</a>"""
+        return f"""배포 완료! <a href="https://{deployment_url}" target="_blank" style="color: #1890ff; text-decoration: underline; cursor: pointer;">https://{deployment_url}</a>"""
 
     except Exception as e:
         return f"배포 중 오류 발생: {str(e)}"
 
+
+# 프롬프트 증강 함수 수정
+async def boost_prompt(prompt: str) -> str:
+    if not prompt:
+        return ""
+    
+    # 증강을 위한 시스템 프롬프트
+    boost_system_prompt = """
+    당신은 웹 개발 프롬프트 전문가입니다. 
+    주어진 프롬프트를 분석하여 더 상세하고 전문적인 요구사항으로 확장하되,
+    원래 의도와 목적은 그대로 유지하면서 다음 관점들을 고려하여 증강하십시오:
+
+    1. 기술적 구현 상세
+    2. UI/UX 디자인 요소
+    3. 사용자 경험 최적화
+    4. 성능과 보안
+    5. 접근성과 호환성
+    
+    기존 SystemPrompt의 모든 규칙을 준수하면서 증강된 프롬프트를 생성하십시오.
+    """
+    
+    try:
+        # Claude API 시도
+        try:
+            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}"
+                }]
+            )
+            return response.content[0].text
+            
+        except Exception as claude_error:
+            print(f"Claude API 에러, OpenAI로 전환: {str(claude_error)}")
+            
+            # OpenAI API 시도
+            response = await openai_client.chat.completions.create(
+                model="gpt-4",
+                messages=[
+                    {"role": "system", "content": boost_system_prompt},
+                    {"role": "user", "content": f"다음 프롬프트를 분석하고 증강하시오: {prompt}"}
+                ],
+                max_tokens=2000,
+                temperature=0.7
+            )
+            return response.choices[0].message.content
+            
+    except Exception as e:
+        print(f"프롬프트 증강 중 오류 발생: {str(e)}")
+        return prompt  # 오류 발생시 원본 프롬프트 반환
+
+# Boost 버튼 이벤트 핸들러 수정
+async def handle_boost(prompt: str):
+    try:
+        boosted_prompt = await 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()
@@ -899,15 +962,17 @@ with gr.Blocks(css_paths="app.css",theme=theme) as demo:
                             allow_clear=True, 
                             placeholder=random.choice(DEMO_LIST)['description']
                         )
-
-
-# UI 수정 부분 - antd.Col(span=24, md=8) 내부의 버튼 컨테이너에 배포 버튼 추가:
+        
                         with antd.Flex(gap="small", justify="space-between"):
                             btn = antd.Button("Send", type="primary", size="large")
+                            boost_btn = antd.Button("Boost", type="default", size="large")  # Boost 버튼 추가
                             execute_btn = antd.Button("Code 실행", type="default", size="large")
-                            deploy_btn = antd.Button("배포하기", type="default", size="large")  # 추가
+                            deploy_btn = antd.Button("배포하기", type="default", size="large")
                             clear_btn = antd.Button("Clear", type="default", size="large")
 
+                        deploy_result = gr.HTML(label="배포 결과")
+
+
 # 배포 결과를 표시할 텍스트 영역 추가
  
                         deploy_result = gr.HTML(label="배포 결과")
@@ -1024,7 +1089,12 @@ with gr.Blocks(css_paths="app.css",theme=theme) as demo:
                 outputs=[history]
             )
 
-
+# UI의 Boost 버튼 이벤트 연결 수정
+            boost_btn.click(
+                fn=handle_boost,
+                inputs=[input],
+                outputs=[input, state_tab]
+            ) 
 
 # 이벤트 핸들러 수정
             deploy_btn.click(