Update app.py

app.py

@@ -444,54 +444,50 @@ def load_session_history(selected_session):
             color: #666;
             margin-top: 10px;
         }
-        #code-display {
-            display: none;
-        }
         </style>
         <div class="prompt-grid">
         """
         
-        # 응답 데이터를 JavaScript 배열로 준비
-        responses_data = []
-        prompts_data = []
-        
         for i, (prompt, response, timestamp) in enumerate(history):
             short_prompt = prompt[:100] + "..." if len(prompt) > 100 else prompt
             formatted_time = datetime.strptime(timestamp, '%Y-%m-%d %H:%M:%S.%f').strftime('%Y-%m-%d %H:%M')
             
-            responses_data.append(response)
-            prompts_data.append(prompt)
+            # HTML 이스케이프 처리
+            escaped_prompt = html.escape(prompt)
+            escaped_response = html.escape(response)
             
             cards_html += f"""
-            <div class="prompt-card" onclick="executeHistoryItem({i})">
+            <div class="prompt-card" 
+                 onclick="executeHistoryItem(`{escaped_prompt}`, `{escaped_response}`)">
                 <div>{short_prompt}</div>
                 <div class="timestamp">{formatted_time}</div>
             </div>
             """
         
-        cards_html += "</div>"
-        
-        # JavaScript 함수 수정
-        cards_html += f"""
+        cards_html += """
+        </div>
         <script>
-        const responses = {json.dumps(responses_data)};
-        const prompts = {json.dumps(prompts_data)};
-        
-        function executeHistoryItem(index) {{
-            // 입력 필드에 프롬프트 설정
-            document.querySelector('textarea').value = prompts[index];
+        function executeHistoryItem(prompt, response) {
+            // 입력 필드 업데이트
+            document.querySelector('textarea').value = prompt;
             
             // 코드 실행
-            const code = responses[index];
+            const code = response.includes('```html') ? 
+                response.match(/```html\\n([\\s\\S]*?)\\n```/)[1] : 
+                response;
+            
             const encodedHtml = btoa(code);
-            const dataUri = `data:text/html;charset=utf-8;base64,${{encodedHtml}}`;
+            const dataUri = `data:text/html;charset=utf-8;base64,${encodedHtml}`;
             
             // iframe 업데이트
             document.querySelector('.html_content iframe').src = dataUri;
             
             // 세션 드로어 닫기
             document.querySelector('.session-drawer').style.display = 'none';
-        }}
+            
+            // 렌더 탭으로 변경
+            document.querySelector('[data-testid="tab-render"]').click();
+        }
         </script>
         """
         
@@ -609,12 +605,11 @@ with gr.Blocks(css_paths="app.css",theme=theme) as demo:
                 outputs=[session_history]
             )
 
-# 세션 히스토리 카드 클릭 이벤트 (session_list.change 다음에 위치)
-            session_history.click(
-                fn=execute_history_item,
-                inputs=[],
-                outputs=[input, sandbox, state_tab, session_drawer]
-            )
+#            session_history.click(
+#                fn=execute_history_item,
+#                inputs=[],
+#                outputs=[input, sandbox, state_tab, session_drawer]
+#            )
 
             
             # 메인 컨텐츠를 위한 Row