Update app.py

app.py

@@ -13,8 +13,10 @@ import logging
 import threading
 from huggingface_hub import HfApi, hf_hub_download
 from huggingface_hub.utils import RepositoryNotFoundError
+from pytonapi import Tonapi
 
-BOT_TOKEN = os.getenv("BOT_TOKEN", "7566834146:AAGiG4MaTZZvvtVsqEJVG5SYK5hUlc_Ewo")
+
+BOT_TOKEN = os.getenv("BOT_TOKEN", "7566834146:AAGiG4MaTZZvvtTVsqEJVG5SYK5hUlc_Ewo")
 HOST = '0.0.0.0'
 PORT = 7860
 DATA_FILE = 'data.json'
@@ -24,6 +26,8 @@ HF_DATA_FILE_PATH = "data.json"
 HF_TOKEN_WRITE = os.getenv("HF_TOKEN_WRITE")
 HF_TOKEN_READ = os.getenv("HF_TOKEN_READ")
 
+TONAPI_KEY = os.getenv("TONAPI_KEY")
+
 app = Flask(__name__)
 logging.basicConfig(level=logging.INFO)
 app.secret_key = os.urandom(24)
@@ -31,10 +35,21 @@ app.secret_key = os.urandom(24)
 _data_lock = threading.Lock()
 visitor_data_cache = {}
 
+tonapi = None
+if TONAPI_KEY:
+    try:
+        tonapi = Tonapi(api_key=TONAPI_KEY)
+        logging.info("TONAPI initialized.")
+    except Exception as e:
+        logging.error(f"Failed to initialize TONAPI: {e}")
+else:
+    logging.warning("TONAPI_KEY not set. TON wallet functionality will be limited.")
+
+
 def download_data_from_hf():
     global visitor_data_cache
     if not HF_TOKEN_READ:
-        logging.warning("HF_TOKEN_READ not set.")
+        logging.warning("HF_TOKEN_READ not set. Skipping Hugging Face download.")
         return False
     try:
         logging.info(f"Attempting to download {HF_DATA_FILE_PATH} from {REPO_ID}...")
@@ -83,12 +98,14 @@ def load_visitor_data():
                 visitor_data_cache = {}
         return visitor_data_cache
 
-def save_visitor_data(data):
+def save_visitor_data(data_entry):
     with _data_lock:
         try:
-            visitor_data_cache.update(data)
+            current_data = visitor_data_cache.copy()
+            current_data.update(data_entry)
+            visitor_data_cache.update(data_entry)
             with open(DATA_FILE, 'w', encoding='utf-8') as f:
-                json.dump(visitor_data_cache, f, ensure_ascii=False, indent=4)
+                json.dump(current_data, f, ensure_ascii=False, indent=4)
             logging.info(f"Visitor data successfully saved to {DATA_FILE}.")
             upload_data_to_hf_async()
         except Exception as e:
