Example_Loader.py

import torch
import torch.nn as nn
import numpy as np
import random
import os
import io
import http.server
import socketserver
from PIL import Image, ImageDraw, ImageFont
import json

# ============================================================
# 1. НЕЙРОСЕТЬ
# ============================================================

class AndreyNN(nn.Module):
    def __init__(self, vocab_size):
        super().__init__()
        self.fc1 = nn.Linear(vocab_size, 128)
        self.fc2 = nn.Linear(128, 128)
        self.fc3 = nn.Linear(128, 64)
        self.fc4 = nn.Linear(64, 32)
        self.fc5 = nn.Linear(32, vocab_size)
        self.relu = nn.ReLU()
        self.dropout = nn.Dropout(0.2)
        self.softmax = nn.Softmax(dim=1)
    
    def forward(self, x):
        x = self.relu(self.fc1(x))
        x = self.dropout(x)
        x = self.relu(self.fc2(x))
        x = self.dropout(x)
        x = self.relu(self.fc3(x))
        x = self.dropout(x)
        x = self.relu(self.fc4(x))
        x = self.softmax(self.fc5(x))
        return x

# ============================================================
# 2. ЗАГРУЗЧИК
# ============================================================

class AndreyWorldLoader:
    def __init__(self, model_path='andrey_world_model.pt', device='cpu'):
        self.device = torch.device(device)
        
        if not os.path.exists(model_path):
            raise FileNotFoundError(f"Файл {model_path} не найден!")
        
        checkpoint = torch.load(model_path, map_location='cpu')
        
        self.vocab = checkpoint['vocab']
        self.word_to_idx = checkpoint['word_to_idx']
        self.idx_to_word = checkpoint['idx_to_word']
        self.vocab_size = checkpoint['vocab_size']
        self.is_trained = checkpoint['is_trained']
        
        self.model = AndreyNN(self.vocab_size).to(self.device)
        self.model.load_state_dict(checkpoint['model_state'])
        self.model.eval()
        
        self.themes = ['свет', 'тьма', 'вода', 'огонь', 'ветер', 'земля']
        self.colors = ['золотой', 'синий', 'зелёный', 'розовый', 'фиолетовый', 'голубой']
        self.creatures = ['существа', 'птицы', 'змеи', 'великаны', 'духи', 'кони']
        self.places = ['долина', 'гора', 'лес', 'пустыня', 'океан', 'небо']
        
        print(f"[ГОТОВО] Модель загружена! Словарь: {self.vocab_size} слов")
    
    def generate_word(self, context=None):
        x = np.zeros(self.vocab_size)
        if context:
            for word in context:
                if word in self.word_to_idx:
                    x[self.word_to_idx[word]] = 1
        
        if np.sum(x) == 0:
            x[np.random.randint(0, self.vocab_size)] = 1
        
        x_tensor = torch.tensor(x, dtype=torch.float32).to(self.device).unsqueeze(0)
        
        with torch.no_grad():
            output = self.model(x_tensor)
            probs = output.cpu().numpy()[0]
            probs = probs / np.sum(probs)
            idx = np.random.choice(self.vocab_size, p=probs)
            return self.idx_to_word[idx]
    
    def generate_world(self):
        theme = self.generate_word(['мир', 'из'])
        color = self.generate_word(['цвет', 'как'])
        creature = self.generate_word(['существо', 'живёт'])
        place = self.generate_word(['место', 'где'])
        
        if theme not in self.themes:
            theme = random.choice(self.themes)
        if color not in self.colors:
            color = random.choice(self.colors)
        if creature not in self.creatures:
            creature = random.choice(self.creatures)
        if place not in self.places:
            place = random.choice(self.places)
        
        prefixes = ["Великий", "Тихий", "Золотой", "Древний", "Светлый", "Тёмный"]
        name = f"{random.choice(prefixes)} {place.capitalize()}"
        
        desc_parts = [
            f"Мир, где всё состоит из {theme}.",
            f"Здесь {creature} охраняют {place}.",
            f"{color.capitalize()} {place} наполнен {theme}.",
            f"{creature} танцуют в {place}."
        ]
        description = ' '.join(random.sample(desc_parts, 2))
        
        return {
            'name': name,
            'description': description,
            'theme': theme,
            'color': color,
            'creature': creature,
            'place': place
        }

