Update app.py

app.py

@@ -564,11 +564,28 @@ def get_random_placeholder():
 def update_placeholder():
     return gr.update(placeholder=get_random_placeholder())
 
+
+# 함수를 상위 레벨로 이동하고 들여쓰기 수정
+async def handle_generation(query, setting, is_search):
+    try:
+        if is_search:
+            result = await execute_search_and_generate(query, setting)  # await 추가
+            
+            return await execute_search_and_generate(query, setting)
+        else:
+            async for result in demo_instance.generation_code(query, setting):
+                final_result = result
+            return final_result
+    except Exception as e:
+        print(f"Generation error: {str(e)}")
+        return ["", None, gr.update(active_key="error"), gr.update(open=False)]
+
+        
 def create_main_interface():
     """메인 인터페이스 생성 함수"""
     
     #NEW - 검색 결과를 통합한 응답 생성 함수
-    def execute_search_and_generate(query, setting):
+    async def execute_search_and_generate(query, setting):
         try:
             # 검색 실행
             url = "https://api.serphouse.com/serp/live"
@@ -615,7 +632,11 @@ def create_main_interface():
             Create a comprehensive visual response that incorporates relevant information from the search results.
             """
             
-            return demo_instance.generation_code(enhanced_prompt, setting)
+            # async generator를 처리하기 위한 수정
+            async for result in demo_instance.generation_code(enhanced_prompt, setting):
+                final_result = result
+            return final_result
+            
         except Exception as e:
             print(f"Search error: {str(e)}")
             return [
@@ -1004,20 +1025,11 @@ Use the "Generate" button for basic creation, "Enhance" button for prompt improv
 
                         # 여기에 추가 (들여쓰기 맞춤)
                         mode.change(
-                            fn=lambda x: x == "Web Search",
+                            fn=lambda x: x == "Generate + Web Search",
                             inputs=[mode],
                             outputs=[search_mode]
                         )
 
-                        async def handle_generation(query, setting, is_search):
-                            try:
-                                if is_search:
-                                    return await execute_search_and_generate(query, setting)
-                                return await demo_instance.generation_code(query, setting)
-                            except Exception as e:
-                                print(f"Generation error: {str(e)}")
-                                return ["", None, gr.update(active_key="error"), gr.update(open=False)]
-
 
                         btn.click(
                             fn=handle_generation,