Update app.py

app.py

@@ -445,18 +445,56 @@ def load_json_data():
 # 전역 변수로 템플릿 데이터 캐시
 TEMPLATE_CACHE = None
 
-def load_session_history(selected_session=None):
+def load_session_history(template_type="best"):
     global TEMPLATE_CACHE
     
     try:
-        if TEMPLATE_CACHE:
-            return TEMPLATE_CACHE
-
-        # 최소한의 카드만 먼저 로드 (첫 12개)
-        json_data = load_json_data()[:16]
+        json_data = load_json_data()
+        
+        # 데이터를 세 섹션으로 나누기
+        templates = {
+            "best": json_data[:12],      # 베스트 템플릿
+            "trending": json_data[12:24], # 트렌딩 템플릿
+            "new": json_data[24:36]      # NEW 템플릿
+        }
+        
+        titles = {
+            "best": "🏆 베스트 템플릿",
+            "trending": "🔥 트렌딩 템플릿",
+            "new": "✨ NEW 템플릿"
+        }
         
         html_content = """
         <style>
+            .template-nav {
+                display: flex;
+                gap: 10px;
+                margin: 20px;
+                position: sticky;
+                top: 0;
+                background: white;
+                z-index: 100;
+                padding: 10px 0;
+                border-bottom: 1px solid #eee;
+            }
+            .template-btn {
+                padding: 8px 16px;
+                border: 1px solid #1890ff;
+                border-radius: 4px;
+                cursor: pointer;
+                background: white;
+                color: #1890ff;
+                font-weight: bold;
+                transition: all 0.3s;
+            }
+            .template-btn:hover {
+                background: #1890ff;
+                color: white;
+            }
+            .template-btn.active {
+                background: #1890ff;
+                color: white;
+            }
             .prompt-grid {
                 display: grid;
                 grid-template-columns: repeat(auto-fit, minmax(300px, 1fr));
@@ -501,18 +539,35 @@ def load_session_history(selected_session=None):
                 padding: 8px;
                 border-radius: 4px;
             }
+            .template-section {
+                display: none;
+            }
+            .template-section.active {
+                display: block;
+            }
         </style>
-        <div class="prompt-grid" id="promptGrid">
+        <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>
         """
 
-        for item in json_data:
+        # 각 섹션의 템플릿 생성
+        for section, items in templates.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>
+            <div class="template-section" id="{section}-templates">
+                <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>"
 
         html_content += """
         <script>
@@ -525,12 +580,32 @@ def load_session_history(selected_session=None):
                 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>
         """
         
-        TEMPLATE_CACHE = gr.HTML(value=html_content)
-        return TEMPLATE_CACHE
+        return gr.HTML(value=html_content)
         
     except Exception as e:
         print(f"Error in load_session_history: {str(e)}")
@@ -686,16 +761,16 @@ with gr.Blocks(css_paths="app.css",theme=theme) as demo:
                 outputs=[history_drawer]
             )
 
-
-
 # 세션 버튼 클릭 이벤트 수정
             sessionBtn.click(
                 fn=lambda: (gr.update(open=True), load_session_history()),
                 inputs=[],
                 outputs=[session_drawer, session_history],
-                queue=False  # 큐 비활성화로 즉시 실행
+                queue=False
             )
 
+
+
 # 세션 드로어 닫기 이벤트 수정  
             session_drawer.close(
                 lambda: (gr.update(open=False), gr.HTML("")),