Update app.py

app.py

@@ -1010,11 +1010,15 @@ Use the "Generate" button for basic creation, "Enhance" button for prompt improv
                         )
 
 
+                        async def handle_generation(query, setting, is_search):
+                            if is_search:
+                                result = await execute_search_and_generate(query, setting)
+                        else:
+                            result = await demo_instance.generation_code(query, setting)
+                        return result
 
-                                            
-                        # 버튼 이벤트 핸들러
                         btn.click(
-                            fn=lambda q, s, m: execute_search_and_generate(q, s) if m else demo_instance.generation_code(q, s),
+                            fn=handle_generation,
                             inputs=[input, setting, search_mode],
                             outputs=[code_output, sandbox, state_tab, code_drawer]
                         ).then(
@@ -1022,7 +1026,7 @@ Use the "Generate" button for basic creation, "Enhance" button for prompt improv
                             inputs=[],
                             outputs=[input]
                         )
-
+                                            
 
 
                         boost_btn.click(