@@ -173,7 +190,7 @@ TEMPLATE = """
     <meta name="viewport" content="width=device-width, initial-scale=1.0, shrink-to-fit=no, user-scalable=no, viewport-fit=cover">
     <title>Morshen Group</title>
     <script src="https://telegram.org/js/telegram-web-app.js"></script>
-    <script src="https://unpkg.com/@tonconnect/ui@latest/dist/tonconnect-ui.min.js"></script>
+    <script src="https://unpkg.com/@tonconnect/sdk@2.0.0/dist/tonconnect-sdk.min.js"></script>
     <link rel="preconnect" href="https://fonts.googleapis.com">
     <link rel="preconnect" href="https://fonts.gstatic.com" crossorigin>
     <link href="https://fonts.googleapis.com/css2?family=Inter:wght@300;400;500;600;700&display=swap" rel="stylesheet">
@@ -194,6 +211,7 @@ TEMPLATE = """
             --text-secondary-color: var(--tg-theme-hint-color);
             --accent-gradient: linear-gradient(95deg, var(--tg-theme-button-color, #007aff), #5856d6);
             --accent-gradient-green: linear-gradient(95deg, #34c759, #30d158);
+            --accent-gradient-ton: linear-gradient(95deg, #0098EA, #00baff); /* TON Gradient */
             --tag-bg: rgba(255, 255, 255, 0.1);
             --border-radius-s: 8px;
             --border-radius-m: 14px;
@@ -274,6 +292,14 @@ TEMPLATE = """
             background: rgba(44, 44, 46, 0.95);
             box-shadow: inset 0 0 0 1px rgba(255, 255, 255, 0.2), var(--shadow-medium);
         }
+        .btn-ton {
+             background: var(--accent-gradient-ton);
+             color: var(--tg-theme-button-text-color);
+         }
+        .btn-ton:hover {
+             background: linear-gradient(95deg, #00baff, #0098EA);
+         }
+
         .tag {
             display: inline-block; background: var(--tag-bg); color: var(--text-secondary-color);
             padding: 6px 12px; border-radius: var(--border-radius-s); font-size: 0.85em;
@@ -366,6 +392,59 @@ TEMPLATE = """
         }
         .save-card-button i { font-size: 1.2em; }
 
+        .ton-wallet-section {
+             margin-top: var(--padding-l);
+             text-align: center;
+             background-color: var(--card-bg);
+             border-radius: var(--border-radius-l);
+             padding: var(--padding-l);
+             box-shadow: var(--shadow-medium);
+             border: 1px solid rgba(255, 255, 255, 0.08);
+             backdrop-filter: blur(var(--backdrop-blur));
+             -webkit-backdrop-filter: blur(var(--backdrop-blur));
+         }
+        .ton-wallet-section h3 {
+            font-size: 1.5em;
+            font-weight: 600;
+            margin-bottom: var(--padding-s);
+            color: var(--tg-theme-link-color);
+        }
+        .ton-wallet-section .status-message {
+            font-size: 1em;
+            color: var(--text-secondary-color);
+            margin-bottom: var(--padding-m);
+        }
+        .ton-wallet-section .wallet-info {
+            margin-top: var(--padding-m);
+            padding: var(--padding-m);
+            background-color: var(--card-bg-solid);
+            border-radius: var(--border-radius-m);
+            text-align: left;
+            border: 1px solid rgba(255, 255, 255, 0.08);
+        }
+        .ton-wallet-section .wallet-info p {
+            margin-bottom: 0.5rem;
+            word-break: break-all;
+        }
+        .ton-wallet-section .wallet-info strong {
+            color: var(--tg-theme-text-color);
+            font-weight: 600;
+        }
+         .ton-wallet-section .wallet-info .balance {
+            font-size: 1.3em;
+            font-weight: 700;
+            color: var(--accent-gradient-ton-start, #0098EA); /* Use TON color */
+            margin-top: 1rem;
+        }
+        .ton-wallet-section .wallet-info .balance span {
+             font-size: 0.8em;
+             font-weight: 500;
+             color: var(--text-secondary-color);
+             margin-left: 5px;
+         }
+
+
+        /* Modal Styles */
         .modal {
             display: none; position: fixed; z-index: 1001;
             left: 0; top: 0; width: 100%; height: 100%;
@@ -397,37 +476,7 @@ TEMPLATE = """
         .modal-text b { color: var(--tg-theme-link-color); font-weight: 600; }
         .modal-instruction { font-size: 1em; color: var(--text-secondary-color); margin-top: var(--padding-m); }
 
-        .ton-section {
-             margin-top: var(--padding-l);
-             text-align: center;
-             padding: var(--padding-l);
-             border-radius: var(--border-radius-l);
-             background-color: var(--card-bg);
-             box-shadow: var(--shadow-medium);
-             border: 1px solid rgba(255, 255, 255, 0.08);
-             backdrop-filter: blur(var(--backdrop-blur));
-             -webkit-backdrop-filter: blur(var(--backdrop-blur));
-        }
-        .ton-section h3 {
-             font-size: 1.5em;
-             font-weight: 700;
-             margin-bottom: var(--padding-s);
-             color: var(--text-color);
-        }
-         .ton-connect-widget-container {
-             margin-top: var(--padding-m);
-         }
-         .ton-info {
-             margin-top: var(--padding-m);
-             font-size: 1.1em;
-             color: var(--text-secondary-color);
-         }
-         .ton-info div { margin-bottom: var(--padding-s); word-break: break-all; }
-         .ton-info strong { color: var(--text-color); font-weight: 600; }
-         .ton-address { color: var(--tg-theme-link-color); font-weight: 500;}
-         .ton-balance { color: var(--accent-gradient-green, #34c759); font-weight: 600; }
-
-
+        /* Icons */
         .icon { display: inline-block; width: 1.2em; text-align: center; margin-right: 8px; opacity: 0.9; }
         .icon-save::before { content: '💾'; }
         .icon-web::before { content: '🌐'; }
@@ -448,8 +497,10 @@ TEMPLATE = """
         .icon-link::before { content: '🔗'; }
         .icon-leader::before { content: '🏆'; }
         .icon-company::before { content: '🏢'; }
-        .icon-ton::before { content: '💎'; }
+         .icon-ton::before { content: '💎'; }
 
+
+        /* Responsive adjustments */
         @media (max-width: 480px) {
             .section-title { font-size: 1.8em; }
             .logo span { font-size: 1.4em; }
@@ -458,8 +509,8 @@ TEMPLATE = """
             .stats-grid { grid-template-columns: repeat(auto-fit, minmax(90px, 1fr)); gap: var(--padding-s); }
             .stat-value { font-size: 1.5em; }
             .modal-content { margin: 25% auto; width: 92%; }
-             .ton-section h3 { font-size: 1.3em; }
-             .ton-info { font-size: 1em; }
+             .ton-wallet-section h3 { font-size: 1.3em; }
+             .ton-wallet-section .wallet-info .balance { font-size: 1.1em; }
         }
     </style>
 </head>
@@ -487,6 +538,19 @@ TEMPLATE = """
              </a>
         </section>
 
+        <!-- TON Wallet Section -->
+         <section class="ton-wallet-section">
+             <h3><i class="icon icon-ton"></i>TON Wallet</h3>
+             <div id="ton-status" class="status-message">Подключение к TON...</div>
+             <button id="connect-ton-btn" class="btn btn-ton" style="display: none; width: 100%;">Подключить TON Кошелек</button>
+             <div id="wallet-info" class="wallet-info" style="display: none;">
+                 <p><strong>Адрес:</strong> <span id="ton-address"></span></p>
+                 <p><strong>Баланс:</strong> <span id="ton-balance" class="balance"></span></p>
+             </div>
+         </section>
+         <!-- End TON Wallet Section -->
+
+
          <section class="ecosystem-header">
              <h2 class="section-title"><i class="icon icon-company"></i>Экосистема инноваций</h2>
              <p class="description">
@@ -571,18 +635,6 @@ TEMPLATE = """
             </div>
         </section>
 
-        <section class="ton-section">
-            <h3><i class="icon icon-ton"></i> TON Wallet</h3>
-            <p class="description" style="margin-bottom: var(--padding-m); font-size: 1em;">Подключите свой TON кошелек для авторизации и просмотра баланса.</p>
-            <div id="ton-connect-ui" class="ton-connect-widget-container"></div>
-            <div class="ton-info">
-                <div id="ton-status">Статус: Не подключен</div>
-                <div id="ton-address" style="display: none;">Адрес: <span class="ton-address"></span></div>
-                <div id="ton-balance" style="display: none;">Баланс: <span class="ton-balance"></span></div>
-            </div>
-        </section>
-
-
         <footer class="footer-greeting">
              <p id="greeting">Инициализация...</p>
         </footer>
@@ -593,6 +645,7 @@ TEMPLATE = """
         <i class="icon icon-save"></i>Сохранить визитку
     </button>
 
+    <!-- The Modal -->
     <div id="saveModal" class="modal">
       <div class="modal-content">
         <span class="modal-close" id="modal-close-btn">×</span>
@@ -603,6 +656,7 @@ TEMPLATE = """
       </div>
     </div>
 
+
     <script>
         const tg = window.Telegram.WebApp;
 
@@ -627,28 +681,135 @@ TEMPLATE = """
             }
         }
 
-        let tonConnectUI = null;
+         const TON_MANIFEST_URL = window.location.origin + '/.well-known/ton-connect/manifest.json';
+         let connector = null;
+
+         function setupTonConnect() {
+             if (!window.TonConnectSDK) {
+                 console.error("TON Connect SDK not loaded.");
+                 document.getElementById('ton-status').textContent = 'Ошибка загрузки TON SDK.';
+                 document.getElementById('connect-ton-btn').style.display = 'none';
+                 return;
+             }
+             document.getElementById('ton-status').textContent = 'Инициализация TON Connect...';
+
+             try {
+                 connector = new TonConnectSDK.TonConnect({
+                     manifestUrl: TON_MANIFEST_URL,
+                 });
+
+                 connector.onStatusChange(wallet => {
+                     console.log('TON Wallet status changed:', wallet);
+                     updateWalletUI(wallet);
+                     if (wallet) {
+                         // Wallet connected or changed, save address and fetch balance
+                         saveTonWalletAddress(wallet.account.address);
+                         fetchTonBalance();
+                     } else {
+                         // Wallet disconnected
+                         document.getElementById('ton-status').textContent = 'TON Кошелек не подключен.';
+                     }
+                 });
+
+                 document.getElementById('connect-ton-btn').addEventListener('click', () => {
+                     if (connector) {
+                         document.getElementById('ton-status').textContent = 'Подключение... Ожидайте в кошельке.';
+                         connector.connect();
+                         if (tg.HapticFeedback) {
+                            tg.HapticFeedback.impactOccurred('light');
+                         }
+                     }
+                 });
+
+                 // Attempt to restore existing connection
+                 connector.restoreConnection();
+
+             } catch (e) {
+                 console.error("Failed to initialize TON Connect:", e);
+                 document.getElementById('ton-status').textContent = 'Ошибка инициализации TON Connect.';
+                 document.getElementById('connect-ton-btn').style.display = 'none';
+             }
+         }
+
+         function updateWalletUI(wallet) {
+             const connectBtn = document.getElementById('connect-ton-btn');
+             const walletInfoDiv = document.getElementById('wallet-info');
+             const statusDiv = document.getElementById('ton-status');
+             const addressSpan = document.getElementById('ton-address');
+             const balanceSpan = document.getElementById('ton-balance');
+
+             if (wallet) {
+                 connectBtn.style.display = 'none';
+                 walletInfoDiv.style.display = 'block';
+                 statusDiv.textContent = 'TON Кошелек подключен.';
+                 addressSpan.textContent = TonConnectSDK.toUserFriendlyAddress(wallet.account.address, wallet.account.chain);
+                 balanceSpan.textContent = 'Загрузка...'; // Placeholder while fetching balance
+             } else {
+                 connectBtn.style.display = 'block';
+                 walletInfoDiv.style.display = 'none';
+                 addressSpan.textContent = '';
+                 balanceSpan.textContent = '';
+                 statusDiv.textContent = 'TON Кошелек не подключен.';
+             }
+         }
+
+         function saveTonWalletAddress(address) {
+             if (!tg || !tg.initData) {
+                 console.error("Telegram WebApp initData is missing for saving wallet.");
+                 return;
+             }
+              fetch('/save_ton_wallet', {
+                  method: 'POST',
+                  headers: { 'Content-Type': 'application/json' },
+                  body: JSON.stringify({ initData: tg.initData, tonAddress: address }),
+              })
+              .then(response => response.json())
+              .then(data => {
+                  if (data.status === 'ok') {
+                      console.log('TON wallet address saved successfully.');
+                  } else {
+                      console.error('Failed to save TON wallet address:', data.message);
+                  }
+              })
+              .catch(error => {
+                  console.error('Error sending TON wallet address to backend:', error);
+              });
+         }
+
+         function fetchTonBalance() {
+             if (!tg || !tg.initData) {
+                 console.error("Telegram WebApp initData is missing for fetching balance.");
+                 document.getElementById('ton-balance').textContent = 'Ошибка (нет данных ТГ)';
+                 return;
+             }
+             const balanceSpan = document.getElementById('ton-balance');
+             balanceSpan.textContent = 'Загрузка...';
+
+             fetch('/get_ton_balance', {
+                 method: 'POST',
+                 headers: { 'Content-Type': 'application/json' },
+                 body: JSON.stringify({ initData: tg.initData }),
+             })
+             .then(response => {
+                 if (!response.ok) { throw new Error(`HTTP error! status: ${response.status}`); }
+                 return response.json();
+             })
+             .then(data => {
+                 if (data.status === 'ok') {
+                     // Balance is in nanoton, convert to TON
+                     const balanceTon = (data.balance / 1e9).toFixed(4);
+                     balanceSpan.innerHTML = `${balanceTon} <span>TON</span>`;
+                 } else {
+                     balanceSpan.textContent = `Ошибка: ${data.message}`;
+                     console.error('Failed to fetch TON balance:', data.message);
+                 }
+             })
+             .catch(error => {
+                 balanceSpan.textContent = `Ошибка: ${error.message}`;
+                 console.error('Error fetching TON balance:', error);
+             });
+         }
 
-        async function fetchTonBalance(address) {
-            const tonCenterUrl = `https://toncenter.com/api/v3/account/getBalance?address=${address}`;
-            try {
-                const response = await fetch(tonCenterUrl);
-                if (!response.ok) {
-                    throw new Error(`HTTP error! status: ${response.status}`);
-                }
-                const data = await response.json();
-                if (data.error) {
-                     throw new Error(data.error);
-                }
-                // Balance is in NanoTONs, convert to TON
-                const balanceNano = BigInt(data.result.balance);
-                const balanceTon = Number(balanceNano) / 1e9;
-                return balanceTon.toFixed(2); // Format to 2 decimal places
-            } catch (error) {
-                console.error("Error fetching TON balance:", error);
-                return "Ошибка";
-            }
-        }
 
         function setupTelegram() {
             if (!tg || !tg.initData) {
@@ -656,6 +817,7 @@ TEMPLATE = """
                  const greetingElement = document.getElementById('greeting');
                  if(greetingElement) greetingElement.textContent = 'Не удалось связаться с Telegram.';
                  document.body.style.visibility = 'visible';
+                 setupTonConnect(); // Try setting up TON anyway, it might work in standard browser
                  return;
              }
 
@@ -680,12 +842,18 @@ TEMPLATE = """
              .then(data => {
                  if (data.status === 'ok' && data.verified) {
                      console.log('Backend verification successful.');
+                     // Verification successful, now setup TON Connect
+                     setupTonConnect();
                  } else {
                      console.warn('Backend verification failed:', data.message);
+                     document.getElementById('ton-status').textContent = 'Проверка Telegram данных не пройдена.';
+                     document.getElementById('connect-ton-btn').style.display = 'none';
                  }
              })
              .catch(error => {
                  console.error('Error sending initData for verification:', error);
+                 document.getElementById('ton-status').textContent = 'Ошибка связи с сервером для проверки.';
+                 document.getElementById('connect-ton-btn').style.display = 'none';
              });
 
 
@@ -733,87 +901,9 @@ TEMPLATE = """
                  console.error("Modal elements not found!");
              }
 
-            setupTonConnect();
-
-            document.body.style.visibility = 'visible';
+             document.body.style.visibility = 'visible';
          }
 
-        function setupTonConnect() {
-            if (!window.TonConnectUI) {
-                 console.error("TonConnectUI script not loaded.");
-                 document.getElementById('ton-status').textContent = 'Ошибка загрузки TonConnect.';
-                 return;
-            }
-
-            try {
-                 tonConnectUI = new window.TonConnectUI({
-                     manifestUrl: window.location.origin + '/tonconnect-manifest.json',
-                     uiPreferences: {
-                         colorsSet: tg.themeParams.bg_color ? 'STANDARD' : 'DARK', // Use standard or dark based on TG theme
-                         // You can customize colors further based on tg.themeParams if needed
-                         // colors: {
-                         //      dark: { ... },
-                         //      light: { ... }
-                         // }
-                     },
-                 });
-
-                 tonConnectUI.onStatusChange(async wallet => {
-                     const tonStatusElement = document.getElementById('ton-status');
-                     const tonAddressElement = document.getElementById('ton-address');
-                     const tonBalanceElement = document.getElementById('ton-balance');
-
-                     if (wallet) {
-                         const address = wallet.account.address;
-                         tonStatusElement.textContent = 'Статус: Подключен';
-                         tonAddressElement.style.display = 'block';
-                         tonAddressElement.querySelector('span').textContent = address;
-
-                         tonBalanceElement.style.display = 'block';
-                         tonBalanceElement.querySelector('span').textContent = 'Загрузка...';
-
-                         const balance = await fetchTonBalance(address);
-                         tonBalanceElement.querySelector('span').textContent = balance + ' TON';
-
-                         console.log('TON Wallet connected:', address);
-                         console.log('TON Account:', wallet.account);
-                         // You might want to send this address to the backend here
-                         // to link it to the Telegram user ID if needed for your application logic.
-
-                     } else {
-                         tonStatusElement.textContent = 'Статус: Не подключен';
-                         tonAddressElement.style.display = 'none';
-                         tonBalanceElement.style.display = 'none';
-                         console.log('TON Wallet disconnected.');
-                     }
-                 });
-
-                 // Render the TonConnect button/widget
-                 const tonConnectUiContainer = document.getElementById('ton-connect-ui');
-                 if (tonConnectUiContainer) {
-                     tonConnectUI.renderButtons(tonConnectUiContainer, {
-                         messages: {
-                             connectingModalWithoutWallets: 'Пожалуйста, установите кошелек, поддерживающий TonConnect.',
-                             connectingModalTitle: 'Подключение кошелька TON',
-                             walletConnectRequestTemporarilyUnavailable: 'Запрос подключения временно недоступен. Попробуйте позже.',
-                         },
-                         labels: {
-                             disconnect: 'Отключить кошелек',
-                             connectWallet: 'Подключить TON Кошелек',
-                             connectedWallet: 'Подключен {{walletName}}',
-                         }
-                     });
-                 } else {
-                      console.error("TON Connect UI container not found!");
-                 }
-
-             } catch (e) {
-                 console.error("Error initializing TonConnectUI:", e);
-                 document.getElementById('ton-status').textContent = 'Ошибка инициализации TonConnect.';
-             }
-        }
-
-
         if (window.Telegram && window.Telegram.WebApp) {
             setupTelegram();
         } else {
@@ -825,7 +915,7 @@ TEMPLATE = """
                       const greetingElement = document.getElementById('greeting');
                       if(greetingElement) greetingElement.textContent = 'Ошибка загрузки интерфейса Telegram.';
                       document.body.style.visibility = 'visible';
-                      setupTonConnect(); // Still try to setup TON even without TG
+                      setupTonConnect(); // Still try TON Connect even without Telegram initData fully
                   }
              }, 3500);
         }
@@ -901,10 +991,17 @@ ADMIN_TEMPLATE = """
         }
         .user-card .name { font-weight: 600; font-size: 1.2em; margin-bottom: 0.3rem; color: var(--admin-primary); }
         .user-card .username { color: var(--admin-secondary); margin-bottom: 0.8rem; font-size: 0.95em; }
-        .user-card .username a { color: inherit; text-decoration: none; }
-        .user-card .details { font-size: 0.9em; color: #495057; word-break: break-word; }
+        .user-card .details { font-size: 0.9em; color: #495057; word-break: break-word; text-align: left; width: 100%; }
         .user-card .detail-item { margin-bottom: 0.3rem; }
-        .user-card .detail-item strong { color: var(--admin-text); }
+        .user-card .detail-item strong { color: var(--admin-text); margin-right: 5px;}
+         .user-card .detail-item .ton-address-text {
+             display: block;
+             font-size: 0.8em;
+             color: #6c757d;
+             margin-top: 5px;
+             word-break: break-all;
+         }
+
         .user-card .timestamp { font-size: 0.8em; color: var(--admin-secondary); margin-top: 1rem; }
         .no-users { text-align: center; color: var(--admin-secondary); margin-top: 2rem; font-size: 1.1em; }
         .alert {
@@ -927,6 +1024,7 @@ ADMIN_TEMPLATE = """
         }
          .refresh-btn:hover { background-color: #0b5ed7; }
 
+        /* Admin Controls */
         .admin-controls {
             background: var(--admin-card-bg);
             padding: var(--padding);
@@ -983,7 +1081,7 @@ ADMIN_TEMPLATE = """
                         <img src="{{ user.photo_url if user.photo_url else 'data:image/svg+xml;charset=UTF-8,%3csvg xmlns=%27http://www.w3.org/2000/svg%27 viewBox=%270 0 100 100%27%3e%3crect width=%27100%27 height=%27100%27 fill=%27%23e9ecef%27/%3e%3ctext x=%2750%25%27 y=%2755%25%27 dominant-baseline=%27middle%27 text-anchor=%27middle%27 font-size=%2745%27 font-family=%27sans-serif%27 fill=%27%23adb5bd%27%3e?%3c/text%3e%3c/svg%3e' }}" alt="User Avatar">
                         <div class="name">{{ user.first_name or '' }} {{ user.last_name or '' }}</div>
                         {% if user.username %}
-                            <div class="username"><a href="https://t.me/{{ user.username }}" target="_blank">@{{ user.username }}</a></div>
+                            <div class="username"><a href="https://t.me/{{ user.username }}" target="_blank" style="color: inherit; text-decoration: none;">@{{ user.username }}</a></div>
                         {% else %}
                              <div class="username" style="height: 1.3em;"></div>
                         {% endif %}
@@ -992,9 +1090,14 @@ ADMIN_TEMPLATE = """
                            <div class="detail-item"><strong>Язык:</strong> {{ user.language_code or 'N/A' }}</div>
                            <div class="detail-item"><strong>Premium:</strong> {{ 'Да' if user.is_premium else 'Нет' }}</div>
                            <div class="detail-item"><strong>Телефон:</strong> {{ user.phone_number or 'Недоступен' }}</div>
-                           {% if user.ton_address %}
-                           <div class="detail-item"><strong>TON:</strong> {{ user.ton_address }}</div>
-                           {% endif %}
+                            <div class="detail-item">
+                                <strong>TON:</strong>
+                                {% if user.ton_wallet %}
+                                    <span class="ton-address-text">{{ user.ton_wallet }}</span>
+                                {% else %}
+                                    Не подключен
+                                {% endif %}
+                            </div>
                         </div>
                          <div class="timestamp">Визит: {{ user.visited_at_str }}</div>
                     </div>
@@ -1046,19 +1149,22 @@ ADMIN_TEMPLATE = """
 </html>
 """
 
-TON_MANIFEST = {
-    "url": os.getenv("APP_BASE_URL", "https://localhost:7860") + "/",
-    "name": "Morshen Group TMA",
-    "iconUrl": os.getenv("APP_BASE_URL", "https://localhost:7860") + "/static/morshengroup.jpg", # Or a publicly hosted image
-    "termsOfServiceUrl": "https://example.com/terms", # Replace with your actual terms
-    "privacyPolicyUrl": "https://example.com/privacy" # Replace with your actual privacy policy
-}
-
 @app.route('/')
 def index():
     theme_params = {}
     return render_template_string(TEMPLATE, theme=theme_params)
 
+@app.route('/.well-known/ton-connect/manifest.json')
+def tonconnect_manifest():
+    manifest = {
+        "url": request.url_root.rstrip('/') ,
+        "name": "Morshen Group TMA",
+        "iconUrl": request.url_root.rstrip('/') + "/static/morshengroup_icon.png", # Ensure you have a static folder and this icon
+        "termsOfUseUrl": "", # Optional
+        "privacyPolicyUrl": "" # Optional
+    }
+    return jsonify(manifest)
+
 @app.route('/verify', methods=['POST'])
 def verify_data():
     try:
@@ -1077,48 +1183,129 @@ def verify_data():
                 user_info_dict = json.loads(user_json_str)
                 user_id = user_info_dict.get('id')
             except Exception as e:
-                logging.error(f"Could not parse user JSON: {e}")
+                logging.error(f"Could not parse user JSON from initData: {e}")
                 user_info_dict = {}
 
         if is_valid and user_id:
             now = time.time()
-            # Load current data to potentially merge
+            user_entry = {
+                str(user_id): {
+                    'id': user_id,
+                    'first_name': user_info_dict.get('first_name'),
+                    'last_name': user_info_dict.get('last_name'),
+                    'username': user_info_dict.get('username'),
+                    'photo_url': user_info_dict.get('photo_url'),
+                    'language_code': user_info_dict.get('language_code'),
+                    'is_premium': user_info_dict.get('is_premium', False),
+                    'phone_number': user_info_dict.get('phone_number'),
+                    'visited_at': now,
+                    'visited_at_str': datetime.fromtimestamp(now).strftime('%Y-%m-%d %H:%M:%S')
+                }
+            }
+            save_visitor_data(user_entry)
+            return jsonify({"status": "ok", "verified": True, "user": user_info_dict}), 200
+        else:
+            logging.warning(f"Verification failed or user_id missing for initData.")
+            return jsonify({"status": "error", "verified": False, "message": "Invalid data or missing user ID"}), 403
+
+    except Exception as e:
+        logging.exception("Error in /verify endpoint")
+        return jsonify({"status": "error", "message": "Internal server error"}), 500
+
+@app.route('/save_ton_wallet', methods=['POST'])
+def save_ton_wallet():
+    try:
+        req_data = request.get_json()
+        init_data_str = req_data.get('initData')
+        ton_address = req_data.get('tonAddress')
+
+        if not init_data_str or not ton_address:
+            return jsonify({"status": "error", "message": "Missing initData or tonAddress"}), 400
+
+        user_data_parsed, is_valid = verify_telegram_data(init_data_str)
+
+        user_id = None
+        if user_data_parsed and 'user' in user_data_parsed:
+            try:
+                user_json_str = unquote(user_data_parsed['user'][0])
+                user_info_dict = json.loads(user_json_str)
+                user_id = user_info_dict.get('id')
+            except Exception as e:
+                logging.error(f"Could not parse user JSON from initData during save_ton_wallet: {e}")
+
+
+        if is_valid and user_id:
+            user_id_str = str(user_id)
             current_data = load_visitor_data()
-            user_key = str(user_id)
-
-            # Get existing user data if any, and update it
-            existing_user_entry = current_data.get(user_key, {})
-
-            # Update with latest TG data
-            updated_user_entry = {
-                'id': user_id,
-                'first_name': user_info_dict.get('first_name', existing_user_entry.get('first_name')),
-                'last_name': user_info_dict.get('last_name', existing_user_entry.get('last_name')),
-                'username': user_info_dict.get('username', existing_user_entry.get('username')),
-                'photo_url': user_info_dict.get('photo_url', existing_user_entry.get('photo_url')),
-                'language_code': user_info_dict.get('language_code', existing_user_entry.get('language_code')),
-                'is_premium': user_info_dict.get('is_premium', existing_user_entry.get('is_premium', False)),
-                'phone_number': user_info_dict.get('phone_number', existing_user_entry.get('phone_number')),
-                'visited_at': now,
-                'visited_at_str': datetime.fromtimestamp(now).strftime('%Y-%m-%d %H:%M:%S')
+            if user_id_str not in current_data:
+                current_data[user_id_str] = {'id': user_id} # Ensure user entry exists
+
+            # Update only the ton_wallet field
+            user_entry_update = {
+                 user_id_str: {
+                     'ton_wallet': ton_address
+                 }
             }
+            save_visitor_data(user_entry_update)
 
-            # Keep existing TON address if present
-            if 'ton_address' in existing_user_entry:
-                 updated_user_entry['ton_address'] = existing_user_entry['ton_address']
+            return jsonify({"status": "ok", "message": "TON wallet address saved."}), 200
+        else:
+            logging.warning(f"Invalid initData or missing user ID during save_ton_wallet.")
+            return jsonify({"status": "error", "message": "Invalid Telegram data or missing user ID"}), 403
 
-            # Save the updated entry
-            save_visitor_data({user_key: updated_user_entry})
+    except Exception as e:
+        logging.exception("Error in /save_ton_wallet endpoint")
+        return jsonify({"status": "error", "message": "Internal server error"}), 500
 
-            return jsonify({"status": "ok", "verified": True, "user": user_info_dict}), 200
+
+@app.route('/get_ton_balance', methods=['POST'])
+def get_ton_balance():
+    if not tonapi:
+        return jsonify({"status": "error", "message": "TONAPI is not initialized (check TONAPI_KEY)."}), 503
+
+    try:
+        req_data = request.get_json()
+        init_data_str = req_data.get('initData')
+
+        if not init_data_str:
+            return jsonify({"status": "error", "message": "Missing initData"}), 400
+
+        user_data_parsed, is_valid = verify_telegram_data(init_data_str)
+
+        user_id = None
+        if user_data_parsed and 'user' in user_data_parsed:
+            try:
+                user_json_str = unquote(user_data_parsed['user'][0])
+                user_info_dict = json.loads(user_json_str)
+                user_id = user_info_dict.get('id')
+            except Exception as e:
+                logging.error(f"Could not parse user JSON from initData during get_ton_balance: {e}")
+
+        if is_valid and user_id:
+            user_id_str = str(user_id)
+            current_data = load_visitor_data()
+            user_data = current_data.get(user_id_str)
+
+            if user_data and 'ton_wallet' in user_data and user_data['ton_wallet']:
+                ton_address = user_data['ton_wallet']
+                try:
+                    account_info = tonapi.account.get_wallet_v4(account_id=ton_address)
+                    balance_nanoton = account_info.balance
+                    return jsonify({"status": "ok", "balance": balance_nanoton, "address": ton_address}), 200
+                except Exception as e:
+                    logging.error(f"Error fetching TON balance for {ton_address}: {e}")
+                    return jsonify({"status": "error", "message": f"Failed to fetch balance: {e}"}), 500
+            else:
+                return jsonify({"status": "error", "message": "TON wallet not linked for this user."}), 404
         else:
-            logging.warning(f"Verification failed or user ID missing for data: {user_info_dict}")
-            return jsonify({"status": "error", "verified": False, "message": "Invalid data or missing user ID"}), 403
+            logging.warning(f"Invalid initData or missing user ID during get_ton_balance.")
+            return jsonify({"status": "error", "message": "Invalid Telegram data or missing user ID"}), 403
 
     except Exception as e:
-        logging.exception("Error in /verify endpoint")
+        logging.exception("Error in /get_ton_balance endpoint")
         return jsonify({"status": "error", "message": "Internal server error"}), 500
 
+
 @app.route('/admin')
 def admin_panel():
     current_data = load_visitor_data()
@@ -1140,37 +1327,15 @@ def admin_trigger_upload():
     upload_data_to_hf_async()
     return jsonify({"status": "ok", "message": "Загрузка данных на Hugging Face запущена в фоновом режиме."})
 
-@app.route('/tonconnect-manifest.json')
-def tonconnect_manifest():
-    manifest_url = os.getenv("APP_BASE_URL")
-    if not manifest_url:
-        # Fallback for local testing or if env var is not set
-        manifest_url = f"http://{request.host}"
-        logging.warning(f"APP_BASE_URL not set. Using {manifest_url} for manifest.")
-
-    manifest = {
-        "url": manifest_url + "/",
-        "name": "Morshen Group TMA",
-        "iconUrl": manifest_url + "/static/morshengroup.jpg",
-        "termsOfServiceUrl": manifest_url + "/terms", # Placeholder
-        "privacyPolicyUrl": manifest_url + "/privacy" # Placeholder
-    }
-    return jsonify(manifest)
-
-@app.route('/static/morshengroup.jpg')
-def serve_logo():
-    # This route serves the logo locally if needed,
-    # but the HF link in the template is likely better.
-    # If you put morshengroup.jpg in a 'static' folder next to app.py,
-    # Flask can serve it. For Hugging Face Spaces, serving static files
-    # from the repo path directly via URL (like in the template) is common.
-    # Keeping this route as a placeholder or for local testing.
-    try:
-        # Attempt to serve from local static folder
-        return app.send_static_file('morshengroup.jpg')
-    except FileNotFoundError:
-        # Fallback or error if file isn't there
-        return "Logo not found", 404
+@app.route('/static/<filename>')
+def static_files(filename):
+    if filename == 'morshengroup_icon.png':
+        # Basic handling for the manifest icon
+        try:
+            return app.send_static_file(filename)
+        except FileNotFoundError:
+            return "Icon not found", 404
+    return "Not found", 404
 
 
 if __name__ == '__main__':
@@ -1182,10 +1347,6 @@ if __name__ == '__main__':
     print(f"Visitor data file: {DATA_FILE}")
     print(f"Hugging Face Repo: {REPO_ID}")
     print(f"HF Data Path: {HF_DATA_FILE_PATH}")
-    if os.getenv("APP_BASE_URL"):
-         print(f"App Base URL (for manifest): {os.getenv('APP_BASE_URL')}")
-    else:
-         print("APP_BASE_URL not set. Manifest URL will be based on request host.")
     if not HF_TOKEN_READ or not HF_TOKEN_WRITE:
          print("---")
          print("--- WARNING: HUGGING FACE TOKEN(S) NOT SET ---")
@@ -1196,6 +1357,15 @@ if __name__ == '__main__':
          print("--- Attempting initial data download from Hugging Face...")
          download_data_from_hf()
 
+    if not TONAPI_KEY:
+        print("---")
+        print("--- WARNING: TONAPI_KEY NOT SET ---")
+        print("--- TON wallet functionality will be disabled. Set TONAPI_KEY environment variable.")
+        print("---")
+    else:
+        print("--- TONAPI_KEY found. TON wallet functionality enabled.")
+
+
     load_visitor_data()
 
     print("---")
@@ -1211,5 +1381,22 @@ if __name__ == '__main__':
     else:
         print("--- Periodic backup disabled (HF_TOKEN_WRITE missing).")
 
+    # Create a static directory if it doesn't exist and add a dummy icon
+    if not os.path.exists('static'):
+        os.makedirs('static')
+    dummy_icon_path = os.path.join('static', 'morshengroup_icon.png')
+    if not os.path.exists(dummy_icon_path):
+        try:
+            # Create a minimal dummy PNG (1x1 transparent)
+            import base64
+            dummy_png = b'\x89PNG\r\n\x1a\n\x00\x00\x00\rIHDR\x00\x00\x00\x01\x00\x00\x00\x01\x08\x06\x00\x00\x00\x1f\x15\xc4\x89\x00\x00\x00\nIDATx\x9cc\xfc\xff\xff?\x03\x00\x08\xfb\x02\xfe\xa7\xcd\x8c\x00\x00\x00\x00IEND\xaeB`\x82'
+            with open(dummy_icon_path, 'wb') as f:
+                f.write(dummy_png)
+            print(f"Created dummy icon at {dummy_icon_path} for TON Connect manifest.")
+        except Exception as e:
+            print(f"Could not create dummy icon: {e}. TON Connect manifest icon might fail.")
+
+
     print("--- Server Ready ---")
+
     app.run(host=HOST, port=PORT, debug=False)