Update app.py

app.py

@@ -4,6 +4,7 @@
 # HF Spaces Secrets 설정:
 #   NAVER_CLIENT_ID     = 발급받은 Client ID
 #   NAVER_CLIENT_SECRET = 발급받은 Client Secret
+#   ALLOWED_IPS         = 123.123.123.123,2406:xxxx:xxxx::1234
 
 import os
 import html
@@ -23,9 +24,41 @@ def _get_env(name: str) -> str:
     return v
 
 
+# ✅ 허용 IP 목록 로드
+raw_ips = _get_env("ALLOWED_IPS")
+ALLOWED_IPS = [ip.strip() for ip in raw_ips.split(",") if ip.strip()]
+
+
+# ✅ 프록시 환경 고려: 실제 클라이언트 IP 추출
+def _get_client_ip(request: gr.Request) -> str:
+    xff = request.headers.get("x-forwarded-for")
+    if xff:
+        return xff.split(",")[0].strip()
+
+    xri = request.headers.get("x-real-ip")
+    if xri:
+        return xri.strip()
+
+    return request.client.host
+
+
+# ✅ IP 접근 차단
+def gatekeeper(request: gr.Request):
+    if not ALLOWED_IPS:
+        return gr.update(visible=True), ""
+
+    client_ip = _get_client_ip(request)
+
+    if client_ip not in ALLOWED_IPS:
+        return (
+            gr.update(visible=False),
+            f"⛔ 접근이 허용되지 않은 IP입니다. (현재 IP: {client_ip})",
+        )
+
+    return gr.update(visible=True), ""
+
+
 def _strip_tags(text: str) -> str:
-    # 네이버 뉴스 API 결과의 title/description은 <b>...</b>가 포함될 수 있어 제거
-    # (HTML 태그 제거 + 엔티티 처리)
     if not text:
         return ""
     text = re.sub(r"<[^>]+>", "", text)
@@ -33,13 +66,10 @@ def _strip_tags(text: str) -> str:
 
 
 def _format_pubdate(pub_date: str) -> str:
-    # 예: "Mon, 19 Jan 2026 09:30:00 +0900"
-    # 파싱 실패 시 원문 반환
     if not pub_date:
         return ""
     try:
         dt = datetime.strptime(pub_date, "%a, %d %b %Y %H:%M:%S %z")
-        # 한국 환경에서 보기 좋은 포맷
         return dt.strftime("%Y-%m-%d %H:%M:%S %z")
     except Exception:
         return pub_date
@@ -80,7 +110,6 @@ def naver_news_search(
         timeout=timeout,
     )
 
-    # 네이버 API가 에러를 반환하는 경우, body에 상세가 있을 수 있어 그대로 보여주기 위해 처리
     if r.status_code != 200:
         try:
             detail = r.json()
@@ -110,7 +139,6 @@ def render_results(data: Dict[str, Any], max_items: int = 10) -> str:
 
         lines.append(f"{i}. **{title}**")
 
-        # 하위 항목 4칸 들여쓰기
         if pub:
             lines.append(f"    - 발행: {pub}")
         if origin:
@@ -127,10 +155,6 @@ def render_results(data: Dict[str, Any], max_items: int = 10) -> str:
 
 
 def dedup_items(all_items: List[Dict[str, Any]]) -> List[Dict[str, Any]]:
-    """
-    결과 중복 제거:
-    - originallink 우선, 없으면 link, 없으면 title+pubDate 해시
-    """
     seen = set()
     out = []
     for it in all_items:
@@ -154,10 +178,7 @@ def aggregate_search(
     display: int,
     sort: str,
 ) -> Tuple[List[str], List[Dict[str, Any]], int]:
-    """
-    사용자 입력 문장을 그대로 query로 사용하여 API 호출
-    반환: (사용된 쿼리 목록, 최종 아이템 목록, total)
-    """
+
     queries = [sentence]
 
     all_items: List[Dict[str, Any]] = []
