Update app.py

app.py

@@ -9,6 +9,7 @@ import markdown
 import tempfile
 import base64
 from datetime import datetime
+import re
 
 # 로깅 설정
 logging.basicConfig(
@@ -151,12 +152,52 @@ def extract_keywords(text: str, top_k: int = 5) -> str:
     2) 공백 기준 토큰 분리
     3) 최대 top_k개만
     """
-    import re
     text = re.sub(r"[^a-zA-Z0-9가-힣\s]", "", text)
     tokens = text.split()
     key_tokens = tokens[:top_k]
     return " ".join(key_tokens)
 
+# Mock 검색 결과 생성 함수 추가 (API 키가 없거나 연결 실패 시 대체 사용)
+def generate_mock_search_results(query):
+    """API 연결이 안될 때 사용할 가상 검색 결과 생성"""
+    current_time = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+    mock_results = [
+        {
+            "title": f"{query}에 관한 최신 정보",
+            "link": "https://example.com/article1",
+            "snippet": f"{query}에 관한 가상 검색 결과입니다. 이 결과는 API 연결 문제로 인해 생성된 가상 데이터입니다. 실제 검색 결과가 아님을 참고하세요. 생성 시간: {current_time}",
+            "displayed_link": "example.com/article1"
+        },
+        {
+            "title": f"{query} 관련 연구 동향",
+            "link": "https://example.org/research",
+            "snippet": "이것은 API 연결 문제로 인한 가상 검색 결과입니다. 실제 검색 결과를 보여드리지 못해 죄송합니다. 대신 AI의 기존 지식을 활용하여 답변드리겠습니다.",
+            "displayed_link": "example.org/research"
+        }
+    ]
+    
+    summary_lines = []
+    for idx, item in enumerate(mock_results, start=1):
+        title = item.get("title", "No title")
+        link = item.get("link", "#")
+        snippet = item.get("snippet", "No description")
+        displayed_link = item.get("displayed_link", link)
+        
+        summary_lines.append(
+            f"### Result {idx}: {title}\n\n"
+            f"{snippet}\n\n"
+            f"**출처**: [{displayed_link}]({link})\n\n"
+            f"---\n"
+        )
+    
+    notice = """
+# 가상 검색 결과 (API 연결 문제로 인해 생성됨)
+아래는 API 연결 문제로 인해 생성된 가상 검색 결과입니다. 실제 검색 결과가 아님을 참고하세요.
+대신 AI의 기존 지식을 활용하여 최대한 정확한 답변을 드리겠습니다.
+"""
+    
+    return notice + "\n".join(summary_lines)
+
 # 웹 검색 함수
 def do_web_search(query: str) -> str:
     """
@@ -171,9 +212,9 @@ def do_web_search(query: str) -> str:
             "q": query,
             "domain": "google.com",
             "serp_type": "web",  # 기본 웹 검색
