Update app.py

app.py

@@ -2,97 +2,89 @@ import gradio as gr
 from huggingface_hub import InferenceClient
 from PIL import Image
 import os
-import io
+import random
 
-# Берем токен из секретов
+# Настройка клиента
 hf_token = os.getenv("HF_TOKEN")
-# Используем мощную модель SDXL
 client = InferenceClient("stabilityai/stable-diffusion-xl-base-1.0", token=hf_token)
 
-def generate_core(prompt, resolution, pixelate):
-    """Основная функция генерации"""
-    if not hf_token:
-        return None
+def forge_texture(prompt, resolution, pixelate, style, seed):
+    if not hf_token: return None, None
     
-    # Промпт заточен под Minecraft
-    full_prompt = f"minecraft asset, {prompt}, 2d flat game texture, simple background, high contrast, sharp edges"
-    neg_prompt = "shading, realistic, 3d, shadow, blurry, text, watermark, photo, messy"
+    # Улучшение промпта в зависимости от стиля
+    styles = {
+        "Cursed 🔮": "dark occult aesthetic, purple and black glow, twisted textures, cursed energy",
+        "Classic ⛏️": "minecraft vanilla style, flat colors, blocky look, simplistic",
+        "HD Realism ✨": "highly detailed, 8k, realistic material, sharp shadows"
+    }
+    
+    # Если seed -1, выбираем случайный
+    used_seed = random.randint(0, 1000000) if seed == -1 else seed
+    
+    full_prompt = f"minecraft 2d texture, {prompt}, {styles[style]}, flat game asset, white background"
+    neg_prompt = "lowres, text, error, cropped, worst quality, low quality, 3d render, perspective, messy"
     
     try:
-        # Сама генерация
-        image = client.text_to_image(full_prompt, negative_prompt=neg_prompt, guidance_scale=8.0)
+        image = client.text_to_image(
+            full_prompt, 
+            negative_prompt=neg_prompt, 
+            guidance_scale=8.0,
+            seed=used_seed
+        )
         
-        # Разрешение
+        # Ресайз
         res_map = {"16x16": 16, "32x32": 32, "64x64": 64, "128x128": 128, "512x512": 512}
         size = res_map[resolution]
+        image = image.resize((size, size), Image.Resampling.NEAREST if pixelate else Image.Resampling.LANCZOS)
+        
+        # Сохраняем как PNG
+        file_name = f"texture_{random.randint(1000, 9999)}.png"
+        image.save(file_name, "PNG")
         
-        if pixelate:
-            image = image.resize((size, size), Image.Resampling.NEAREST)
-        else:
-            image = image.resize((size, size), Image.Resampling.LANCZOS)
-            
-        return image
+        return image, file_name
     except Exception as e:
         print(f"Error: {e}")
-        return None
+        return None, None
 
-def forge_single(prompt, resolution, pixelate):
-    img = generate_core(prompt, resolution, pixelate)
-    if img is None:
-        # Создаем картинку-ошибку, если API не ответил
-        err = Image.new('RGB', (512, 512), color=(40, 0, 0))
-        return err
-    return img
+def forge_set(prompt, resolution, pixelate, style):
+    # Генератор для сета (блок + предмет)
+    block_img, block_file = forge_texture(f"block of {prompt}", resolution, pixelate, style, -1)
+    item_img, item_file = forge_texture(f"item {prompt} ingot or tool", resolution, pixelate, style, -1)
+    return block_img, block_file, item_img, item_file
 
-def forge_set(prompt, resolution, pixelate):
-    """Генерирует сразу блок и предмет"""
-    block = generate_core(f"block of {prompt} texture", resolution, pixelate)
-    item = generate_core(f"item icon of {prompt} sword or tool", resolution, pixelate)
-    
-    if block is None: block = Image.new('RGB', (512, 512), color=(20, 20, 20))
-    if item is None: item = Image.new('RGB', (512, 512), color=(20, 20, 20))
-    
-    return block, item
-
-# Красивый темный интерфейс
-css = """
-.gradio-container { background-color: #0a0a0a !important; color: #d8b4fe !important; }
-.btn-primary { background: linear-gradient(to right, #7c3aed, #db2777) !important; border: none !important; }
-footer { display: none !important; }
-"""
+# UI
+css = ".gradio-container {background: #050505; color: #a78bfa;} .main-btn {background: linear-gradient(90deg, #4c1d95, #7c3aed) !important;}"
 
 with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
