Update app.py

app.py

@@ -2,27 +2,36 @@
 import gradio as gr
 from datasets import load_dataset
 
-# Загружаем только 500 случайных изображений (можно увеличить до 1000–2000)
-dataset = load_dataset("deepghs/sankaku_full", split="train[:500]")
+# Загружаем ТОЛЬКО метаданные, без изображений
+dataset = load_dataset(
+    "deepghs/sankaku_full",
+    split="train[:2000]",        # первые 2000 записей
+    streaming=False,
+    use_auth_token=False
+)
 
-def search_images(tag):
+# Преобразуем в список один раз (иначе streaming будет мешать)
+data_list = list(dataset)
+
+def search_by_tag(tag):
     results = []
-    for item in dataset:
+    for item in data_list:
         tags = item.get("tags", "")
         if tag.lower() in tags.lower():
-            # Добавляем миниатюру + теги
-            results.append((item["image"], f"Tags: {tags}"))
-            if len(results) >= 10:  # показываем максимум 10 результатов
+            rating = item.get("rating", "?")
+            # НЕ пытаемся показать image — его нет или он битый
+            results.append(f"ID: {item.get('id', 'N/A')} | Rating: {rating}\nTags: {tags[:100]}...")
+            if len(results) >= 10:
                 break
-    return results if results else [("https://via.placeholder.com/300?text=No+Results", "Ничего не найдено")]
+    return "\n\n".join(results) if results else "Ничего не найдено."
 
-with gr.Blocks(title="Sankaku Viewer") as demo:
-    gr.Markdown("# 🔞 Sankaku Full Dataset Viewer")
-    gr.Markdown("⚠️ **NSFW content. For adults only.**")
+with gr.Blocks() as demo:
+    gr.Markdown("### 🔍 Sankaku Dataset — Metadata Viewer")
+    gr.Markdown("⚠️ Изображения НЕ отображаются (недоступны). Только теги и ID.")
     
-    tag_input = gr.Textbox(label="Поиск по тегу (например: wolf, furry, animal ears)")
-    gallery = gr.Gallery(label="Результаты", columns=3, rows=4, height=600)
-
-    tag_input.change(fn=search_images, inputs=tag_input, outputs=gallery)
+    tag = gr.Textbox(label="Поиск по тегу (например: wolf, furry, nsfw)")
+    output = gr.Textbox(label="Результаты", lines=15)
+    
+    tag.change(fn=search_by_tag, inputs=tag, outputs=output)
 
 demo.launch()