Update app.py

app.py

@@ -1,3 +1,15 @@
+# 개선 아이디어 (한글):
+# 1. 비즈니스 로직(예: API 호출, 데이터 처리 함수)과 UI 구성을 분리하여 코드 구조를 간결하게 유지하세요.
+# 2. 예외 처리 구간에서 구체적인 예외 타입(ValueError, TimeoutError 등)을 사용하고 logging 모듈을 활용해 디버깅과 유지보수를 개선하세요.
+# 3. 배포와 관련된 민감한 토큰이나 키는 절대로 코드에 직접 작성하지 말고 환경 변수나 안전한 설정 파일을 통해 관리하세요.
+# 4. 반복되는 코드(예: load_json_data, create_template_html 등)를 통합하거나 작은 함수로 나누어 재사용성을 높이세요.
+# 5. 대규모 함수를 작은 기능 단위로 분리해 유지보수성을 향상시키고, 함수마다 명확한 주석과 타입 힌트를 추가하세요.
+# 6. 외부 API 호출 시 asyncio와 aiohttp 등을 적용해 비동기 처리를 통해 더 나은 응답 시간을 지원하도록 개선할 수 있습니다.
+# 7. 환경설정(테마, CSS, HTML 템플릿 등)을 별도의 파일이나 상수로 관리하여 코드 본문 가독성을 높이세요.
+# 8. 사용자 입력에 대한 유효성 검사와 예외 처리를 강화하여, 불완전한 입력이 들어와도 프로그램이 안전하게 동작하도록 개선하세요.
+# 9. 주요 함수에 대한 단위 테스트 코드를 추가하여, 수정 시 기능이 정상 동작하는지 쉽게 검증할 수 있도록 하세요.
+# 10. Gradio 버전을 5.6.0으로 명시하여, 시스템 요구사항 및 호환성을 보장하세요.
+
 import os
 import re
 import random
@@ -20,8 +32,9 @@ import urllib.parse
 from huggingface_hub import HfApi, create_repo
 import string
 import random
+import requests
 