# ============================================================
# 3. ГЕНЕРАЦИЯ КАРТИНКИ МИРА
# ============================================================

def generate_world_image(world_data, size=(640, 480)):
    img = Image.new('RGB', size, color=(20, 20, 40))
    draw = ImageDraw.Draw(img)
    
    colors = {
        'свет': (255, 240, 200),
        'тьма': (30, 30, 60),
        'вода': (0, 100, 200),
        'огонь': (255, 100, 50),
        'ветер': (180, 200, 220),
        'земля': (120, 80, 40)
    }
    
    base_color = colors.get(world_data.get('theme', 'свет'), (50, 50, 80))
    
    for y in range(size[1]):
        r = int(base_color[0] * (0.5 + 0.5 * y / size[1]))
        g = int(base_color[1] * (0.5 + 0.5 * y / size[1]))
        b = int(base_color[2] * (0.5 + 0.5 * y / size[1]))
        draw.line([(0, y), (size[0], y)], fill=(r, g, b))
    
    ground_y = int(size[1] * 0.6)
    draw.rectangle([(0, ground_y), (size[0], size[1])], fill=(60, 40, 20))
    
    for _ in range(30):
        x = random.randint(0, size[0])
        y = random.randint(ground_y, size[1])
        h = random.randint(5, 15)
        draw.line([(x, y), (x + random.randint(-3, 3), y - h)], fill=(50, 150, 50), width=1)
    
    for _ in range(random.randint(3, 5)):
        cx = random.randint(100, size[0] - 100)
        cy = ground_y - random.randint(30, 100)
        r = random.randint(60, 150)
        draw.ellipse([(cx - r, cy - r), (cx + r, cy + r)], fill=(100, 80, 60))
    
    sx = random.randint(50, 200)
    sy = random.randint(30, 100)
    draw.ellipse([(sx - 30, sy - 30), (sx + 30, sy + 30)], fill=(255, 200, 100))
    
    if world_data.get('theme') == 'тьма':
        for _ in range(50):
            sx2 = random.randint(0, size[0])
            sy2 = random.randint(0, ground_y)
            br = random.randint(1, 3)
            draw.ellipse([(sx2 - br, sy2 - br), (sx2 + br, sy2 + br)], fill=(255, 255, 200))
    
    for _ in range(random.randint(8, 15)):
        tx = random.randint(0, size[0])
        ty = ground_y + random.randint(0, 30)
        draw.rectangle([(tx - 3, ty - 20), (tx + 3, ty)], fill=(80, 50, 30))
        cr = random.randint(15, 30)
        draw.ellipse([(tx - cr, ty - 20 - cr), (tx + cr, ty - 20 + cr)], fill=(40, 140, 40))
    
    for _ in range(random.randint(3, 6)):
        ox = random.randint(0, size[0])
        oy = random.randint(20, ground_y - 50)
        ow = random.randint(60, 120)
        oh = random.randint(20, 40)
        draw.ellipse([(ox, oy), (ox + ow, oy + oh)], fill=(200, 200, 220, 100))
    
    try:
        font = ImageFont.load_default()
        draw.text((20, 20), f"🌍 {world_data.get('name', '')}", fill=(255, 255, 255), font=font)
        draw.text((20, 50), f"🏷️ {world_data.get('theme', '')} | {world_data.get('color', '')}", fill=(200, 200, 200), font=font)
        draw.text((20, 80), f"🦄 {world_data.get('description', '')[:50]}...", fill=(180, 180, 180), font=font)
    except:
        pass
    
    draw.rectangle([(0, 0), (size[0]-1, size[1]-1)], outline=(100, 100, 150), width=2)
    
    return img

# ============================================================
# 4. HTTP-СЕРВЕР
# ============================================================

# Загружаем модель
loader = AndreyWorldLoader('andrey_world_model.pt')

