Update app.py

app.py

@@ -2,8 +2,14 @@ import gradio as gr
 from datasets import load_dataset
 from rapidfuzz import process, fuzz
 
-# 1) 데이터셋 스트리밍 로드(메타만 메모리에 보관)
-ds = load_dataset("nyuuzyou/clker-svg", split="train", streaming=True)
+# ──────────────────────────────────────────────────────────
+# 1) 데이터셋 로드 (스트리밍) ─ 메모리에 메타만 저장
+# ──────────────────────────────────────────────────────────
+ds = load_dataset(
+    "nyuuzyou/clker-svg",
+    split="train",
+    streaming=True,  # .jsonl.zst → 자동 스트리밍 해제
+)
 
 records = []
 for ex in ds:
@@ -17,50 +23,55 @@ for ex in ds:
         }
     )
 
+# ──────────────────────────────────────────────────────────
 # 2) 검색 함수
-def search_svg(query, top_k):
-    if not query:
+# ──────────────────────────────────────────────────────────
+def search_svg(query: str, top_k: int):
+    if not query.strip():
         return "⚠️ 검색어를 입력하세요.", None
 
-    # title+tags 스트링 하나로 결합
-    choices = {i: r["title"] + " " + r["tags"] for i, r in enumerate(records)}
+    # choices: index(int) ➜ title+tags 한 줄 문자열
+    choices = {i: f"{r['title']} {r['tags']}" for i, r in enumerate(records)}
+
+    # Rapidfuzz: (choice_text, score, key) 반환
     matched = process.extract(
         query,
         choices,
         scorer=fuzz.WRatio,
-        limit=top_k,
+        limit=int(top_k),
     )
 
-    # Rapidfuzz 결과 → SVG HTML 변환
     html_snippets = []
-    for idx, score, _ in matched:
+    for _choice_text, score, idx in matched:  # idx 가 실제 list 인덱스
         r = records[idx]
         svg_html = (
-            f"<div style='border:1px solid #ddd;margin:8px;padding:8px'>"
+            "<div style='border:1px solid #ddd;margin:8px;padding:8px'>"
             f"<strong>{r['title']}</strong> "
             f"(score {score})<br>"
             f"<em>{r['tags']}</em><br>"
             f"<a href='{r['url']}' target='_blank'>원본&nbsp;다운로드</a><br>"
-            f"{r['svg']}"  # raw SVG → 브라우저가 바로 렌더
-            f"</div>"
+            f"{r['svg']}"  # 브라우저가 즉시 SVG 렌더
+            "</div>"
         )
         html_snippets.append(svg_html)
 
     return "", "\n".join(html_snippets)
 
-
+# ──────────────────────────────────────────────────────────
+# 3) Gradio UI
+# ──────────────────────────────────────────────────────────
 with gr.Blocks(title="Clker SVG 검색") as demo:
     gr.Markdown("## 🔍 Clker.com SVG Public-Domain Clipart 검색")
     with gr.Row():
-        query = gr.Textbox(label="검색어", placeholder="예: cat, tree, …")
-        top_k = gr.Slider(1, 50, value=10, step=1, label="결과 개수")
-    warn = gr.Markdown()
-    out = gr.HTML()
+        query_box = gr.Textbox(label="검색어", placeholder="예: cat, tree, ...")
+        top_slider = gr.Slider(1, 50, value=10, step=1, label="결과 개수")
+    warning_md = gr.Markdown()
+    output_html = gr.HTML()
 
-    query.submit(
+    query_box.submit(
         fn=search_svg,
-        inputs=[query, top_k],
-        outputs=[warn, out],
+        inputs=[query_box, top_slider],
+        outputs=[warning_md, output_html],
     )
 
 if __name__ == "__main__":