Update app.py

app.py

@@ -397,20 +397,39 @@ def update_session_list():
         print(f"Error updating session list: {e}")
         return gr.update(choices=[])
 
-# db.json 파일 읽기 함수 추가
 def load_json_data():
     try:
-        with open('db.json', 'r', encoding='utf-8') as f:
-            return json.load(f)
+        current_dir = os.path.dirname(os.path.abspath(__file__))
+        json_path = os.path.join(current_dir, 'db.json')
+        print(f"Attempting to load JSON from: {json_path}")
+        
+        with open(json_path, 'r', encoding='utf-8') as f:
+            data = json.load(f)
+            print(f"Successfully loaded {len(data)} items from db.json")
+            return data
     except Exception as e:
-        print(f"Error loading db.json: {e}")
-        return []
+        print(f"Error loading db.json: {str(e)}")
+        # 에러 시 기본 데이터 반환
+        return [
+            {
+                "name": "MBTI 진단 서비스",
+                "image_url": "mbti.png",
+                "prompt": "MBTI 진단을 위해 15개의 질문과 객관식 답변을 통해 MBTI 진단 결과 및 해당 성격에 대한 상세한 결과를 출력하라"
+            },
+            {
+                "name": "투자 포트폴리오 대시보드",
+                "image_url": "chart.png",
+                "prompt": "Create an interactive dashboard with Chart.js showing different types of charts..."
+            }
+        ]
 
-# 세션 히스토리 로드 함수 수정
 def load_session_history(selected_session=None):
     try:
-        # db.json 데이터 로드
+        print("Loading session history...")
         json_data = load_json_data()
+        print(f"Loaded {len(json_data)} items from JSON")
+        
+        current_dir = os.path.dirname(os.path.abspath(__file__))
         
         html_content = """
         <style>
@@ -428,6 +447,7 @@ def load_session_history(selected_session=None):
                 cursor: pointer;
                 transition: all 0.3s ease;
                 box-shadow: 0 2px 5px rgba(0,0,0,0.1);
+                min-height: 300px;
             }
             .prompt-card:hover {
                 transform: translateY(-2px);
@@ -443,6 +463,7 @@ def load_session_history(selected_session=None):
             .card-name {
                 font-weight: bold;
                 margin-bottom: 8px;
+                font-size: 16px;
             }
             .card-prompt {
                 font-size: 0.9em;
@@ -453,21 +474,20 @@ def load_session_history(selected_session=None):
                 overflow: hidden;
             }
         </style>
-        """
-
-        html_content += """
-        <script src="https://code.jquery.com/jquery-3.6.0.min.js"></script>
         <div class="prompt-grid">
         """
 
         for item in json_data:
             name = html.escape(item.get('name', ''))
-            image_url = html.escape(item.get('image_url', ''))
+            image_url = os.path.join(current_dir, item.get('image_url', ''))
             prompt = html.escape(item.get('prompt', ''))
             
+            print(f"Processing item: {name}")
+            print(f"Image path: {image_url}")
+            
             html_content += f"""
             <div class="prompt-card" data-prompt="{prompt}" onclick="handleCardClick(this)">
-                <img src="{image_url}" class="card-image" alt="{name}">
+                <img src="{image_url}" class="card-image" alt="{name}" onerror="this.src='default.png'">
                 <div class="card-name">{name}</div>
                 <div class="card-prompt">{prompt}</div>
             </div>
@@ -477,20 +497,24 @@ def load_session_history(selected_session=None):
         </div>
         <script>
         function handleCardClick(card) {
+            console.log('Card clicked');
             const prompt = card.getAttribute('data-prompt');
             const textarea = document.querySelector('textarea');
             if (textarea) {
+                console.log('Setting prompt:', prompt);
                 textarea.value = prompt;
                 
-                setTimeout(() => {
-                    const executeButton = $('button:contains("Code 실행")');
-                    if (executeButton.length) {
-                        executeButton[0].click();
-                    }
-                }, 100);
+                // Send 버튼 클릭
+                const sendButton = document.querySelector('button:contains("Send")');
+                if (sendButton) {
+                    console.log('Clicking Send button');
+                    sendButton.click();
+                }
                 
+                // 드로어 닫기
                 const drawer = document.querySelector('.session-drawer');
                 if (drawer) {
+                    console.log('Closing drawer');
                     drawer.style.display = 'none';
                 }
             }
@@ -498,11 +522,12 @@ def load_session_history(selected_session=None):
         </script>
         """
         
+        print("Generated HTML content length:", len(html_content))
         return gr.HTML(value=html_content)
         
     except Exception as e:
-        print(f"Error loading session history: {e}")
-        return gr.HTML("Error loading history")
+        print(f"Error in load_session_history: {str(e)}")
+        return gr.HTML("Error loading templates")
 
 # 히스토리 아이템 선택 처리 함수
 def handle_history_selection(evt: gr.SelectData):
@@ -584,9 +609,6 @@ with gr.Blocks(css_paths="app.css",theme=theme) as demo:
             with antd.Drawer(open=False, title="history", placement="left", width="900px") as history_drawer:
                 history_output = legacy.Chatbot(show_label=False, flushing=False, height=960, elem_classes="history_chatbot")
 
-
-# session_list 관련 코드를 모두 제거하고 수정된 버전입니다
-
             with antd.Drawer(
                 open=False, 
                 title="Templates", 
@@ -595,6 +617,7 @@ with gr.Blocks(css_paths="app.css",theme=theme) as demo:
                 elem_classes="session-drawer"
             ) as session_drawer:
                 with antd.Flex(vertical=True, gap="middle"):
+                    gr.Markdown("### Available Templates")
                     session_history = gr.HTML(
                         elem_classes="session-history"
                     )
@@ -604,8 +627,6 @@ with gr.Blocks(css_paths="app.css",theme=theme) as demo:
                         elem_classes="close-btn"
                     )
 
-
-
 # 세션 드로어에서 카드 클릭 시 실행할 함수 (Drawer 컴포넌트들 다음에 위치)
             def execute_history_item(evt: gr.SelectData):  # gr.SelectData로 이벤트 데이터 받기
                 try:
@@ -823,12 +844,12 @@ with gr.Blocks(css_paths="app.css",theme=theme) as demo:
 
 
 # 세션 버튼 클릭 이벤트 수정
+
             sessionBtn.click(
-                lambda: (gr.update(open=True), load_session_history()),
+                fn=lambda: (gr.update(open=True), load_session_history()),
                 inputs=[],
                 outputs=[session_drawer, session_history]
             )
-
 # 세션 드로어 닫기 이벤트 수정  
             session_drawer.close(
                 lambda: (gr.update(open=False), gr.HTML("")),
@@ -840,6 +861,7 @@ with gr.Blocks(css_paths="app.css",theme=theme) as demo:
                 outputs=[session_drawer, session_history]
             )
 
+            
 
             btn.click(
                 demo_instance.generation_code,