Update app.py

app.py

@@ -446,7 +446,7 @@ def load_json_data():
         }        
     ]
 
-# 전역 변수로 템플릿 HTML 캐시 저장
+# 템플릿 HTML 캐시 저장
 template_html_cache = None
 
 def load_session_history(selected_session=None):
@@ -521,7 +521,7 @@ def load_session_history(selected_session=None):
 
         for item in json_data:
             html_content += f"""
-            <div class="prompt-card" onclick="copyPrompt(this)" data-prompt="{html.escape(item.get('prompt', ''))}">
+            <div class="prompt-card" onclick="copyToInput('{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>
@@ -530,12 +530,14 @@ def load_session_history(selected_session=None):
 
         html_content += """
         <script>
-        function copyPrompt(card) {
-            const prompt = card.getAttribute('data-prompt');
+        function copyToInput(prompt) {
             const textarea = document.querySelector('.ant-input-textarea-large textarea');
             if (textarea) {
                 textarea.value = prompt;
                 textarea.dispatchEvent(new Event('input', { bubbles: true }));
+                // 드로어 닫기
+                const closeBtn = document.querySelector('.session-drawer .close-btn');
+                if (closeBtn) closeBtn.click();
             }
         }
         </script>
@@ -551,6 +553,8 @@ def load_session_history(selected_session=None):
         print(f"Error in load_session_history: {str(e)}")
         return gr.HTML("Error loading templates")
 
+
+
 # Demo 인스턴스 생성
 demo_instance = Demo()
 
@@ -701,25 +705,14 @@ with gr.Blocks(css_paths="app.css",theme=theme) as demo:
             )
 
 
+
+
 # 세션 버튼 클릭 이벤트 수정
             sessionBtn.click(
                 fn=lambda: (gr.update(open=True), load_session_history()),
                 inputs=[],
-                outputs=[session_drawer, session_history],
-                _js="""
-                () => {
-                    document.body.style.cursor = 'wait';
-                    const drawer = document.querySelector('.session-drawer');
-                    if (drawer) drawer.style.opacity = '0';
-                    setTimeout(() => {
-                        if (drawer) drawer.style.opacity = '1';
-                        document.body.style.cursor = 'default';
-                    }, 100);
-                }
-                """
+                outputs=[session_drawer, session_history]
             )
-
-
 # 세션 드로어 닫기 이벤트 수정  
             session_drawer.close(
                 lambda: (gr.update(open=False), gr.HTML("")),