Update app.py

app.py

@@ -756,21 +756,18 @@ with gr.Blocks(css_paths="app.css",theme=theme) as demo:
                             placeholder=random.choice(DEMO_LIST)['description']
                         )
 
-# 버튼들을 가로로 배치하기 위한 Flex 컨테이너들
-                        with antd.Flex(vertical=True, gap="middle"):  # vertical을 사용하여 세로 방향으로 배치
+                        # 버튼들을 가로로 배치하기 위한 Flex 컨테이너
+                        with antd.Flex(vertical=True, gap="middle"):
                             with antd.Flex(gap="small", justify="space-between"):
                                 btn = antd.Button("Send", type="primary", size="large")
                                 execute_btn = antd.Button("Code 실행", type="default", size="large")
                                 clear_btn = antd.Button("Clear", type="default", size="large")
 
-    # 템플릿 버튼들을 위한 새로운 Flex 컨테이너
+                            # 템플릿 버튼들을 위한 새로운 Flex 컨테이너
                             with antd.Flex(gap="small", justify="space-between"):
                                 best_btn = antd.Button("🏆 베스트 템플릿", type="default", size="large")
                                 trending_btn = antd.Button("🔥 트렌딩 템플릿", type="default", size="large")
                                 new_btn = antd.Button("✨ NEW 템플릿", type="default", size="large")
-
-                        
-
                                            
                 # 우측 패널
                 with antd.Col(span=24, md=16):
@@ -789,6 +786,25 @@ 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,