-            "device": "desktop",
-            "lang": "en",
-            "num": "20"  # 최대 20개 결과만 요청
+            "device": "desktop", 
+            "lang": "ko",  # 한국어 검색 결과를 위해 변경
+            "num": "10"  # 결과 수를 10개로 줄임 (빠른 응답을 위해)
         }
         
         headers = {
@@ -183,8 +224,13 @@ def do_web_search(query: str) -> str:
         logging.info(f"SerpHouse API 호출 중... 검색어: {query}")
         logging.info(f"요청 URL: {url} - 파라미터: {params}")
         
-        # GET 요청 수행
-        response = requests.get(url, headers=headers, params=params, timeout=60)
+        # Mock 응답 (실제 API 연결이 어려울 경우)
+        if not SERPHOUSE_API_KEY or "mock" in SERPHOUSE_API_KEY.lower():
+            logging.warning("API 키가 없거나 Mock 모드입니다. 모의 검색 결과를 반환합니다.")
+            return generate_mock_search_results(query)
+        
+        # 타임아웃 줄임 (30초)
+        response = requests.get(url, headers=headers, params=params, timeout=30)
         response.raise_for_status()
         
         logging.info(f"SerpHouse API 응답 상태 코드: {response.status_code}")
@@ -209,13 +255,13 @@ def do_web_search(query: str) -> str:
             
         if not organic:
             logging.warning("응답에서 organic 결과를 찾을 수 없습니다.")
-            logging.debug(f"응답 구조: {list(data.keys())}")
+            logging.debug(f"응답 구조: {list(data.keys()) if isinstance(data, dict) else 'not a dict'}")
             if isinstance(results, dict):
                 logging.debug(f"results 구조: {list(results.keys())}")
-            return "No web search results found or unexpected API response structure."
+            return "검색 결과를 찾을 수 없습니다. 대신 기존 지식을 활용하여 답변하겠습니다."
 
         # 결과 수 제한 및 컨텍스트 길이 최적화
-        max_results = min(20, len(organic))
+        max_results = min(10, len(organic))
         limited_organic = organic[:max_results]
         
         # 결과 형식 개선 - 마크다운 형식으로 출력하여 가독성 향상
@@ -248,9 +294,12 @@ def do_web_search(query: str) -> str:
         logging.info(f"검색 결과 {len(limited_organic)}개 처리 완료")
         return search_results
     
+    except requests.exceptions.Timeout:
+        logging.error("Web search timed out")
+        return "Web search timed out. 검색 시간이 초과되었습니다. 기존 지식을 활용하여 답변하겠습니다."
     except Exception as e:
         logging.error(f"Web search failed: {e}")
-        return f"Web search failed: {str(e)}"
+        return f"Web search failed: {str(e)}. 검색 중 오류가 발생했습니다. 기존 지식을 활용하여 답변하겠습니다."
 
 def chatbot_interface():
     st.title("Ginigen Blog")
@@ -320,7 +369,6 @@ def chatbot_interface():
                 latest_blog = msg["content"]
                 
                 # 타이틀 추출 시도 (첫 번째 제목 태그 사용)
-                import re
                 title_match = re.search(r'# (.*?)(\n|$)', latest_blog)
                 if title_match:
                     latest_blog_title = title_match.group(1).strip()
@@ -402,14 +450,20 @@ def chatbot_interface():
             system_prompt = get_system_prompt()
             if st.session_state.use_web_search:
                 with st.spinner("웹에서 관련 정보를 검색 중..."):
-                    search_query = extract_keywords(prompt, top_k=5)
-                    search_results = do_web_search(search_query)
-                    if "search failed" not in search_results.lower():
-                        # 시스템 프롬프트에 검색 결과 추가
-                        system_prompt += f"\n\n검색 결과:\n{search_results}\n"
-                        st.success(f"검색 완료: '{search_query}'에 대한 정보를 수집했습니다.")
-                    else:
-                        st.error("웹 검색 중 오류가 발생했습니다.")
+                    try:
+                        search_query = extract_keywords(prompt, top_k=5)
+                        search_results = do_web_search(search_query)
+                        
+                        if "search failed" not in search_results.lower() and "timed out" not in search_results.lower():
+                            # 시스템 프롬프트에 검색 결과 추가
+                            system_prompt += f"\n\n검색 결과:\n{search_results}\n"
+                            st.success(f"검색 완료: '{search_query}'에 대한 정보를 수집했습니다.")
+                        else:
+                            st.warning("웹 검색을 진행했으나 결과를 가져오는데 실패했습니다. 기존 지식을 활용하여 답변합니다.")
+                            logging.warning(f"웹 검색 실패 또는 타임아웃: {search_results}")
+                    except Exception as e:
+                        st.error(f"웹 검색 중 오류가 발생했습니다: {str(e)}")
+                        logging.error(f"웹 검색 오류: {str(e)}")
             
             # API 호출
             with client.messages.stream(