@@ -173,12 +194,8 @@ def aggregate_search(
 
 
 def render_results_from_items(items: List[Dict[str, Any]]) -> str:
-    """
-    items 리스트를 동일 스타일로 출력
-    """
     lines: List[str] = []
-    # lines.append(f"- 최종 반환 개수: {len(items)}건")
-    lines.append("")  # (요청: 수평선은 상단 문구 다음 줄에만 출력)
+    lines.append("")
 
     for i, it in enumerate(items, start=1):
         title = _strip_tags(it.get("title", ""))
@@ -218,17 +235,9 @@ def handle_search(
     try:
         _, items, total = aggregate_search(sentence=q, display=int(display), sort=sort)
 
-        total_to_show = total if total > 0 else len(items)
-
         lines: List[str] = []
-        # lines.append(
-        #     f"\"{q}\"에 대한 검색 결과는 **{total_to_show:,}건** 이며, 최종 응답 개수는 **{len(items)}건** 입니다."
-        # )
-        # lines.append("---")
-        # lines.append("")
-        # lines.append("---")  # ✅ 수평선은 “... 입니다” 다음 줄에만
         lines.append("")
-        lines.append(render_results_from_items(items))  # ✅ 결과는 1번만 출력
+        lines.append(render_results_from_items(items))
         lines.append("")
 
         assistant_text = "\n".join(lines).strip()
@@ -240,52 +249,64 @@ def handle_search(
     return chat_history, ""
 
 
-with gr.Blocks(title="Naver News Search (Chat UI)") as demo:          
-    with gr.Accordion("검색 옵셥 - 뉴스 개수와 정렬 방식을 선택하세요.", open=False):
-        with gr.Row():
-            display = gr.Slider(
-                minimum=1, maximum=100, value=30, step=1, label="뉴스 개수"
-            )
-            sort = gr.Dropdown(
-                choices=[("최신순", "date"), ("정확도순(연관도순)", "sim")],
-                value="date",
-                label="정렬 방식",
-            )
+with gr.Blocks(title="Naver News Search (Chat UI)") as demo:
+
+    warning = gr.Markdown("")
+    main_ui = gr.Column(visible=True)
 
-    chatbot = gr.Chatbot(
-        value=[],
-        label="NewsChat_v0.1",
-        type="messages",
-        height=600,
+    # ✅ 페이지 로드시 IP 검사
+    demo.load(
+        fn=gatekeeper,
+        inputs=None,
+        outputs=[main_ui, warning],
     )
 
-    with gr.Row():
-        query_in = gr.Textbox(
-            label="검색어를 입력하세요.",
-            placeholder="(예: 행안부의 인공지능 사업)",
-            scale=8,
+    with main_ui:
+        with gr.Accordion("검색 옵셥 - 뉴스 개수와 정렬 방식을 선택하세요.", open=False):
+            with gr.Row():
+                display = gr.Slider(
+                    minimum=1, maximum=100, value=30, step=1, label="뉴스 개수"
+                )
+                sort = gr.Dropdown(
+                    choices=[("최신순", "date"), ("정확도순(연관도순)", "sim")],
+                    value="date",
+                    label="정렬 방식",
+                )
+
+        chatbot = gr.Chatbot(
+            value=[],
+            label="NewsChat_v0.1",
+            type="messages",
+            height=600,
+        )
+
+        with gr.Row():
+            query_in = gr.Textbox(
+                label="검색어를 입력하세요.",
+                placeholder="(예: 행안부의 인공지능 사업)",
+                scale=8,
+            )
+            with gr.Column(scale=2):
+                search_btn = gr.Button("검색", variant="primary")
+                clear_btn = gr.Button("대화 지우기", variant="secondary")
+
+        search_btn.click(
+            fn=handle_search,
+            inputs=[query_in, chatbot, display, sort],
+            outputs=[chatbot, query_in],
+            api_name="search",
+        )
+        query_in.submit(
+            fn=handle_search,
+            inputs=[query_in, chatbot, display, sort],
+            outputs=[chatbot, query_in],
+            api_name="search_submit",
         )
-        with gr.Column(scale=2):
-            search_btn = gr.Button("검색", variant="primary")
-            clear_btn = gr.Button("대화 지우기", variant="secondary")
-
-    search_btn.click(
-        fn=handle_search,
-        inputs=[query_in, chatbot, display, sort],
-        outputs=[chatbot, query_in],
-        api_name="search",
-    )
-    query_in.submit(
-        fn=handle_search,
-        inputs=[query_in, chatbot, display, sort],
-        outputs=[chatbot, query_in],
-        api_name="search_submit",
-    )
 
-    def clear_chat():
-        return []
+        def clear_chat():
+            return []
 
-    clear_btn.click(fn=clear_chat, inputs=[], outputs=[chatbot], api_name="clear")
+        clear_btn.click(fn=clear_chat, inputs=[], outputs=[chatbot], api_name="clear")
 
 
 if __name__ == "__main__":