Update app.py

app.py

@@ -456,17 +456,48 @@ def search_serphouse(query, country, page=1, num_result=10):
     }
 
     try:
+        # 세션 설정 개선
         session = requests.Session()
-        retries = Retry(total=3, backoff_factor=0.5)
-        session.mount('https://', HTTPAdapter(max_retries=retries))
         
-        response = session.post(url, json=payload, headers=headers, timeout=(5, 15))
+        # 재시도 설정 강화
+        retries = Retry(
+            total=5,  # 최대 재시도 횟수 증가
+            backoff_factor=1,  # 재시도 간격 증가
+            status_forcelist=[500, 502, 503, 504, 429],  # 재시도할 HTTP 상태 코드
+            allowed_methods=["POST"]  # POST 요청에 대한 재시도 허용
+        )
+        
+        # 타임아웃 설정 조정
+        adapter = HTTPAdapter(max_retries=retries)
+        session.mount('http://', adapter)
+        session.mount('https://', adapter)
+        
+        # 타임아웃 값 증가 (connect timeout, read timeout)
+        response = session.post(
+            url, 
+            json=payload, 
+            headers=headers, 
+            timeout=(30, 30)  # 연결 타임아웃 30초, 읽기 타임아웃 30초
+        )
+        
         response.raise_for_status()
         return {"results": response.json(), "translated_query": translated_query}
-    except requests.RequestException as e:
-        return {"error": f"Error: {str(e)}", "translated_query": query}
-
-
+        
+    except requests.exceptions.Timeout:
+        return {
+            "error": "검색 시간이 초과되었습니다. 잠시 후 다시 시도해주세요.",
+            "translated_query": query
+        }
+    except requests.exceptions.RequestException as e:
+        return {
+            "error": f"검색 중 오류가 발생했습니다: {str(e)}",
+            "translated_query": query
+        }
+    except Exception as e:
+        return {
+            "error": f"예기치 않은 오류가 발생했습니다: {str(e)}",
+            "translated_query": query
+        }
 
 def format_results_from_raw(response_data):
     if "error" in response_data:
@@ -995,7 +1026,7 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, title="NewsAI 서비스") as
 
     
     # AI 번역 탭 추가
-        with gr.Tab("AI 번역"):
+        with gr.Tab("AI 기사 생성"):
             gr.Markdown("뉴스 URL을 입력하면 AI가 한국어로 번역하여 기사 형식으로 작성합니다.")
         
             with gr.Column():