class WorldHandler(http.server.BaseHTTPRequestHandler):
    def log_message(self, format, *args):
        pass
    
    def do_GET(self):
        if self.path == '/':
            # Генерируем мир
            world = loader.generate_world()
            img = generate_world_image(world)
            
            # Сохраняем в буфер
            buffer = io.BytesIO()
            img.save(buffer, format='PNG')
            img_data = buffer.getvalue()
            
            # Страница
            html = f'''
            <!DOCTYPE html>
            <html>
            <head>
                <title>AndreyWorld</title>
                <meta http-equiv="refresh" content="10">
                <style>
                    body {{
                        background: #0d0d1a;
                        color: #e0e0e0;
                        font-family: 'Segoe UI', sans-serif;
                        display: flex;
                        justify-content: center;
                        align-items: center;
                        height: 100vh;
                        margin: 0;
                        padding: 20px;
                    }}
                    .container {{
                        max-width: 900px;
                        width: 100%;
                        background: #1a1a2e;
                        padding: 30px;
                        border-radius: 20px;
                        border: 1px solid #2a2a4e;
                        box-shadow: 0 0 50px rgba(0,0,0,0.8);
                        text-align: center;
                    }}
                    h1 {{
                        color: #6a8fbd;
                        margin-top: 0;
                    }}
                    img {{
                        width: 100%;
                        border-radius: 15px;
                        border: 2px solid #2a2a4e;
                        margin: 15px 0;
                    }}
                    .info {{
                        background: #0d0d1a;
                        padding: 15px;
                        border-radius: 10px;
                        margin: 10px 0;
                    }}
                    .tag {{
                        display: inline-block;
                        background: #2a2a4e;
                        padding: 4px 12px;
                        border-radius: 12px;
                        margin: 3px;
                        font-size: 0.9em;
                    }}
                    .meta {{
                        color: #666;
                        font-size: 0.85em;
                        margin-top: 15px;
                    }}
                    .refresh {{
                        color: #4a6fa5;
                        text-decoration: none;
                    }}
                </style>
            </head>
            <body>
                <div class="container">
                    <h1>🌍 AndreyWorld</h1>
                    <div class="info">
                        <h2>🌍 {world.get('name', '')}</h2>
                        <p>{world.get('description', '')}</p>
                        <div>
                            <span class="tag">🏷️ {world.get('theme', '')}</span>
                            <span class="tag">🎨 {world.get('color', '')}</span>
                            <span class="tag">🦄 {world.get('creature', '')}</span>
                            <span class="tag">📍 {world.get('place', '')}</span>
                        </div>
                    </div>
                    <img src="/world.png" alt="AndreyWorld">
                    <div class="meta">
                        🔄 Страница обновляется каждые 10 секунд<br>
                        📡 <a href="/" class="refresh">Обновить сейчас</a>
                    </div>
                </div>
            </body>
            </html>
            '''
            
            self.send_response(200)
            self.send_header('Content-Type', 'text/html; charset=utf-8')
            self.end_headers()
            self.wfile.write(html.encode('utf-8'))
        
        elif self.path == '/world.png':
            # Генерируем новый мир и отдаём картинку
            world = loader.generate_world()
            img = generate_world_image(world)
            
            buffer = io.BytesIO()
            img.save(buffer, format='PNG')
            img_data = buffer.getvalue()
            
            self.send_response(200)
            self.send_header('Content-Type', 'image/png')
            self.end_headers()
            self.wfile.write(img_data)
        
        else:
            self.send_response(404)
            self.end_headers()
            self.wfile.write(b'Not Found')

# ============================================================
# 5. ЗАПУСК
# ============================================================

def main():
    PORT = 6767
    
    print("\n" + "="*50)
    print("🌍 ANDREYWORLD")
    print("   http://localhost:" + str(PORT))
    print("="*50 + "\n")
    
    with socketserver.TCPServer(("", PORT), WorldHandler) as httpd:
        print(f"[ГОТОВО] Сервер запущен на порту {PORT}")
        print(f"[ССЫЛКА] Открой http://localhost:{PORT}")
        print("[ИНФО] Страница обновляется каждые 10 секунд\n")
        try:
            httpd.serve_forever()
        except KeyboardInterrupt:
            print("\n[СТОП] Остановка...")
            httpd.shutdown()

if __name__ == "__main__":
    main()