-    gr.Markdown("# 🌌 The Cursed: Texture Forge v3.0")
+    gr.Markdown("# 🔮 The Cursed: Texture Forge v4.0")
     
-    if not hf_token:
-        gr.HTML("<div style='background: red; padding: 10px; color: white;'>⚠️ ОШИБКА: HF_TOKEN не найден! Добавьте его в Settings -> Secrets</div>")
-
     with gr.Tabs():
         with gr.TabItem("Одиночная текстура"):
             with gr.Row():
                 with gr.Column():
-                    p_in = gr.Textbox(label="Описание (English)", placeholder="e.g. cursed soul sand with screaming faces")
-                    r_in = gr.Dropdown(["16x16", "32x32", "64x64", "128x128", "512x512"], value="128x128", label="Разрешение")
+                    p_in = gr.Textbox(label="Описание", placeholder="dark obsidian with eyes")
+                    with gr.Row():
+                        r_in = gr.Dropdown(["16x16", "32x32", "64x64", "128x128", "512x512"], value="128x128", label="Разрешение")
+                        s_in = gr.Dropdown(["Cursed 🔮", "Classic ⛏️", "HD Realism ✨"], value="Cursed 🔮", label="Стиль")
                     px_in = gr.Checkbox(label="Pixel Art Mode", value=True)
-                    btn_p = gr.Button("🔥 ВЫКОВАТЬ", variant="primary")
+                    seed_in = gr.Number(label="Seed (-1 для случайного)", value=-1)
+                    btn_p = gr.Button("🔥 ВЫКОВАТЬ", variant="primary", elem_classes="main-btn")
                 with gr.Column():
-                    out_p = gr.Image(label="Результат", type="pil")
-                    gr.Markdown("📥 *Чтобы скачать в .png: Нажми на иконку загрузки в углу картинки или ПКМ -> Сохранить как.*")
+                    out_img = gr.Image(label="Предпросмотр", type="pil")
+                    out_file = gr.File(label="Скачать .PNG для Minecraft")
 
-        with gr.TabItem("📦 ГЕНЕРАЦИЯ СЕТА"):
-            gr.Markdown("Создает одновременно текстуру блока и иконку предмета в едином стиле.")
+        with gr.TabItem("📦 Генерация сета"):
             with gr.Row():
                 with gr.Column():
-                    s_in = gr.Textbox(label="Тема сета", placeholder="e.g. void amethyst crystal")
-                    s_res = gr.Dropdown(["16x16", "32x32", "64x64", "128x128", "512x512"], value="128x128", label="Разрешение")
-                    s_px = gr.Checkbox(label="Pixel Art Mode", value=True)
+                    set_p = gr.Textbox(label="Тема сета", placeholder="void crystal")
                     btn_s = gr.Button("⚒️ СОЗДАТЬ СЕТ", variant="primary")
                 with gr.Column():
                     with gr.Row():
-                        out_s_block = gr.Image(label="Блок", type="pil")
-                        out_s_item = gr.Image(label="Предмет", type="pil")
+                        out_b = gr.Image(label="Блок"); out_i = gr.Image(label="Предмет")
+                    with gr.Row():
+                        out_b_f = gr.File(label="PNG Блока"); out_i_f = gr.File(label="PNG Предмета")
 
-    btn_p.click(forge_single, [p_in, r_in, px_in], out_p)
-    btn_s.click(forge_set, [s_in, s_res, s_px], [out_s_block, out_s_item])
+    btn_p.click(forge_texture, [p_in, r_in, px_in, s_in, seed_in], [out_img, out_file])
+    btn_s.click(forge_set, [set_p, r_in, px_in, s_in], [out_b, out_b_f, out_i, out_i_f])
 
 demo.launch()