Spaces:

XBFDI
/

neuron

Runtime error

App Files Files Community

X commited on Jun 26

Commit

b83bc51

verified ·

1 Parent(s): 9ddaadb

Create app.py

Browse files

Files changed (1) hide show

app.py +692 -0

app.py ADDED Viewed

	@@ -0,0 +1,692 @@

+import json
+import uuid
+import os
+import subprocess
+import threading
+import time
+import socket
+from pathlib import Path
+from fastapi import FastAPI, Request, Form
+from fastapi.responses import HTMLResponse, JSONResponse
+from fastapi.staticfiles import StaticFiles
+from fastapi.templating import Jinja2Templates
+import uvicorn
+# ========== КОНФИГ ==========
+PORT = int(os.getenv("PORT", 7860))
+DATA_DIR = Path("/data")
+DATA_DIR.mkdir(exist_ok=True)
+INVITES_FILE = DATA_DIR / "invites.json"
+PEERS_FILE = DATA_DIR / "peers.json"
+# Создаем файлы если их нет
+if not INVITES_FILE.exists():
+    with open(INVITES_FILE, "w") as f:
+        json.dump({}, f)
+if not PEERS_FILE.exists():
+    with open(PEERS_FILE, "w") as f:
+        json.dump({}, f)
+# ========== РАБОТА С ДАННЫМИ ==========
+def load_invites():
+    with open(INVITES_FILE, "r") as f:
+        return json.load(f)
+def save_invites(invites):
+    with open(INVITES_FILE, "w") as f:
+        json.dump(invites, f, indent=2)
+def load_peers():
+    with open(PEERS_FILE, "r") as f:
+        return json.load(f)
+def save_peers(peers):
+    with open(PEERS_FILE, "w") as f:
+        json.dump(peers, f, indent=2)
+# ========== ЗАПУСК PEERJS SERVER ==========
+def start_peerjs():
+    """Запускает PeerJS Server в фоновом процессе"""
+    try:
+        # Проверяем, запущен ли уже PeerJS
+        sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+        result = sock.connect_ex(('127.0.0.1', 9000))
+        sock.close()
+        if result != 0:
+            # Порт 9000 свободен - запускаем PeerJS
+            print("🚀 Запускаем PeerJS Server...")
+            subprocess.Popen(
+                ["peer", "--port", "9000", "--path", "/peerjs"],
+                stdout=subprocess.DEVNULL,
+                stderr=subprocess.DEVNULL,
+                start_new_session=True
+            )
+            time.sleep(2)  # Даем время запуститься
+            print("✅ PeerJS Server запущен на порту 9000")
+    except Exception as e:
+        print(f"⚠️ Ошибка запуска PeerJS: {e}")
+# Запускаем PeerJS в фоне (при старте приложения)
+threading.Thread(target=start_peerjs, daemon=True).start()
+# ========== FASTAPI ==========
+app = FastAPI()
+# ========== HTML СТРАНИЦА ==========
+@app.get("/", response_class=HTMLResponse)
+async def index(request: Request):
+    html = """
+    <!DOCTYPE html>
+    <html>
+    <head>
+        <meta charset="UTF-8">
+        <meta name="viewport" content="width=device-width, initial-scale=1.0">
+        <title>💬 HF Message</title>
+        <script src="https://unpkg.com/peerjs@1.5.1/dist/peerjs.min.js"></script>
+        <style>
+            * {
+                margin: 0;
+                padding: 0;
+                box-sizing: border-box;
+            }
+            body {
+                font-family: 'Segoe UI', Tahoma, Geneva, Verdana, sans-serif;
+                background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
+                min-height: 100vh;
+                display: flex;
+                justify-content: center;
+                align-items: center;
+                padding: 20px;
+            }
+            .container {
+                background: white;
+                border-radius: 20px;
+                box-shadow: 0 20px 60px rgba(0,0,0,0.3);
+                max-width: 600px;
+                width: 100%;
+                padding: 30px;
+            }
+            h1 {
+                color: #333;
+                text-align: center;
+                margin-bottom: 5px;
+                font-size: 28px;
+            }
+            .subtitle {
+                text-align: center;
+                color: #6c757d;
+                font-size: 14px;
+                margin-bottom: 20px;
+            }
+            .section {
+                background: #f8f9fa;
+                border-radius: 12px;
+                padding: 20px;
+                margin-bottom: 20px;
+            }
+            .section h3 {
+                color: #495057;
+                margin-bottom: 15px;
+                font-size: 16px;
+            }
+            input {
+                width: 100%;
+                padding: 12px;
+                border: 2px solid #dee2e6;
+                border-radius: 8px;
+                font-size: 14px;
+                margin-bottom: 10px;
+                transition: border-color 0.3s;
+            }
+            input:focus {
+                outline: none;
+                border-color: #667eea;
+            }
+            button {
+                width: 100%;
+                padding: 12px;
+                background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
+                color: white;
+                border: none;
+                border-radius: 8px;
+                font-size: 16px;
+                font-weight: 600;
+                cursor: pointer;
+                transition: transform 0.2s, box-shadow 0.2s;
+            }
+            button:hover {
+                transform: translateY(-2px);
+                box-shadow: 0 5px 20px rgba(102, 126, 234, 0.4);
+            }
+            button:active {
+                transform: translateY(0);
+            }
+            .code-display {
+                background: #2d3748;
+                color: #f7fafc;
+                padding: 15px;
+                border-radius: 8px;
+                font-family: 'Courier New', monospace;
+                font-size: 24px;
+                text-align: center;
+                letter-spacing: 3px;
+                word-break: break-all;
+                margin: 10px 0;
+            }
+            .status {
+                padding: 10px;
+                border-radius: 8px;
+                margin-top: 10px;
+                font-size: 14px;
+            }
+            .status.success {
+                background: #d4edda;
+                color: #155724;
+            }
+            .status.error {
+                background: #f8d7da;
+                color: #721c24;
+            }
+            .status.info {
+                background: #d1ecf1;
+                color: #0c5460;
+            }
+            .chat-box {
+                border: 2px solid #dee2e6;
+                border-radius: 8px;
+                height: 300px;
+                overflow-y: auto;
+                padding: 15px;
+                background: white;
+                margin-bottom: 10px;
+                display: none;
+            }
+            .chat-box.active {
+                display: block;
+            }
+            .message {
+                margin-bottom: 8px;
+                padding: 8px 12px;
+                border-radius: 8px;
+                max-width: 80%;
+                word-wrap: break-word;
+            }
+            .message.sent {
+                background: #667eea;
+                color: white;
+                margin-left: auto;
+            }
+            .message.received {
+                background: #e9ecef;
+                color: #333;
+                margin-right: auto;
+            }
+            .message.system {
+                background: #fff3cd;
+                color: #856404;
+                text-align: center;
+                font-style: italic;
+                max-width: 100%;
+            }
+            .chat-input {
+                display: none;
+                gap: 10px;
+            }
+            .chat-input.active {
+                display: flex;
+            }
+            .chat-input input {
+                flex: 1;
+                margin-bottom: 0;
+            }
+            .chat-input button {
+                width: auto;
+                padding: 12px 24px;
+                white-space: nowrap;
+            }
+            .peer-id {
+                font-size: 12px;
+                color: #6c757d;
+                text-align: center;
+                margin-top: 10px;
+                word-break: break-all;
+            }
+            .hidden {
+                display: none;
+            }
+            .btn-copy {
+                background: #28a745 !important;
+                margin-top: 10px;
+            }
+            .btn-copy:hover {
+                background: #218838 !important;
+                box-shadow: 0 5px 20px rgba(40, 167, 69, 0.4) !important;
+            }
+            .btn-danger {
+                background: #dc3545 !important;
+                margin-top: 10px;
+            }
+            .btn-danger:hover {
+                background: #c82333 !important;
+                box-shadow: 0 5px 20px rgba(220, 53, 69, 0.4) !important;
+            }
+            .stats {
+                text-align: center;
+                font-size: 12px;
+                color: #6c757d;
+                margin-top: 15px;
+                padding-top: 15px;
+                border-top: 1px solid #dee2e6;
+            }
+            .stats span {
+                font-weight: 600;
+                color: #333;
+            }
+            @media (max-width: 500px) {
+                .container {
+                    padding: 20px;
+                }
+                .code-display {
+                    font-size: 18px;
+                    letter-spacing: 2px;
+                }
+            }
+        </style>
+    </head>
+    <body>
+        <div class="container">
+            <h1>💬 HF Message</h1>
+            <div class="subtitle">Peer‑to‑Peer чат с инвайт‑кодами</div>
+            <!-- Секция 1: Инвайт-код -->
+            <div class="section" id="invite-section">
+                <h3>🎫 Создать инвайт‑код</h3>
+                <button onclick="generateInvite()">Сгенерировать код</button>
+                <div id="invite-result" class="hidden">
+                    <div class="code-display" id="invite-code"></div>
+                    <p style="font-size:12px;color:#6c757d;text-align:center;margin:5px 0;">
+                        Код сохраняется в /data и не пропадает после перезапуска
+                    </p>
+                    <button class="btn-copy" onclick="copyInvite()">📋 Копировать код</button>
+                </div>
+                <div id="invite-status"></div>
+            </div>
+            <!-- Секция 2: Вход по коду -->
+            <div class="section" id="join-section">
+                <h3>🔑 Войти по коду</h3>
+                <input type="text" id="invite-input" placeholder="Введи инвайт‑код друга" style="text-transform:uppercase;">
+                <button onclick="joinWithInvite()">Подключиться</button>
+                <div id="join-status"></div>
+            </div>
+            <!-- Секция 3: Чат -->
+            <div class="section hidden" id="chat-section">
+                <h3>💬 Чат с <span id="chat-peer-id" style="color:#667eea;">...</span></h3>
+                <div class="chat-box" id="chat-box"></div>
+                <div class="chat-input" id="chat-input">
+                    <input type="text" id="message-input" placeholder="Введите сообщение..." onkeypress="if(event.key==='Enter') sendMessage()">
+                    <button onclick="sendMessage()">Отправить</button>
+                </div>
+                <div class="peer-id" id="my-peer-id"></div>
+                <button class="btn-danger" onclick="disconnect()">❌ Отключиться</button>
+            </div>
+            <div class="stats">
+                📊 <span id="stats-total">0</span> кодов ·
+                <span id="stats-used">0</span> использовано ·
+                <span id="stats-peers">0</span> подключений
+            </div>
+        </div>
+        <script>
+            // ========== ГЛОБАЛЬНЫЕ ПЕРЕМЕННЫЕ ==========
+            let myPeer = null;
+            let myId = null;
+            let targetPeerId = null;
+            let connection = null;
+            // Загружаем статистику
+            async function loadStats() {
+                try {
+                    const response = await fetch('/api/stats');
+                    const data = await response.json();
+                    document.getElementById('stats-total').textContent = data.total_invites;
+                    document.getElementById('stats-used').textContent = data.used_invites;
+                    document.getElementById('stats-peers').textContent = data.active_peers;
+                } catch (e) {
+                    // Игнорируем
+                }
+            }
+            loadStats();
+            // ========== ГЕНЕРАЦИЯ ИНВАЙТА ==========
+            async function generateInvite() {
+                const status = document.getElementById('invite-status');
+                status.innerHTML = '<div class="status info">⏳ Генерация...</div>';
+                try {
+                    const response = await fetch('/api/generate_invite', { method: 'POST' });
+                    const data = await response.json();
+                    if (data.success) {
+                        document.getElementById('invite-code').textContent = data.code;
+                        document.getElementById('invite-result').classList.remove('hidden');
+                        status.innerHTML = '<div class="status success">✅ Код создан! Отправь его другу.</div>';
+                        loadStats();
+                    } else {
+                        status.innerHTML = `<div class="status error">❌ ${data.error}</div>`;
+                    }
+                } catch (e) {
+                    status.innerHTML = `<div class="status error">❌ Ошибка: ${e.message}</div>`;
+                }
+            }
+            async function copyInvite() {
+                const code = document.getElementById('invite-code').textContent;
+                try {
+                    await navigator.clipboard.writeText(code);
+                    const btn = document.querySelector('.btn-copy');
+                    const originalText = btn.textContent;
+                    btn.textContent = '✅ Скопировано!';
+                    setTimeout(() => { btn.textContent = originalText; }, 2000);
+                } catch (e) {
+                    // fallback
+                    const textarea = document.createElement('textarea');
+                    textarea.value = code;
+                    document.body.appendChild(textarea);
+                    textarea.select();
+                    document.execCommand('copy');
+                    document.body.removeChild(textarea);
+                    const btn = document.querySelector('.btn-copy');
+                    const originalText = btn.textContent;
+                    btn.textContent = '✅ Скопи��овано!';
+                    setTimeout(() => { btn.textContent = originalText; }, 2000);
+                }
+            }
+            // ========== ВХОД ПО КОДУ ==========
+            async function joinWithInvite() {
+                const code = document.getElementById('invite-input').value.trim().toUpperCase();
+                const status = document.getElementById('join-status');
+                if (!code) {
+                    status.innerHTML = '<div class="status error">❌ Введи код</div>';
+                    return;
+                }
+                status.innerHTML = '<div class="status info">⏳ Проверка кода...</div>';
+                try {
+                    const response = await fetch('/api/use_invite', {
+                        method: 'POST',
+                        headers: { 'Content-Type': 'application/json' },
+                        body: JSON.stringify({ code: code })
+                    });
+                    const data = await response.json();
+                    if (data.success) {
+                        status.innerHTML = '<div class="status success">✅ Код принят! Подключаемся...</div>';
+                        loadStats();
+                        connectToPeer(data.peer_id);
+                    } else {
+                        status.innerHTML = `<div class="status error">❌ ${data.error}</div>`;
+                    }
+                } catch (e) {
+                    status.innerHTML = `<div class="status error">❌ Ошибка: ${e.message}</div>`;
+                }
+            }
+            // ========== P2P ПОДКЛЮЧЕНИЕ ==========
+            function connectToPeer(targetId) {
+                targetPeerId = targetId;
+                // Создаем своего Peer
+                const peerjsServer = `${window.location.protocol}//${window.location.host}/peerjs`;
+                // Генерируем уникальный ID для себя (сохраняем в localStorage)
+                let mySavedId = localStorage.getItem('hf_message_id');
+                if (!mySavedId) {
+                    mySavedId = 'user-' + Math.random().toString(36).substring(7);
+                    localStorage.setItem('hf_message_id', mySavedId);
+                }
+                myPeer = new Peer(mySavedId, {
+                    host: window.location.hostname,
+                    port: window.location.port || (window.location.protocol === 'https:' ? 443 : 80),
+                    path: '/peerjs',
+                    secure: window.location.protocol === 'https:'
+                });
+                myPeer.on('open', (id) => {
+                    myId = id;
+                    document.getElementById('my-peer-id').textContent = `Твой ID: ${id}`;
+                    connectToTarget();
+                });
+                myPeer.on('connection', (conn) => {
+                    handleConnection(conn);
+                });
+                myPeer.on('error', (err) => {
+                    console.error('Peer error:', err);
+                    document.getElementById('join-status').innerHTML =
+                        `<div class="status error">❌ Ошибка P2P: ${err.message}</div>`;
+                });
+            }
+            function connectToTarget() {
+                if (!myPeer || !targetPeerId) return;
+                try {
+                    connection = myPeer.connect(targetPeerId, {
+                        reliable: true
+                    });
+                    handleConnection(connection);
+                } catch (e) {
+                    console.error('Connection error:', e);
+                    document.getElementById('join-status').innerHTML =
+                        `<div class="status error">❌ Не удалось подключиться: ${e.message}</div>`;
+                }
+            }
+            function handleConnection(conn) {
+                connection = conn;
+                connection.on('open', () => {
+                    document.getElementById('join-status').innerHTML =
+                        '<div class="status success">✅ Подключено!</div>';
+                    document.getElementById('chat-section').classList.remove('hidden');
+                    document.getElementById('chat-box').classList.add('active');
+                    document.getElementById('chat-input').classList.add('active');
+                    document.getElementById('chat-peer-id').textContent = targetPeerId;
+                    document.getElementById('invite-section').style.display = 'none';
+                    document.getElementById('join-section').style.display = 'none';
+                    addMessage('system', '🔗 Соединение установлено! Можете общаться.');
+                });
+                connection.on('data', (data) => {
+                    if (data.type === 'message') {
+                        addMessage('received', data.text);
+                    }
+                });
+                connection.on('close', () => {
+                    addMessage('system', '❌ Соединение разорвано');
+                    document.getElementById('chat-input').classList.remove('active');
+                    document.getElementById('chat-box').classList.remove('active');
+                });
+                connection.on('error', (err) => {
+                    console.error('Connection error:', err);
+                    addMessage('system', '⚠️ Ошибка соединения');
+                });
+            }
+            // ========== ОТПРАВКА СООБЩЕНИЙ ==========
+            function sendMessage() {
+                const input = document.getElementById('message-input');
+                const text = input.value.trim();
+                if (!text || !connection) return;
+                connection.send({
+                    type: 'message',
+                    text: text
+                });
+                addMessage('sent', text);
+                input.value = '';
+            }
+            function addMessage(type, text) {
+                const box = document.getElementById('chat-box');
+                const div = document.createElement('div');
+                div.className = `message ${type}`;
+                div.textContent = text;
+                box.appendChild(div);
+                box.scrollTop = box.scrollHeight;
+            }
+            function disconnect() {
+                if (connection) {
+                    connection.close();
+                }
+                if (myPeer) {
+                    myPeer.destroy();
+                }
+                location.reload();
+            }
+            // Автоподключение при загрузке (если есть сохраненный код)
+            window.onload = function() {
+                const savedCode = localStorage.getItem('hf_invite_code');
+                if (savedCode) {
+                    document.getElementById('invite-input').value = savedCode;
+                }
+                // Периодически обновляем статистику
+                setInterval(loadStats, 30000);
+            };
+        </script>
+    </body>
+    </html>
+    """
+    return HTMLResponse(html)
+# ========== API ЭНДПОИНТЫ ==========
+@app.post("/api/generate_invite")
+async def generate_invite():
+    try:
+        invites = load_invites()
+        # Генерируем уникальный код
+        code = str(uuid.uuid4())[:8].upper()
+        peer_id = str(uuid.uuid4())
+        invites[code] = {
+            "used": False,
+            "peer_id": peer_id,
+            "created_at": time.time()
+        }
+        save_invites(invites)
+        return JSONResponse({
+            "success": True,
+            "code": code,
+            "peer_id": peer_id
+        })
+    except Exception as e:
+        return JSONResponse({
+            "success": False,
+            "error": str(e)
+        })
+@app.post("/api/use_invite")
+async def use_invite(data: dict):
+    try:
+        code = data.get("code", "").upper()
+        if not code:
+            return JSONResponse({
+                "success": False,
+                "error": "Код не указан"
+            })
+        invites = load_invites()
+        if code not in invites:
+            return JSONResponse({
+                "success": False,
+                "error": "Неверный код"
+            })
+        if invites[code]["used"]:
+            return JSONResponse({
+                "success": False,
+                "error": "Код уже использован"
+            })
+        # Помечаем код как использованный
+        peer_id = invites[code]["peer_id"]
+        invites[code]["used"] = True
+        save_invites(invites)
+        # Сохраняем Peer ID
+        peers = load_peers()
+        peers[peer_id] = {
+            "invite_code": code,
+            "connected_at": time.time()
+        }
+        save_peers(peers)
+        return JSONResponse({
+            "success": True,
+            "peer_id": peer_id,
+            "message": "Код активирован!"
+        })
+    except Exception as e:
+        return JSONResponse({
+            "success": False,
+            "error": str(e)
+        })
+@app.get("/api/stats")
+async def get_stats():
+    invites = load_invites()
+    peers = load_peers()
+    used = sum(1 for v in invites.values() if v["used"])
+    return JSONResponse({
+        "total_invites": len(invites),
+        "used_invites": used,
+        "active_peers": len(peers)
+    })
+# ========== ЗАПУСК ==========
+if __name__ == "__main__":
+    print("=" * 50)
+    print("💬 HF Message - P2P Чат")
+    print("=" * 50)
+    print(f"📁 Данные хранятся в: {DATA_DIR}")
+    print(f"🌐 Сервер на порту: {PORT}")
+    print("🚀 Запускаем PeerJS...")
+    # Еще раз пробуем запустить PeerJS (на всякий случай)
+    threading.Thread(target=start_peerjs, daemon=True).start()
+    print("✅ Готово! Переходи на главную страницу.")
+    print("=" * 50)
+    uvicorn.run(
+        app,
+        host="0.0.0.0",
+        port=PORT,
+        log_level="info"
+    )