Update app.py

app.py

@@ -569,9 +569,24 @@ def update_placeholder():
 def create_main_interface():
     """메인 인터페이스 생성 함수"""
     
-    #NEW - 검색 결과를 통합한 응답 생성 함수
+    async def handle_generation(query, setting, is_search):
+        try:
+            print(f"Mode: {'Web Search' if is_search else 'Generate'}")  # 디버깅용
+            if is_search:
+                print("Executing search and generate...")  # 디버깅용
+                return await execute_search_and_generate(query, setting)
+            else:
+                print("Executing normal generation...")  # 디버깅용
+                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)]
+
     async def execute_search_and_generate(query, setting):
         try:
+            print(f"Executing web search for query: {query}")  # 디버깅용
             # 검색 실행
             url = "https://api.serphouse.com/serp/live"
             payload = {
@@ -594,10 +609,12 @@ def create_main_interface():
                 'Content-Type': 'application/json'
             }
             
+            print("Sending search request...")  # 디버깅용
             response = requests.post(url, headers=headers, json=payload)
             results = response.json()
+            print(f"Search results received: {len(results.get('results', []))} items")  # 디버깅용
             
-            # 검색 결과를 HTML로 변환 - 수정된 부분
+            # 검색 결과를 HTML로 변환
             search_content = "<div class='search-summary'><h2>Search Results</h2>"
             if 'results' in results and isinstance(results['results'], list):
                 search_results = results['results']
@@ -639,25 +656,6 @@ def create_main_interface():
                 gr.update(active_key="error"),
                 gr.update(open=False)
             ]
-
-    async def handle_generation(query, setting, is_search):
-        try:
-            print(f"Mode: {'Web Search' if is_search else 'Generate'}")  # 디버깅용
-            if is_search:
-            print("Executing search and generate...")  # 디버깅용
-
-                return await execute_search_and_generate(query, setting)
-            else:
-            print("Executing normal generation...")  # 디버깅용
-            
-                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 execute_code(query: str):
         if not query or query.strip() == '':