Update app.py

app.py

@@ -32,7 +32,9 @@ from safetensors.torch import load_file
 from diffusers import FluxPipeline
 import torch
 from os import path  # 이 줄을 추가
-
+from datetime import datetime, timedelta
+from requests.adapters import HTTPAdapter
+from requests.packages.urllib3.util.retry import Retry
 # 캐시 경로 설정
 cache_path = path.join(path.dirname(path.abspath(__file__)), "models")
 os.environ["TRANSFORMERS_CACHE"] = cache_path
@@ -587,30 +589,44 @@ def create_main_interface():
     async def execute_search_and_generate(query, setting):
         try:
             print(f"Executing web search for query: {query}")  # 디버깅용
+            
             # 검색 실행
             url = "https://api.serphouse.com/serp/live"
+            
+            # 날짜 범위 설정
+            now = datetime.utcnow()
+            yesterday = now - timedelta(days=1)
+            date_range = f"{yesterday.strftime('%Y-%m-%d')},{now.strftime('%Y-%m-%d')}"
+            
             payload = {
                 "data": {
                     "q": query,
-                    "domain": "google.com", 
+                    "domain": "google.com",
+                    "loc": "United States",
                     "lang": "en",
                     "device": "desktop",
-                    "serp_type": "web",
-                    "loc": "Alba,Texas,United States",
-                    "loc_id": "1026201",
-                    "verbatim": "0",
-                    "gfilter": "0", 
+                    "serp_type": "news",  # 뉴스 검색으로 변경
                     "page": "1",
-                    "num_result": "5"
+                    "num": "10",
+                    "date_range": date_range,
+                    "sort_by": "date"
                 }
             }
+            
             headers = {
-                'Authorization': 'Bearer V38CNn4HXpLtynJQyOeoUensTEYoFy8PBUxKpDqAW1pawT1vfJ2BWtPQ98h6',
-                'Content-Type': 'application/json'
+                "accept": "application/json",
+                "content-type": "application/json",
+                "authorization": "Bearer V38CNn4HXpLtynJQyOeoUensTEYoFy8PBUxKpDqAW1pawT1vfJ2BWtPQ98h6"
             }
             
+            # 세션 설정 및 재시도 로직 추가
+            session = requests.Session()
+            retries = Retry(total=3, backoff_factor=0.5)
+            session.mount('https://', HTTPAdapter(max_retries=retries))
+            
             print("Sending search request...")  # 디버깅용
-            response = requests.post(url, headers=headers, json=payload)
+            response = session.post(url, json=payload, headers=headers, timeout=(5, 15))
+            response.raise_for_status()
             results = response.json()
             print(f"Search results received: {len(results.get('results', []))} items")  # 디버깅용
             
@@ -625,22 +641,24 @@ def create_main_interface():
                     title = result.get('title', 'No Title')
                     url = result.get('url', '#')
                     snippet = result.get('snippet', 'No description available')
+                    time = result.get('time', '')
                     
                     search_content += f"""
                         <div class="search-item">
                             <h3><a href="{url}" target="_blank">{title}</a></h3>
                             <p>{snippet}</p>
+                            <span class="search-time">{time}</span>
                         </div>
                     """
             search_content += "</div>"
             
             # 검색 결과를 포함한 프롬프트 생성
-            enhanced_prompt = f"""Based on these search results, create a visually appealing and informative response:
+            enhanced_prompt = f"""Based on these recent news results, create a visually appealing and informative response:
             {search_content}
             
             Original query: {query}
             
-            Create a comprehensive visual response that incorporates relevant information from the search results.
+            Create a comprehensive visual response that incorporates relevant information from the news results.
             """
             
             # async generator를 처리하기 위한 수정
@@ -656,7 +674,7 @@ def create_main_interface():
                 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")