Update app.py

app.py

@@ -588,6 +588,8 @@ def create_main_interface():
 
     async def execute_search_and_generate(query, setting):
           try:
+               print(f"Executing search for query: {query}")  # 디버깅용
+               
                # 검색 실행
                url = "https://api.serphouse.com/serp/live"
                payload = {
@@ -596,70 +598,87 @@ def create_main_interface():
                          "domain": "google.com", 
                          "lang": "en",
                          "device": "desktop",
-                         "serp_type": "news",  # 뉴스 검색으로 변경
+                         "serp_type": "news",
+                         "loc": "United States",
                          "page": "1",
-                         "num": "10"
+                         "num": "10",
+                         "ie": "UTF-8",
+                         "tbm": "nws"
                     }
                }
                headers = {
-                    "accept": "application/json",
-                    "content-type": "application/json",
-                    "authorization": "Bearer V38CNn4HXpLtynJQyOeoUensTEYoFy8PBUxKpDqAW1pawT1vfJ2BWtPQ98h6"
+                    "Authorization": "Bearer V38CNn4HXpLtynJQyOeoUensTEYoFy8PBUxKpDqAW1pawT1vfJ2BWtPQ98h6",
+                    "Content-Type": "application/json"
                }
                
+               print("Sending API request...")  # 디버깅용
                response = requests.post(url, headers=headers, json=payload)
+               print(f"API Response status: {response.status_code}")  # 디버깅용
+               
+               if response.status_code != 200:
+                    print(f"API Error: {response.text}")  # 디버깅용
+                    raise Exception(f"API request failed with status {response.status_code}")
+                    
                results = response.json()
+               print(f"Got search results: {results.keys()}")  # 디버깅용
                
                # 검색 결과를 HTML로 변환하고 요약
                search_summary = "<div class='search-results'>\n"
                search_summary += "<h2>최신 뉴스 검색 결과</h2>\n"
                
-               if 'results' in results and 'news' in results['results']:
-                    for item in results['results']['news'][:5]:  # 상위 5개 결과만 사용
+               if 'results' in results:
+                    news_results = results['results'].get('news', [])
+                    print(f"Found {len(news_results)} news items")  # 디버깅용
+                    
+                    for item in news_results[:5]:  # 상위 5개 결과만 사용
                          search_summary += f"""
                               <div class="search-item">
-                                   <h3><a href="{item['url']}" target="_blank">{item['title']}</a></h3>
-                                   <p>{item['snippet']}</p>
+                                   <h3><a href="{item.get('url', '#')}">{item.get('title', 'No Title')}</a></h3>
+                                   <p>{item.get('snippet', 'No description available')}</p>
                                    <div class="search-meta">
-                                        <span class="source">{item['channel']}</span>
-                                        <span class="time">{item['time']}</span>
+                                        <span class="source">{item.get('channel', 'Unknown')}</span>
+                                        <span class="time">{item.get('time', '')}</span>
                                    </div>
                               </div>
                          """
                search_summary += "</div>"
 
                # 검색 결과를 포함한 향상된 프롬프트 생성
-               enhanced_prompt = f"""Based on the latest news search results, create a comprehensive and visually appealing response:
+               enhanced_prompt = f"""Based on these news search results, create a comprehensive visual summary:
 
-Search Query: {query}
+Original Query: {query}
 
-Search Results Summary:
 {search_summary}
 
-Please create a response that:
-1. Summarizes the key information from the search results
-2. Presents the information in a clear and organized way
-3. Uses appropriate HTML formatting for better visual presentation
-4. Includes relevant quotes or statistics when available
-5. Provides proper attribution to sources
+Please create a visually appealing HTML response that:
+1. Summarizes the key points from the news results
+2. Organizes information in a clear and structured way
+3. Uses appropriate styling for better readability
+4. Includes important quotes and facts
+5. Cites sources properly
 
-Response should be in HTML format with appropriate styling.
+Make sure to maintain proper HTML formatting in the response.
 """
 
+               print("Generating response with enhanced prompt...")  # 디버깅용
+               
                # async generator를 처리하기 위한 수정
                async for result in demo_instance.generation_code(enhanced_prompt, setting):
                     final_result = result
+               
+               print("Response generation completed")  # 디버깅용
                return final_result
                     
           except Exception as e:
-               print(f"Search error: {str(e)}")
+               print(f"Search error: {str(e)}")  # 디버깅용
+               print(f"Full error details: {str(e.__class__.__name__)}: {str(e)}")  # 디버깅용
                return [
                     "",
                     None,
                     gr.update(active_key="error"),
                     gr.update(open=False)
                ]
-
+              
     def execute_code(query: str):
         if not query or query.strip() == '':
             return None, gr.update(active_key="empty")