-
+# SystemPrompt 정의
 SystemPrompt = """너의 이름은 'MOUSE'이다. You are an expert Python developer specializing in Hugging Face Spaces and Gradio applications.
 Your task is to create functional and aesthetically pleasing web applications using Python, Gradio, and Hugging Face integration.
 
@@ -85,16 +98,17 @@ from config import DEMO_LIST
 
 class Role:
     SYSTEM = "system"
-    USER = "user"
+    USER = "user" 
     ASSISTANT = "assistant"
 
 History = List[Tuple[str, str]]
 Messages = List[Dict[str, str]]
 
-# 이미지 캐시를 메모리에 저장
+# 이미지 캐시
 IMAGE_CACHE = {}
 
 def get_image_base64(image_path):
+    # Improvement: Consider adding logging and handling specific exceptions
     if image_path in IMAGE_CACHE:
         return IMAGE_CACHE[image_path]
     try:
@@ -102,9 +116,10 @@ def get_image_base64(image_path):
             encoded_string = base64.b64encode(image_file.read()).decode()
             IMAGE_CACHE[image_path] = encoded_string
             return encoded_string
-    except:
+    except Exception as e:
+        # TODO: Log exception details
         return IMAGE_CACHE.get('default.png', '')
-        
+
 def history_to_messages(history: History, system: str) -> Messages:
     messages = [{'role': Role.SYSTEM, 'content': system}]
     for h in history:
@@ -126,52 +141,6 @@ YOUR_OPENAI_TOKEN = os.getenv('OPENAI_API_KEY')
 claude_client = anthropic.Anthropic(api_key=YOUR_ANTHROPIC_TOKEN)
 openai_client = openai.OpenAI(api_key=YOUR_OPENAI_TOKEN)
 
-async def try_claude_api(system_message, claude_messages, timeout=15):
-    try:
-        start_time = time.time()
-        with claude_client.messages.stream(
-            model="claude-3-5-sonnet-20241022",
-            max_tokens=7800,
-            system=system_message,
-            messages=claude_messages
-        ) as stream:
-            collected_content = ""
-            for chunk in stream:
-                current_time = time.time()
-                if current_time - start_time > timeout:
-                    print(f"Claude API response time: {current_time - start_time:.2f} seconds")
-                    raise TimeoutError("Claude API timeout")
-                if chunk.type == "content_block_delta":
-                    collected_content += chunk.delta.text
-                    yield collected_content
-                    await asyncio.sleep(0)
-                
-                start_time = current_time
-                
-    except Exception as e:
-        print(f"Claude API error: {str(e)}")
-        raise e
-
-async def try_openai_api(openai_messages):
-    try:
-        stream = openai_client.chat.completions.create(
-            model="gpt-4o",
-            messages=openai_messages,
-            stream=True,
-            max_tokens=4096,
-            temperature=0.7
-        )
-        
-        collected_content = ""
-        for chunk in stream:
-            if chunk.choices[0].delta.content is not None:
-                collected_content += chunk.choices[0].delta.content
-                yield collected_content
-                
-    except Exception as e:
-        print(f"OpenAI API error: {str(e)}")
-        raise e
-
 # Built-in modules that don't need to be in requirements.txt
 BUILTIN_MODULES = {
     'os', 'sys', 're', 'time', 'json', 'csv', 'math', 'random', 'datetime', 'calendar',
@@ -322,20 +291,12 @@ IMPORT_TO_PACKAGE = {
 
 def get_package_name(import_name):
     """임포트명으로부터 실제 패키지명을 반환"""
-    # 내장 모듈이면 None 반환
     if import_name in BUILTIN_MODULES:
         return None
-        
-    # 점이 있는 경우 첫 부분만 사용 (예: matplotlib.pyplot -> matplotlib)
     base_import = import_name.split('.')[0]
-    
-    # 내장 모듈이면 None 반환
     if base_import in BUILTIN_MODULES:
         return None
-        
     return IMPORT_TO_PACKAGE.get(base_import, base_import)
-    
-
 
 def analyze_code(code: str) -> str:
     """코드 분석 결과를 HTML 형식으로 반환"""
@@ -359,15 +320,13 @@ def analyze_code(code: str) -> str:
     for line in code.split('\n'):
         if line.startswith('import ') or line.startswith('from '):
             imports.append(line.strip())
-            # 패키지 이름 추출 및 변환
             if line.startswith('import '):
                 package = line.split('import ')[1].split()[0].split('.')[0]
             else:
                 package = line.split('from ')[1].split()[0].split('.')[0]
             
-            # 내장 모듈이 아닌 경우만 패키지 추가
             package_name = get_package_name(package)
-            if package_name:  # None이 아닌 경우만 추가
+            if package_name:
                 required_packages.add(package_name)
    
     if imports:
@@ -377,16 +336,12 @@ def analyze_code(code: str) -> str:
             analysis.append(f"<li><code>{imp}</code></li>")
         analysis.append("</ul>")
         
-        # requirements.txt 설명 추가
         analysis.append("<h3>📋 Requirements.txt</h3>")
         analysis.append("<p>이 앱을 실행하기 위해 필요한 패키지들입니다:</p>")
         analysis.append("<pre>")
         for pkg in sorted(required_packages):
-            if pkg and pkg not in BUILTIN_MODULES:  # None이 아니고 내장 모듈이 아닌 경우만 추가
-                if pkg == 'gradio':
-                    analysis.append("gradio==5.5.0")
-                else:
-                    analysis.append(pkg)
+            if pkg and pkg not in BUILTIN_MODULES:
+                analysis.append(pkg)
         analysis.append("</pre>")
     
     # 2. 함수 분석
@@ -409,12 +364,10 @@ def analyze_code(code: str) -> str:
                 current_func = []
     
     if functions:
-        analysis.append("<h2>🔧 주요 함수 설명</h2>")
+        analysis.append("<h2>🔧 주요 함수</h2>")
         for func in functions:
             func_name = func.split('def ')[1].split('(')[0]
             analysis.append(f"<h3><code>{func_name}</code></h3>")
-            analysis.append(f"<p>{get_function_description(func)}</p>")
-            # 함수 파라미터 분석
             params = func.split('(')[1].split(')')[0]
             if params.strip():
                 analysis.append("<p>파라미터:</p><ul>")
@@ -434,35 +387,14 @@ def analyze_code(code: str) -> str:
     
     if ui_components:
         analysis.append("<h2>🎨 UI 구성요소</h2>")
-        analysis.append("<p>이 앱은 다음과 같은 Gradio 컴포넌트들로 구성되어 있습니다:</p>")
         analysis.append("<ul>")
         for component in ui_components:
-            analysis.append(f"<li><strong>{component}</strong>: {get_component_description(component)}</li>")
+            analysis.append(f"<li><strong>{component}</strong></li>")
         analysis.append("</ul>")
     
-    # 4. 특징 및 기능
-    analysis.append("<h2>✨ 주요 특징</h2>")
-    analysis.append("<ul>")
-    if 'theme=' in code:
-        analysis.append("<li>커스텀 테마 적용으로 일관된 디자인</li>")
-    if 'with gr.Row' in code:
-        analysis.append("<li>반응형 레이아웃으로 다양한 화면 크기 지원</li>")
-    if 'gr.State' in code:
-        analysis.append("<li>상태 관리를 통한 데이터 유지</li>")
-    if '.click(' in code:
-        analysis.append("<li>이벤트 핸들링을 통한 동적 상호작용</li>")
-    analysis.append("</ul>")
-    
-    # 5. 실행 방법
-    analysis.append("<h2>▶️ 실행 방법</h2>")
-    analysis.append("<ol>")
-    analysis.append("<li>'실행하기' 버튼을 클릭하여 Hugging Face Space에 배포</li>")
-    analysis.append("<li>생성된 링크를 클릭하여 애플리케이션 실행</li>")
-    analysis.append("<li>필요한 입력값을 제공하고 상호작용 시작</li>")
-    analysis.append("</ol>")
-    
     return "\n".join(analysis)
 
+
 def get_function_description(func: str) -> str:
     """함수의 목적을 설명하는 문자열 반환"""
     if 'get_multiplication_table' in func:
@@ -499,6 +431,323 @@ def get_component_description(component: str) -> str:
     return descriptions.get(component, '컴포넌트 설명')
 
 
+async def try_claude_api(system_message, claude_messages, timeout=15):
+    try:
+        start_time = time.time()
+        with claude_client.messages.stream(
+            model="claude-3-5-sonnet-20241022",
+            max_tokens=7800,
+            system=system_message,
+            messages=claude_messages
+        ) as stream:
+            collected_content = ""
+            for chunk in stream:
+                current_time = time.time()
+                if current_time - start_time > timeout:
+                    print(f"Claude API response time: {current_time - start_time:.2f} seconds")
+                    raise TimeoutError("Claude API timeout")
+                if chunk.type == "content_block_delta":
+                    collected_content += chunk.delta.text
+                    yield collected_content
+                    await asyncio.sleep(0)
+                
+                start_time = current_time
+                
+    except Exception as e:
+        print(f"Claude API error: {str(e)}")
+        raise e
+
+async def try_openai_api(openai_messages):
+    try:
+        stream = openai_client.chat.completions.create(
+            model="gpt-4",
+            messages=openai_messages,
+            stream=True,
+            max_tokens=4096,
+            temperature=0.7
+        )
+        
+        collected_content = ""
+        for chunk in stream:
+            if chunk.choices[0].delta.content is not None:
+                collected_content += chunk.choices[0].delta.content
+                yield collected_content
+                
+    except Exception as e:
+        print(f"OpenAI API error: {str(e)}")
+        raise e
+
+def deploy_to_vercel(code: str):
+    try:
+        token = "A8IFZmgW2cqA4yUNlLPnci0N"  # Improvement: Do not hardcode tokens; use environment variables.
+        if not token:
+            return "Vercel 토큰이 설정되지 않았습니다."
+
+        project_name = ''.join(random.choice(string.ascii_lowercase) for i in range(6))
+        deploy_url = "https://api.vercel.com/v13/deployments"
+        
+        headers = {
+            "Authorization": f"Bearer {token}",
+            "Content-Type": "application/json"
+        }
+
+        package_json = {
+            "name": project_name,
+            "version": "1.0.0",
+            "private": True,
+            "dependencies": {
+                "vite": "^5.0.0"
+            },
+            "scripts": {
+                "dev": "vite",
+                "build": "echo 'No build needed' && mkdir -p dist && cp index.html dist/",
+                "preview": "vite preview"
+            }
+        }
+
+        files = [
+            {
+                "file": "index.html",
+                "data": code
+            },
+            {
+                "file": "package.json",
+                "data": json.dumps(package_json, indent=2)
+            }
+        ]
+
+        project_settings = {
+            "buildCommand": "npm run build",
+            "outputDirectory": "dist",
+            "installCommand": "npm install",
+            "framework": None
+        }
+
+        deploy_data = {
+            "name": project_name,
+            "files": files,
+            "target": "production",
+            "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}"
+
+        deployment_url = f"{project_name}.vercel.app"
+        time.sleep(5)
+
+        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)}"
+
+def boost_prompt(prompt: str) -> str:
+    if not prompt:
+        return ""
+    
+    boost_system_prompt = """
+    당신은 Gradio 웹앱 개발 프롬프트 전문가입니다.
+    주어진 프롬프트를 분석하여 더 상세하고 전문적인 요구사항으로 확장하되,
+    원래 의도와 목적은 그대로 유지하면서 다음 관점들을 고려하여 증강하십시오:
+
+    1. UI/UX 디자인 요소
+    2. Gradio 컴포넌트 활용
+    3. 사용자 경험 최적화
+    4. 성능과 보안
+    5. 접근성과 호환성
+    
+    기존 SystemPrompt의 모든 규칙을 준수하면서 증강된 프롬프트를 생성하십시오.
+    """
+    
+    try:
+        try:
+            response = claude_client.messages.create(
+                model="claude-3-5-sonnet-20241022",
+                max_tokens=2000,
+                messages=[{
+                    "role": "user",
+                    "content": f"다음 프롬프트를 분석하고 증강하시오: {prompt}"
+                }]
+            )
+            
+            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)}")
+            
+            completion = 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
+            )
+            
+            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
+
+def handle_boost(prompt: str):
+    try:
+        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")
+
+def execute_code(query: str):
+    if not query or query.strip() == '':
+        return None, gr.update(active_key="empty")
+    
+    try:
+        if '```html' in query and '```' in query:
+            code = remove_code_block(query)
+        else:
+            code = query.strip()
+        
+        return send_to_sandbox(code), gr.update(active_key="render")
+    except Exception as e:
+        print(f"Error executing code: {str(e)}")
+        return None, gr.update(active_key="empty")
+
+def remove_code_block(text):
+    text = re.sub(r'```[python|html]?\n', '', text)
+    text = re.sub(r'\n```', '', text)
+    
+    lines = text.split('\n')
+    filtered_lines = []
+    seen_imports = set()
+    
+    for line in lines:
+        if not line.strip():
+            continue
+            
+        if line.startswith('import ') or line.startswith('from '):
+            import_key = line.split('#')[0].strip()
+            if import_key in seen_imports:
+                continue
+            seen_imports.add(import_key)
+            
+        if 'if __name__ == "__main__":' in line:
+            continue
+        if 'demo.launch()' in line:
+            continue
+            
+        filtered_lines.append(line)
+        
+    return '\n'.join(filtered_lines)
+
+def history_render(history: History):
+    return gr.update(open=True), history
+
+def send_to_sandbox(code):
+    encoded_html = base64.b64encode(code.encode('utf-8')).decode('utf-8')
+    data_uri = f"data:text/html;charset=utf-8;base64,{encoded_html}"
+    return f"<iframe src=\"{data_uri}\" width=\"100%\" height=\"920px\"></iframe>"
+
+def load_json_data():
+    # 템플릿 데이터 반환
+    return [
+        # ... (이전에 정의된 템플릿 데이터)
+    ]
+
+def load_best_templates():
+    json_data = load_json_data()[:12]
+    return create_template_html("🏆 베스트 템플릿", json_data)
+
+def load_trending_templates():
+    json_data = load_json_data()[12:24]
+    return create_template_html("🔥 트렌딩 템플릿", json_data)
+
+def load_new_templates():
+    json_data = load_json_data()[24:44]
+    return create_template_html("✨ NEW 템플릿", json_data)
+
+def create_template_html(title, items):
+    html_content = """
+    <style>
+        .prompt-grid {
+            display: grid;
+            grid-template-columns: repeat(auto-fit, minmax(300px, 1fr));
+            gap: 20px;
+            padding: 20px;
+        }
+        .prompt-card {
+            background: white;
+            border: 1px solid #eee;
+            border-radius: 8px;
+            padding: 15px;
+            cursor: pointer;
+            box-shadow: 0 2px 5px rgba(0,0,0,0.1);
+        }
+        .prompt-card:hover {
+            transform: translateY(-2px);
+            transition: transform 0.2s;
+        }
+        .card-image {
+            width: 100%;
+            height: 180px;
+            object-fit: cover;
+            border-radius: 4px;
+            margin-bottom: 10px;
+        }
+        .card-name {
+            font-weight: bold;
+            margin-bottom: 8px;
+            font-size: 16px;
+            color: #333;
+        }
+        .card-prompt {
+            font-size: 11px;
+            line-height: 1.4;
+            color: #666;
+            display: -webkit-box;
+            -webkit-line-clamp: 6;
+            -webkit-box-orient: vertical;
+            overflow: hidden;
+            height: 90px;
+            background-color: #f8f9fa;
+            padding: 8px;
+            border-radius: 4px;
+        }
+    </style>
+    <div class="prompt-grid">
+    """
+
+    for item in items:
+        html_content += f"""
+        <div class="prompt-card" onclick="copyToInput(this)" data-prompt="{html.escape(item.get('prompt', ''))}">
+            <img src="{item.get('image_url', '')}" class="card-image" loading="lazy" alt="{html.escape(item.get('name', ''))}">
+            <div class="card-name">{html.escape(item.get('name', ''))}</div>
+            <div class="card-prompt">{html.escape(item.get('prompt', ''))}</div>
+        </div>
+        """
+
+    html_content += """
+    <script>
+    function copyToInput(card) {
+        const prompt = card.dataset.prompt;
+        const textarea = document.querySelector('.ant-input-textarea-large textarea');
+        if (textarea) {
+            textarea.value = prompt;
+            textarea.dispatchEvent(new Event('input', { bubbles: true }));
+            document.querySelector('.session-drawer .close-btn').click();
+        }
+    }
+    </script>
+    </div>
+    """
+    return gr.HTML(value=html_content)
 
 class Demo:
     def __init__(self):
@@ -546,7 +795,7 @@ class Demo:
                     yield [
                         code,
                         _history,
-                        analysis,  # 분석 결과를 HTML로 전달
+                        analysis,
                         gr.update(active_key="loading"),
                         gr.update(open=True)
                     ]
@@ -562,7 +811,7 @@ class Demo:
                     yield [
                         code,
                         _history,
-                        analysis,  # 분석 결과를 HTML로 전달
+                        analysis,
                         gr.update(active_key="loading"),
                         gr.update(open=True)
                     ]
@@ -580,7 +829,7 @@ class Demo:
                 yield [
                     collected_content,
                     _history,
-                    analyze_code(collected_content),  # 최종 분석 결과를 HTML로 전달
+                    analyze_code(collected_content),
                     gr.update(active_key="render"),
                     gr.update(open=True)
                 ]
@@ -594,49 +843,9 @@ class Demo:
     def clear_history(self):
         return []
 
-def remove_code_block(text):
-    # Remove markdown code block syntax
-    text = re.sub(r'```[python|html]?\n', '', text)
-    text = re.sub(r'\n```', '', text)
-    
-    # Remove duplicate imports and launch configurations
-    lines = text.split('\n')
-    filtered_lines = []
-    seen_imports = set()
-    
-    for line in lines:
-        # Skip empty lines
-        if not line.strip():
-            continue
-            
-        # Skip duplicate imports
-        if line.startswith('import ') or line.startswith('from '):
-            import_key = line.split('#')[0].strip()
-            if import_key in seen_imports:
-                continue
-            seen_imports.add(import_key)
-            
-        # Skip duplicate launch configurations    
-        if 'if __name__ == "__main__":' in line:
-            continue
-        if 'demo.launch()' in line:
-            continue
-            
-        filtered_lines.append(line)
-        
-    return '\n'.join(filtered_lines)
-
-def history_render(history: History):
-    return gr.update(open=True), history
-
-def send_to_sandbox(code):
-    encoded_html = base64.b64encode(code.encode('utf-8')).decode('utf-8')
-    data_uri = f"data:text/html;charset=utf-8;base64,{encoded_html}"
-    return f"<iframe src=\"{data_uri}\" width=\"100%\" height=\"920px\"></iframe>"
 
-
-
-theme = gr.themes.Soft()
+# theme = gr.themes.Soft()
+theme='earneleh/paris' 
 
 def load_json_data():
     return [
@@ -951,20 +1160,18 @@ def create_template_html(title, items):
             padding: 8px;
             border-radius: 4px;
         }
+        .template-section {
+            display: none;
+        }
+        .template-section.active {
+            display: block;
+        }
     </style>
-    <div class="prompt-grid">
-    """
-
-    for item in items:
-        html_content += f"""
-        <div class="prompt-card" onclick="copyToInput(this)" data-prompt="{html.escape(item.get('prompt', ''))}">
-            <img src="{item.get('image_url', '')}" class="card-image" loading="lazy" alt="{html.escape(item.get('name', ''))}">
-            <div class="card-name">{html.escape(item.get('name', ''))}</div>
-            <div class="card-prompt">{html.escape(item.get('prompt', ''))}</div>
-        </div>
-        """
-
-    html_content += """
+    <div class="template-nav">
+        <button class="template-btn" onclick="showTemplate('best')">🏆 베스트</button>
+        <button class="template-btn" onclick="showTemplate('trending')">🔥 트렌딩</button>
+        <button class="template-btn" onclick="showTemplate('new')">✨ NEW</button>
+    </div>
     <script>
     function copyToInput(card) {
         const prompt = card.dataset.prompt;
@@ -975,14 +1182,37 @@ def create_template_html(title, items):
             document.querySelector('.session-drawer .close-btn').click();
         }
     }
+
+    function showTemplate(type) {
+        document.querySelectorAll('.template-section').forEach(section => {
+            section.style.display = 'none';
+        });
+        document.querySelectorAll('.template-btn').forEach(btn => {
+            btn.classList.remove('active');
+        });
+        document.getElementById(type + '-templates').style.display = 'block';
+        event.target.classList.add('active');
+    }
+    document.addEventListener('DOMContentLoaded', function() {
+        showTemplate('best');
+        document.querySelector('.template-btn').classList.add('active');
+    });
     </script>
-    </div>
     """
