Update app.py

app.py

@@ -217,7 +217,31 @@ with gr.Blocks(css_paths="app.css") as demo:
                                 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 이벤트 핸들러 추가
+            execute_btn.click(
+                fn=execute_code,
+                inputs=[input],
+                outputs=[sandbox, state_tab]
+            )
 
             settingPromptBtn.click(lambda: gr.update(
                 open=True), inputs=[], outputs=[system_prompt_modal])
@@ -243,5 +267,6 @@ with gr.Blocks(css_paths="app.css") as demo:
             
             clear_btn.click(clear_history, inputs=[], outputs=[history])
 
+
 if __name__ == "__main__":
     demo.queue(default_concurrency_limit=20).launch(ssr_mode=False)