-    return gr.HTML(value=html_content)
 
-    
-# 전역 변수로 템플릿 데이터 캐시
-TEMPLATE_CACHE = None
+    # 탭별 템플릿 목록
+    html_content += f'<div id="best-templates" class="template-section">'
+    html_content += '<div class="prompt-grid">'
+    for item in items:
+        html_content += f"""
+        <div class="prompt-card" onclick="copyToInput(this)" data-prompt="{html.escape(item.get('prompt', ''))}">
+            <img src="{item.get('image_url', '')}" class="card-image" loading="lazy" alt="{html.escape(item.get('name', ''))}">
+            <div class="card-name">{html.escape(item.get('name', ''))}</div>
+            <div class="card-prompt">{html.escape(item.get('prompt', ''))}</div>
+        </div>
+        """
+    html_content += '</div></div>'
+    return gr.HTML(value=html_content)
 
 def load_session_history(template_type="best"):
     global TEMPLATE_CACHE
@@ -1150,10 +1380,6 @@ def load_session_history(template_type="best"):
         print(f"Error in load_session_history: {str(e)}")
         return gr.HTML("Error loading templates")
 
-
-
-
-
 # 배포 관련 함수 추가
 def generate_space_name():
     """6자리 랜덤 영문 이름 생성"""
@@ -1217,7 +1443,7 @@ def deploy_to_huggingface(code: str):
         
         # 기본 requirements 설정
         default_requirements = [
-            'gradio==5.5.0',
+            'gradio==5.6.0',
             'huggingface_hub',
             'transformers',
             'torch',
@@ -1243,7 +1469,6 @@ def deploy_to_huggingface(code: str):
         if requirements:
             for req in requirements.split('\n'):
                 if req.strip():
-                    pkg_name = req.split('==')[0] if '==' in req else req
                     final_requirements.add(req.strip())
         
         # 기본 requirements 추가 (없는 것만)
@@ -1272,7 +1497,6 @@ def deploy_to_huggingface(code: str):
 # Demo 인스턴스 생성
 demo_instance = Demo()
 
-
 with gr.Blocks(css_paths="app.css",theme=theme) as demo:
     history = gr.State([])
     setting = gr.State({
@@ -1309,32 +1533,29 @@ with gr.Blocks(css_paths="app.css",theme=theme) as demo:
             # 메인 컨텐츠를 위한 Row
             with antd.Row(gutter=[32, 12]) as layout:    
                 # 좌측 패널
-
-# 좌측 패널 부분을 다음과 같이 수정
                 with antd.Col(span=24, md=8):
                     with antd.Flex(vertical=True, gap="middle", wrap=True):
                         header = gr.HTML(f"""
-                                    <div class="left_header">
-                                     <img src="data:image/gif;base64,{get_image_base64('mouse.gif')}" width="360px" />
-                                     <h1 style="font-size: 18px;">고양이도 발로 코딩하는 'MOUSE-II'</h2>
-                                     <h1 style="font-size: 10px;">템플릿의 프롬프트 내용을 복사해 프롬프트에 입력 'Send'버튼 클릭 -> '실행하기' 버튼 클릭하여 코드 실행. 문의: [email protected] </h2> 
-                                    </div>
-                                     """)
+                            <div class="left_header">
+                                <img src="data:image/gif;base64,{get_image_base64('mouse.gif')}" width="360px" />
+                                <h1 style="font-size: 18px;">고양이도 발로 코딩하는 'MOUSE-II'</h2>
+                                <h1 style="font-size: 10px;">템플릿의 프롬프트를 복사하거나 원하는 내용을 프롬프트에 입력하고 Boost 버튼 클릭시 프롬프트 증강이 됨. -> 배포하기 클릭하면 허깅페이스를 통해 웹서비스가 배포됩니다. 문의: [email protected] </h2>                                 
+                            </div>
+                        """)
+
                         input = antd.InputTextarea(
                             size="large", 
                             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")
-                            deploy_btn = antd.Button("실행하기", type="default", size="large")  # 추가
-                            clear_btn = antd.Button("Clear", type="default", size="large")
+                            boost_btn = antd.Button("Boost", type="default", size="large")
+                            execute_btn = antd.Button("Code실행", type="default", size="large")
+                            deploy_btn = antd.Button("배포", type="default", size="large")
+                            clear_btn = antd.Button("클리어", type="default", size="large")
 
-# 배포 결과를 ��시할 텍스트 영역 추가
- 
                         deploy_result = gr.HTML(label="배포 결과")
 
                 with antd.Col(span=24, md=16):
@@ -1342,14 +1563,12 @@ with gr.Blocks(css_paths="app.css",theme=theme) as demo:
                         with antd.Flex(gap="small", elem_classes="setting-buttons"):
                             codeBtn = antd.Button("🧑‍💻 코드 보기", type="default")
                             historyBtn = antd.Button("📜 히스토리", type="default")
-                            best_btn = antd.Button("🏆 기본 템플릿", type="default")
-                            trending_btn = antd.Button("🔥 응용 템플릿", type="default")
-                            new_btn = antd.Button("✨ 활용 템플릿", type="default")
+                            best_btn = antd.Button("🏆 베스트 템플릿", type="default")
+                            trending_btn = antd.Button("🔥 트렌딩 템플릿", type="default")
+                            new_btn = antd.Button("✨ NEW 템플릿", type="default")
 
                         gr.HTML('<div class="render_header"><span class="header_btn"></span><span class="header_btn"></span><span class="header_btn"></span></div>')
 
-        
-      
                         with antd.Tabs(active_key="empty", render_tab_bar="() => null") as state_tab:
                             with antd.Tabs.Item(key="empty"):
                                 empty = antd.Empty(description="empty input", elem_classes="right_content")
@@ -1358,27 +1577,36 @@ with gr.Blocks(css_paths="app.css",theme=theme) as demo:
                             with antd.Tabs.Item(key="render"):
                                 sandbox = gr.HTML(elem_classes="html_content")
 
-            # Code 실행 버튼 이벤트 핸들러 함수 정의
-            def execute_code(query: str):
-                if not query or query.strip() == '':
-                    return None, gr.update(active_key="empty")
-                
-                try:
-                    # HTML 코드 블록 확인
-                    if '```html' in query and '```' in query:
-                        # HTML 코드 블록 추출
-                        code = remove_code_block(query)
-                    else:
-                        # 입력된 텍스트를 그대로 코드로 사용
-                        code = query.strip()
-                    
-                    return send_to_sandbox(code), gr.update(active_key="render")
-                except Exception as e:
-                    print(f"Error executing code: {str(e)}")
-                    return None, gr.update(active_key="empty")
-
-# 이벤트 핸들러들
+            # 이벤트 핸들러 연결
+            execute_btn.click(
+                fn=execute_code,
+                inputs=[input],
+                outputs=[sandbox, state_tab]
+            )
+
+            boost_btn.click(
+                fn=handle_boost,
+                inputs=[input],
+                outputs=[input, state_tab]
+            )
+
+            btn.click(
+                demo_instance.generation_code,
+                inputs=[input, setting, history],
+                outputs=[code_output, history, sandbox, state_tab, code_drawer]
+            )
+
+            clear_btn.click(
+                demo_instance.clear_history,
+                inputs=[],
+                outputs=[history]
+            )
 
+            deploy_btn.click(
+                fn=lambda code: deploy_to_huggingface(remove_code_block(code)) if code else "코드가 없습니다.",
+                inputs=[code_output],
+                outputs=[deploy_result]
+            )
 
             codeBtn.click(
                 lambda: gr.update(open=True),
@@ -1404,7 +1632,6 @@ with gr.Blocks(css_paths="app.css",theme=theme) as demo:
                 outputs=[history_drawer]
             )
 
-# 템플릿 버튼 이벤트 핸들러
             best_btn.click(
                 fn=lambda: (gr.update(open=True), load_best_templates()),
                 outputs=[session_drawer, session_history],
@@ -1433,26 +1660,6 @@ with gr.Blocks(css_paths="app.css",theme=theme) as demo:
                 outputs=[session_drawer, session_history]
             )
 
-            btn.click(
-                demo_instance.generation_code,
-                inputs=[input, setting, history],
-                outputs=[code_output, history, sandbox, state_tab, code_drawer]
-            )
-
-            clear_btn.click(
-                demo_instance.clear_history,
-                inputs=[],
-                outputs=[history]
-            )
-
-# 이벤트 핸들러 추가
-            deploy_btn.click(
-                fn=lambda code: deploy_to_huggingface(remove_code_block(code)) if code else "코드가 없습니다.",
-                inputs=[code_output],
-                outputs=[deploy_result]
-            )
-
-
 if __name__ == "__main__":
     try:
         demo_instance = Demo()
@@ -1460,4 +1667,3 @@ if __name__ == "__main__":
     except Exception as e:
         print(f"Initialization error: {e}")
         raise
-