Update app.py

app.py

@@ -1,452 +1,705 @@
 import os
 import uuid
-import json
-import logging
-import threading
-import time
 import subprocess
-import sys
+import time
+import socket
+import logging
 import atexit
 from datetime import datetime
-from flask import Flask, request, jsonify, Response, redirect, url_for, flash
-from werkzeug.utils import secure_filename
-from dotenv import load_dotenv
+from flask import Flask, request, jsonify, Response, send_from_directory, redirect, url_for
 import google.generativeai as genai
 from huggingface_hub import HfApi, hf_hub_download
 from huggingface_hub.utils import RepositoryNotFoundError, HfHubHTTPError
-import requests
-
-# --- Load Environment Variables ---
-load_dotenv()
+import requests # Нужен для HfHubHTTPError
 
-# --- Configuration ---
+# --- Конфигурация ---
 app = Flask(__name__)
-app.secret_key = 'super-secret-key-for-generator-app-12345'
 
-# Google AI Configuration
-API_KEY_INTERNAL = os.getenv("GEMINI_API_KEY", "AIzaSyArKidc4o0MwbaCFKStlb2q2AwNg6Pnqns") 
+# ВНИМАНИЕ: Жестко закодированный API ключ - это небезопасно. Используйте переменные окружения.
+API_KEY_INTERNAL = "AIzaSyArKidc4o0MwbaCFKStlb2q2AwNg6Pnqns" # Используется ключ из Кода 1
+SITES_CACHE_DIR = 'generated_sites_cache' # Локальный кэш для .py файлов сайтов
 
-# Hugging Face Configuration
-REPO_ID = "Kgshop/testsynk"
-HF_TOKEN_WRITE = os.getenv("HF_TOKEN")
-HF_TOKEN_READ = os.getenv("HF_TOKEN_READ")
+# Конфигурация Hugging Face (адаптировано из Кода 2)
+REPO_ID = "Kgshop/testsynk" # Указанный репозиторий
+HF_TOKEN_WRITE = os.getenv("HF_TOKEN") # Токен с правом записи
+HF_TOKEN_READ = os.getenv("HF_TOKEN_READ", HF_TOKEN_WRITE) # Токен для чтения (или тот же, что и для записи)
 
-# Local Directories and Files
-GENERATED_SITES_DIR = 'generated_sites'
-DATA_FILE = 'generator_data.json' # State file for the generator app
-SYNC_FILES = [DATA_FILE]
+DOWNLOAD_RETRIES = 3
+DOWNLOAD_DELAY = 5
 
-# Runtime State
-running_sites = {}  # { 'site_id': {'process': Popen_object, 'port': int, 'prompt': str} }
-sites_data = {'sites': {}} # This will be loaded from DATA_FILE
-next_available_port = 7861 # Main app runs on 7860
+# Управление запущенными сайтами
+RUNNING_SITES = {} # {'site_filename.py': {'process': Popen_object, 'port': 5001, 'host': '127.0.0.1'}}
+BASE_PORT_FOR_GENERATED_SITES = 8001
+USED_PORTS = set()
 
-# Logging Configuration
+# --- Настройка Логирования ---
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
 
-# --- Hugging Face Backup & Restore Functions (Adapted from Code 2) ---
+# --- Создание директорий ---
+if not os.path.exists(SITES_CACHE_DIR):
+    os.makedirs(SITES_CACHE_DIR)
+
+# --- Функции для работы с Hugging Face ---
+def download_file_from_hf(repo_filename, local_save_path, retries=DOWNLOAD_RETRIES, delay=DOWNLOAD_DELAY):
+    if not HF_TOKEN_READ:
+        logging.warning("HF_TOKEN_READ not set. Download might fail for private repos.")
+    
+    token_to_use = HF_TOKEN_READ
 
-def download_db_from_hf(specific_file=None, retries=3, delay=5):
-    if not HF_TOKEN_READ and not HF_TOKEN_WRITE:
-        logging.warning("HF_TOKEN_READ/HF_TOKEN_WRITE not set. Download might fail for private repos.")
-    token_to_use = HF_TOKEN_READ if HF_TOKEN_READ else HF_TOKEN_WRITE
-    files_to_download = [specific_file] if specific_file else SYNC_FILES
-    logging.info(f"Attempting download for {files_to_download} from {REPO_ID}...")
-    all_successful = True
+    logging.info(f"Attempting download for {repo_filename} from {REPO_ID} to {local_save_path}...")
+    for attempt in range(retries + 1):
+        try:
+            logging.info(f"Downloading {repo_filename} (Attempt {attempt + 1}/{retries + 1})...")
+            downloaded_path = hf_hub_download(
+                repo_id=REPO_ID,
+                filename=repo_filename,
+                repo_type="dataset", # Обычно 'dataset' для файлов, или 'model'
+                token=token_to_use,
+                local_dir=os.path.dirname(local_save_path),
+                local_dir_use_symlinks=False,
+                force_download=True, # Всегда скачивать свежую версию
+                resume_download=False,
+                # Важно указать имя файла, если local_dir указан, иначе он создаст структуру папок репозитория
+                # hf_hub_download вернет полный путь к файлу, который может включать структуру папок репозитория.
+                # Мы должны убедиться, что он сохраняется как local_save_path
+            )
+            # Переименовываем/перемещаем, если hf_hub_download скачал не точно по пути
+            if downloaded_path != local_save_path:
+                os.makedirs(os.path.dirname(local_save_path), exist_ok=True)
+                os.replace(downloaded_path, local_save_path) # Используем replace для атомарности
+                logging.info(f"File moved from {downloaded_path} to {local_save_path}")
+
+            logging.info(f"Successfully downloaded {repo_filename} to {local_save_path}.")
+            return True
+        except RepositoryNotFoundError:
+            logging.error(f"Repository {REPO_ID} not found. Download cancelled.")
+            return False
+        except HfHubHTTPError as e:
+            if e.response.status_code == 404:
+                logging.warning(f"File {repo_filename} not found in repo {REPO_ID} (404).")
+                return False # Файл не найден, нет смысла ретраить
+            logging.error(f"HTTP error downloading {repo_filename} (Attempt {attempt + 1}): {e}. Retrying in {delay}s...")
+        except requests.exceptions.RequestException as e: # requests используется внутри huggingface_hub
+            logging.error(f"Network error downloading {repo_filename} (Attempt {attempt + 1}): {e}. Retrying in {delay}s...")
+        except Exception as e:
+            logging.error(f"Unexpected error downloading {repo_filename} (Attempt {attempt + 1}): {e}. Retrying in {delay}s...", exc_info=True)
+        
+        if attempt < retries:
+            time.sleep(delay)
+    
+    logging.error(f"Failed to download {repo_filename} after {retries + 1} attempts.")
+    return False
 
-    for file_name in files_to_download:
-        success = False
-        for attempt in range(retries + 1):
-            try:
-                logging.info(f"Downloading {file_name} (Attempt {attempt + 1}/{retries + 1})...")
-                hf_hub_download(
-                    repo_id=REPO_ID, filename=file_name, repo_type="dataset", token=token_to_use,
-                    local_dir=".", local_dir_use_symlinks=False, force_download=True, resume_download=False
-                )
-                logging.info(f"Successfully downloaded {file_name}.")
-                success = True
-                break
-            except RepositoryNotFoundError:
-                 logging.error(f"Repository {REPO_ID} not found. Download cancelled.")
-                 return False
-            except HfHubHTTPError as e:
-                if e.response.status_code == 404:
-                    logging.warning(f"File {file_name} not found in repo {REPO_ID}. Creating empty local file.")
-                    if file_name == DATA_FILE:
-                        with open(file_name, 'w', encoding='utf-8') as f:
-                            json.dump({'sites': {}}, f)
-                    success = True # We consider this a success as we handled it.
-                    break
-                else:
-                    logging.error(f"HTTP error downloading {file_name} (Attempt {attempt + 1}): {e}. Retrying in {delay}s...")
-            except Exception as e:
-                 logging.error(f"Unexpected error downloading {file_name} (Attempt {attempt + 1}): {e}. Retrying...", exc_info=True)
-            if attempt < retries:
-                time.sleep(delay)
-        if not success:
-            logging.error(f"Failed to download {file_name} after {retries + 1} attempts.")
-            all_successful = False
-    return all_successful
-
-def upload_db_to_hf(specific_file=None):
+def upload_file_to_hf(local_filepath, path_in_repo):
     if not HF_TOKEN_WRITE:
-        logging.warning("HF_TOKEN (for writing) not set. Skipping upload to Hugging Face.")
-        return
+        logging.warning("HF_TOKEN_WRITE (for writing) not set. Skipping upload to Hugging Face.")
+        return False
+    if not os.path.exists(local_filepath):
+        logging.error(f"Local file {local_filepath} not found. Skipping upload.")
+        return False
     try:
         api = HfApi()
-        files_to_upload = [specific_file] if specific_file else SYNC_FILES
-        logging.info(f"Starting upload of {files_to_upload} to HF repo {REPO_ID}...")
-        for file_name in files_to_upload:
-            if os.path.exists(file_name):
-                api.upload_file(
-                    path_or_fileobj=file_name, path_in_repo=file_name, repo_id=REPO_ID,
-                    repo_type="dataset", token=HF_TOKEN_WRITE,
-                    commit_message=f"Sync generator state {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}"
-                )
-                logging.info(f"File {file_name} successfully uploaded to Hugging Face.")
-            else:
-                logging.warning(f"File {file_name} not found locally, skipping upload.")
+        logging.info(f"Starting upload of {local_filepath} to {path_in_repo} in HF repo {REPO_ID}...")
+        api.upload_file(
+            path_or_fileobj=local_filepath,
+            path_in_repo=path_in_repo,
+            repo_id=REPO_ID,
+            repo_type="dataset",
+            token=HF_TOKEN_WRITE,
+            commit_message=f"Sync {path_in_repo} {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}"
+        )
+        logging.info(f"File {local_filepath} successfully uploaded to {path_in_repo} in Hugging Face.")
+        return True
     except Exception as e:
-        logging.error(f"Error during Hugging Face upload: {e}", exc_info=True)
+        logging.error(f"Error uploading file {local_filepath} to Hugging Face: {e}", exc_info=True)
+        return False
 
-def periodic_backup():
+# --- Функции управления дочерними сайтами ---
+def find_free_port(start_port):
+    port = start_port
     while True:
-        time.sleep(1800) # 30 minutes
-        logging.info("Starting periodic backup...")
-        upload_db_to_hf()
-
-def load_data():
-    global sites_data
-    default_data = {'sites': {}}
-    try:
-        if not os.path.exists(DATA_FILE):
-             raise FileNotFoundError
-        with open(DATA_FILE, 'r', encoding='utf-8') as file:
-            data = json.load(file)
-        if 'sites' not in data: data['sites'] = {}
-        sites_data = data
-        logging.info(f"Local state loaded successfully from {DATA_FILE}")
-    except (FileNotFoundError, json.JSONDecodeError):
-        logging.warning(f"Local file {DATA_FILE} not found or corrupt. Attempting download from HF.")
-        if download_db_from_hf(specific_file=DATA_FILE):
+        if port in USED_PORTS:
+            port += 1
+            continue
+        with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
             try:
-                with open(DATA_FILE, 'r', encoding='utf-8') as file:
-                    sites_data = json.load(file)
-                if 'sites' not in sites_data: sites_data['sites'] = {}
-                logging.info(f"State loaded from {DATA_FILE} after download.")
-            except (FileNotFoundError, json.JSONDecodeError):
-                logging.error(f"Failed to read file after download. Using default state.")
-                sites_data = default_data
+                s.bind(('127.0.0.1', port))
+                USED_PORTS.add(port)
+                return port
+            except OSError:
+                logging.warning(f"Port {port} is already in use by another application, trying next.")
+                # Добавляем в USED_PORTS, чтобы не пробовать его снова в этой сессии для наших сайтов
+                USED_PORTS.add(port) 
+                port += 1
+        if port > 65535:
+            raise RuntimeError("Could not find a free port.")
+
+
+def start_generated_site(py_filename):
+    if py_filename in RUNNING_SITES:
+        site_info = RUNNING_SITES[py_filename]
+        # Проверить, жив ли процесс
+        if site_info['process'].poll() is None: # None значит, что процесс еще работает
+            logging.info(f"Site {py_filename} is already running on port {site_info['port']}.")
+            return site_info['host'], site_info['port']
         else:
-            logging.error(f"Failed to download state from HF. Using empty default state.")
-            sites_data = default_data
+            logging.warning(f"Site {py_filename} was marked as running but process died. Cleaning up.")
+            USED_PORTS.discard(site_info['port'])
+            del RUNNING_SITES[py_filename]
 
-def save_data():
+    local_py_path = os.path.join(SITES_CACHE_DIR, py_filename)
+    
+    # 1. Скачать/убедиться в наличии файла
+    if not os.path.exists(local_py_path):
+        if not download_file_from_hf(f"sites/{py_filename}", local_py_path):
+            logging.error(f"Failed to download site {py_filename} from Hugging Face.")
+            return None, None
+    
+    # 2. Найти свободный порт и модифицировать код сайта
     try:
-        with open(DATA_FILE, 'w', encoding='utf-8') as file:
-            json.dump(sites_data, file, ensure_ascii=False, indent=4)
-        logging.info(f"Generator state saved to {DATA_FILE}")
-        # Upload to HF after every save for real-time backup
-        upload_db_to_hf(specific_file=DATA_FILE)
+        assigned_port = find_free_port(BASE_PORT_FOR_GENERATED_SITES)
+        with open(local_py_path, 'r', encoding='utf-8') as f:
+            original_code = f.read()
+        
+        # Заменить плейсхолдер порта
+        modified_code = original_code.replace("{{GENERATED_APP_PORT}}", str(assigned_port))
+        
+        # Убедимся, что код пытается запуститься на 0.0.0.0 или 127.0.0.1
+        # Это очень грубая проверка, лучше если AI генерирует предсказуемый app.run
+        if f"app.run(host='0.0.0.0', port={assigned_port})" not in modified_code and \
+           f"app.run(host='127.0.0.1', port={assigned_port})" not in modified_code and \
+           f"app.run(port={assigned_port}, host='0.0.0.0')" not in modified_code and \
+           f"app.run(port={assigned_port}, host='127.0.0.1')" not in modified_code:
+            
+            # Попробуем заменить более общий app.run(port=XXXX) или app.run()
+            import re
+            modified_code = re.sub(
+                r"app\.run\((.*)\)", 
+                f"app.run(host='0.0.0.0', port={assigned_port})", 
+                modified_code, 
+                flags=re.DOTALL
+            )
+            if "app.run(" not in modified_code : # Если app.run не было вообще
+                 modified_code += f"\n\nif __name__ == '__main__':\n    app.run(host='0.0.0.0', port={assigned_port})\n"
+
+
+        with open(local_py_path, 'w', encoding='utf-8') as f:
+            f.write(modified_code)
+
     except Exception as e:
-        logging.error(f"Error saving generator state: {e}", exc_info=True)
+        logging.error(f"Error preparing site {py_filename} for launch: {e}", exc_info=True)
+        if 'assigned_port' in locals() and assigned_port:
+            USED_PORTS.discard(assigned_port)
+        return None, None
 
-# --- Core AI Generation Logic (Modified for Flask Apps) ---
+    # 3. Запустить скрипт
+    try:
+        # Для Windows, если python не в PATH, может понадобиться sys.executable
+        # python_executable = sys.executable  
+        python_executable = "python" # Предполагаем, что python в PATH
+        process = subprocess.Popen(
+            [python_executable, local_py_path],
+            stdout=subprocess.PIPE, # Можно перенаправить в лог-файлы
+            stderr=subprocess.PIPE
+        )
+        time.sleep(2) # Дать время Flask-приложению запуститься
+
+        if process.poll() is not None: # Если процесс сразу завершился
+            stdout, stderr = process.communicate()
+            logging.error(f"Site {py_filename} failed to start. Return code: {process.returncode}")
+            logging.error(f"Stdout: {stdout.decode(errors='ignore')}")
+            logging.error(f"Stderr: {stderr.decode(errors='ignore')}")
+            USED_PORTS.discard(assigned_port)
+            return None, None
+
+        site_host = '127.0.0.1' # Сгенерированные сайты слушают на 0.0.0.0, но доступны через 127.0.0.1
+        RUNNING_SITES[py_filename] = {'process': process, 'port': assigned_port, 'host': site_host}
+        logging.info(f"Site {py_filename} started successfully on http://{site_host}:{assigned_port}")
+        return site_host, assigned_port
+    except Exception as e:
+        logging.error(f"Error launching site {py_filename}: {e}", exc_info=True)
+        if 'assigned_port' in locals() and assigned_port:
+            USED_PORTS.discard(assigned_port)
+        return None, None
+
+def cleanup_running_sites():
+    logging.info("Cleaning up running generated sites...")
+    for filename, site_info in list(RUNNING_SITES.items()): # list() для копии, т.к. будем удалять
+        try:
+            if site_info['process'].poll() is None: # Если процесс еще жив
+                logging.info(f"Terminating site {filename} (PID: {site_info['process'].pid}) on port {site_info['port']}")
+                site_info['process'].terminate() # Сначала пробуем мягко
+                try:
+                    site_info['process'].wait(timeout=5) # Ждем завершения
+                except subprocess.TimeoutExpired:
+                    logging.warning(f"Site {filename} did not terminate gracefully, killing.")
+                    site_info['process'].kill() # Если не помогло - жестко
+            USED_PORTS.discard(site_info['port'])
+            del RUNNING_SITES[filename]
+        except Exception as e:
+            logging.error(f"Error terminating site {filename}: {e}", exc_info=True)
+    logging.info("Cleanup complete.")
 
-def generate_flask_app_code_from_prompt(user_prompt):
+atexit.register(cleanup_running_sites)
+
+
+# --- Функция генерации кода сайта с помощью Google AI ---
+def generate_full_python_site_from_prompt(user_prompt):
     try:
+        # ВНИМАНИЕ: Ключ API жестко закодирован. В продакшене используйте переменные окружения.
         genai.configure(api_key=API_KEY_INTERNAL)
     except Exception as e:
-        raise ValueError(f"Failed to configure Google AI: {e}")
+        logging.error(f"Error configuring GenAI: {e}", exc_info=True)
+        raise ValueError(f"Не удалось настроить Google AI. Проблема с конфигурацией. Ошибка: {e}")
 
     if not user_prompt or not user_prompt.strip():
-        raise ValueError("Prompt cannot be empty.")
+        raise ValueError("Текстовый запрос (промпт) не может быть пустым.")
 
     system_instruction = (
-        "You are an expert Python Flask web developer. Your task is to generate a complete, single-file Flask application "
-        "based on the user's request. The entire application must be in ONE SINGLE Python (.py) file.\n"
-        "Crucial requirements for the generated file:\n"
-        "1.  **Single File:** Everything must be in one Python file. No external templates, CSS, or JS files.\n"
-        "2.  **HTML as Strings:** All HTML/CSS/JS for templates must be stored in multiline Python string variables (e.g., `HOME_PAGE_TEMPLATE = '''<!DOCTYPE html>...'''`).\n"
-        "3.  **Render with `render_template_string`:** Use `from flask import render_template_string` to render the HTML string variables.\n"
-        "4.  **Command-Line Port:** The application MUST accept the port number as a command-line argument. This is NOT optional. The main script will launch it like `python generated_app.py 8001`. "
-        "   Use this exact structure at the end of the file:\n"
+        "You are an expert Python web developer. Your task is to generate a complete, single-file Flask application "
+        "based on the user's request. The entire application must be contained in a single .py file.\n"
+        "Follow these requirements strictly:\n"
+        "1.  **Flask Application:** The code must be a runnable Flask application.\n"
+        "2.  **Single File:** All code, including HTML templates, CSS, and JavaScript, must be in this single Python file.\n"
+        "3.  **HTML Templates:** Embed HTML as Python strings. Use `flask.render_template_string()`.\n"
+        "4.  **CSS/JS:** Inline CSS within `<style>` tags in the HTML strings. Inline JavaScript within `<script>` tags in the HTML strings. Do not use external files for CSS/JS.\n"
+        "5.  **Routes:** Define Flask routes as requested by the user.\n"
+        "6.  **`app.run`:** Crucially, the script MUST end with a block like this for запуска:\n"
         "   ```python\n"
-        "   import sys\n"
         "   if __name__ == '__main__':\n"
-        "       # Default port is 5000, but it will be overridden by the command line argument.\n"
-        "       port = int(sys.argv[1]) if len(sys.argv) > 1 else 5000\n"
-        "       app.run(host='0.0.0.0', port=port, debug=False)\n"
+        "       # IMPORTANT: The port number MUST be the placeholder {{GENERATED_APP_PORT}}\n"
+        "       # The host should be '0.0.0.0' to be accessible.\n"
+        "       app.run(host='0.0.0.0', port={{GENERATED_APP_PORT}})\n"
         "   ```\n"
-        "5.  **Database:** For any data storage, use a simple in-memory Python list or dictionary defined at the top of the script. Do not use file-based databases like SQLite or JSON files.\n"
-        "6.  **Dependencies:** Only use Flask and standard Python libraries. Do not import libraries that are not commonly available, like pandas or numpy, unless absolutely necessary for the request.\n"
-        "7.  **Output Format:** Directly output ONLY the Python code. Do not include any explanatory text, markdown formatting (like ```python), or anything else before `import os` or after the `app.run(...)` call."
+        "7.  **No External File Dependencies:** The script should not rely on any other local files (e.g., for images, data) unless explicitly creating and managing them (e.g. a simple data.json). Images should preferably be placeholder URLs (e.g., from placehold.co) or base64 encoded if small and necessary.\n"
+        "8.  **Self-Contained Database (Optional):** If the site requires data persistence, it should use a simple mechanism like storing data in Python dictionaries or lists within the script itself. For more complex needs, it could read/write to a single JSON file named `data.json` *in its current working directory if absolutely necessary*, and the script should handle its creation. Avoid complex databases or ORMs.\n"
+        "9.  **Direct Python Output:** Output ONLY the Python code. Do NOT include ```python, markdown, or any explanatory text before or after the code block. The output must start with `import flask` or similar Python code and end with the `app.run` block.\n"
+        "10. **Error Handling & Imports:** Include necessary imports (like `from flask import Flask, render_template_string, request, ...`). Add basic error handling if appropriate for the requested site.\n"
+        "11. **Character Encoding:** Ensure any HTML generated uses UTF-8.\n"
+        "12. **Simplicity:** Favor simplicity and clarity in the generated code."
     )
     
     full_prompt = f"{system_instruction}\n\nUser request: \"{user_prompt}\""
     
+    response_obj = None # Для отладки prompt_feedback
     try:
-        model = genai.GenerativeModel('gemini-1.5-flash-latest') # Using a more recent model
-        response = model.generate_content(full_prompt)
+        # Рекомендуется использовать более новые модели для генерации кода, например, Gemini
+        model = genai.GenerativeModel('gemini-1.5-flash-latest') # Заменено с learnlm-2.0-flash-experimental
+        response_obj = model.generate_content(full_prompt)
+
+        generated_text = ""
+        # Обработка ответа согласно документации Gemini
+        if hasattr(response_obj, 'text') and response_obj.text:
+            generated_text = response_obj.text
+        elif hasattr(response_obj, 'parts') and response_obj.parts:
+            generated_text = "".join(part.text for part in response_obj.parts if hasattr(part, 'text'))
         
-        generated_text = response.text.strip()
-
-        if not generated_text:
-             if hasattr(response, 'prompt_feedback') and response.prompt_feedback.block_reason:
-                 reason = response.prompt_feedback.block_reason
-                 raise ValueError(f"Content generation blocked due to safety policy (reason: {reason}). Please try a different prompt.")
+        if not generated_text.strip():
+             # Проверка на блокировку контента
+             if hasattr(response_obj, 'prompt_feedback') and \
+                hasattr(response_obj.prompt_feedback, 'block_reason') and \
+                response_obj.prompt_feedback.block_reason:
+                 reason = response_obj.prompt_feedback.block_reason
+                 raise ValueError(f"Генерация контента заблокирована из-за политики безопасности (причина: {reason}). Попробуйте другой запрос.")
              else:
-                raise ValueError("The model returned an empty result. Please try rephrasing your prompt.")
-
-        # Clean up potential markdown formatting
-        if generated_text.startswith("```python"):
-            generated_text = generated_text[10:]
-        if generated_text.endswith("```"):
-            generated_text = generated_text[:-3]
-
-        # Basic validation
-        if "from flask import Flask" not in generated_text or "app.run" not in generated_text:
-            raise ValueError("The generated code does not appear to be a valid Flask application.")
-
-        return generated_text.strip()
+                raise ValueError("Модель вернула пустой результат. Попробуйте изменить запрос.")
+        
+        # Очистка от markdown, если модель его добавила
+        clean_text = generated_text.strip()
+        if clean_text.startswith("```python"):
+            clean_text = clean_text[9:] # Удаляем ```python
+            if clean_text.endswith("```"):
+                clean_text = clean_text[:-3] # Удаляем ```
+            generated_text = clean_text.strip()
+        elif clean_text.startswith("```"): # Общий случай для ``` без указания языка
+            clean_text = clean_text[3:]
+            if clean_text.endswith("```"):
+                clean_text = clean_text[:-3]
+            generated_text = clean_text.strip()
+
+        if not (generated_text.startswith("import ") or generated_text.startswith("from ")):
+            logging.warning(f"Generated code might not be pure Python. Preview: {generated_text[:200]}")
+            # Дополнительная проверка не помешает, но AI должен следовать инструкциям
+
+        return generated_text
 
     except Exception as e:
         logging.error(f"Error generating content with GenAI: {e}", exc_info=True)
-        raise ValueError(f"An error occurred while generating the site code: {e}")
-
-
-# --- Process and Port Management ---
-
-def find_free_port():
-    global next_available_port
-    # In a real app, you'd check if the port is actually free.
-    # For this demo, we'll just increment.
-    port = next_available_port
-    next_available_port += 1
-    return port
-
-def cleanup():
-    logging.info("Shutting down generator... terminating all running site processes.")
-    for site_id, site_info in list(running_sites.items()):
-        try:
-            logging.info(f"Terminating site {site_id} (PID: {site_info['process'].pid})")
-            site_info['process'].terminate()
-            site_info['process'].wait(timeout=5) # Wait for graceful termination
-        except subprocess.TimeoutExpired:
-            logging.warning(f"Process for site {site_id} did not terminate gracefully, killing.")
-            site_info['process'].kill()
-        except Exception as e:
-            logging.error(f"Error terminating process for site {site_id}: {e}")
-    logging.info("Cleanup complete.")
-
-atexit.register(cleanup)
+        error_message = str(e)
+        # Более детальная обработка ошибок API
+        if "API key not valid" in error_message:
+             raise ValueError("Внутренняя ошибка конфигурации API Google AI. Проверьте ключ.")
+        elif "Billing account not found" in error_message or "billing account" in error_message.lower():
+             raise ValueError("Проблема с биллингом аккаунта Google Cloud.")
+        elif "Could not find model" in error_message:
+             raise ValueError(f"Модель не найдена или недоступна. Убедитесь, что модель 'gemini-1.5-flash-latest' активна.")
+        elif "resource has been exhausted" in error_message.lower() or "quota" in error_message.lower():
+             raise ValueError("Квота запросов к Google AI исчерпана. Попробуйте позже.")
+        elif "content has been blocked" in error_message.lower() or \
+             (response_obj and hasattr(response_obj, 'prompt_feedback') and \
+              hasattr(response_obj.prompt_feedback, 'block_reason') and \
+              response_obj.prompt_feedback.block_reason):
+            reason = "неизвестна"
+            if response_obj and hasattr(response_obj, 'prompt_feedback') and hasattr(response_obj.prompt_feedback, 'block_reason') and response_obj.prompt_feedback.block_reason:
+                reason = response_obj.prompt_feedback.block_reason
+            elif "safety settings" in error_message.lower():
+                reason = "настройки безопасности"
+            raise ValueError(f"Генерация контента заблокирована (причина: {reason}). Попробуйте другой запрос.")
+        # Ошибка из raise ValueError выше
+        elif "Генерация контента заблокирована" in error_message or "Модель вернула пустой результат" in error_message:
+            raise # Перебрасываем уже обработанную ошибку
+        else:
+             raise ValueError(f"Неизвестная ошибка при генерации Python-кода сайта: {e}")
+
+
+# --- HTML шаблон для главной страницы генератора ---
+# (Взят из Кода 1 и немного адаптирован)
+html_template = """
+<!DOCTYPE html>
+<html lang="ru">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0, viewport-fit=cover">
+    <title>EVA - Генератор Python Сайтов</title>
+    <style>
+        :root {
+            --system-gray-100-light: #f2f2f7;
+            --system-gray-75-light: #f8f8fa;
+            --system-gray-50-light: #ffffff;
+            --system-gray-dark-100-light: #000000;
+            --system-gray-dark-75-light: #1c1c1e;
+            --system-gray-dark-50-light: #3a3a3c;
+            --system-gray-light-75-light: #8e8e93;
+            --system-gray-light-50-light: #aeaeb2;
+            --system-blue-light: #007aff;
+            --system-blue-light-hover: #005ecf;
+            --system-red-light: #ff3b30;
+            --system-separator-light: rgba(60, 60, 67, 0.29);
+            --system-separator-opaque-light: #d1d1d6;
+
+            --system-gray-100-dark: #1c1c1e;
+            --system-gray-75-dark: #2c2c2e;
+            --system-gray-50-dark: #000000;
+            --system-gray-dark-100-dark: #ffffff;
+            --system-gray-dark-75-dark: #f2f2f7;
+            --system-gray-dark-50-dark: #e5e5ea;
+            --system-gray-light-75-dark: #8e8e93;
+            --system-gray-light-50-dark: #636366;
+            --system-blue-dark: #0a84ff;
+            --system-blue-dark-hover: #3b9eff;
+            --system-red-dark: #ff453a;
+            --system-separator-dark: rgba(84, 84, 88, 0.65);
+            --system-separator-opaque-dark: #38383a;
+        }
 
-# --- Flask Routes ---
+        @media (prefers-color-scheme: dark) {
+            :root {
+                --bg-color: var(--system-gray-50-dark);
+                --content-bg: var(--system-gray-100-dark);
+                --text-color: var(--system-gray-dark-100-dark);
+                --secondary-text-color: var(--system-gray-light-75-dark);
+                --tertiary-text-color: var(--system-gray-light-50-dark);
+                --border-color: var(--system-separator-dark);
+                --border-color-opaque: var(--system-separator-opaque-dark);
+                --input-bg: var(--system-gray-75-dark);
+                --primary-color: var(--system-blue-dark);
+                --primary-color-hover: var(--system-blue-dark-hover);
+                --error-color: var(--system-red-dark);
+            }
+        }
 
-# The main UI from Code 1, slightly adapted
-@app.route('/')
-def index():
-    # This is the HTML for the generator's frontend
-    html_template = """
-    <!DOCTYPE html>
-    <html lang="ru">
-    <head>
-        <meta charset="UTF-8">
-        <meta name="viewport" content="width=device-width, initial-scale=1.0, viewport-fit=cover">
-        <title>EVA - Генератор Python/Flask Сайтов</title>
-        <style>
+        @media (prefers-color-scheme: light) {
             :root {
-                --system-gray-100-light: #f2f2f7; --system-gray-75-light: #f8f8fa; --system-gray-50-light: #ffffff; --system-gray-dark-100-light: #000000; --system-gray-dark-75-light: #1c1c1e; --system-gray-dark-50-light: #3a3a3c; --system-gray-light-75-light: #8e8e93; --system-gray-light-50-light: #aeaeb2; --system-blue-light: #007aff; --system-blue-light-hover: #005ecf; --system-red-light: #ff3b30; --system-separator-light: rgba(60, 60, 67, 0.29); --system-separator-opaque-light: #d1d1d6;
-                --system-gray-100-dark: #1c1c1e; --system-gray-75-dark: #2c2c2e; --system-gray-50-dark: #000000; --system-gray-dark-100-dark: #ffffff; --system-gray-dark-75-dark: #f2f2f7; --system-gray-dark-50-dark: #e5e5ea; --system-gray-light-75-dark: #8e8e93; --system-gray-light-50-dark: #636366; --system-blue-dark: #0a84ff; --system-blue-dark-hover: #3b9eff; --system-red-dark: #ff453a; --system-separator-dark: rgba(84, 84, 88, 0.65); --system-separator-opaque-dark: #38383a;
+                --bg-color: var(--system-gray-100-light);
+                --content-bg: var(--system-gray-50-light);
+                --text-color: var(--system-gray-dark-100-light);
+                --secondary-text-color: var(--system-gray-light-75-light);
+                --tertiary-text-color: var(--system-gray-light-50-light);
+                --border-color: var(--system-separator-light);
+                 --border-color-opaque: var(--system-separator-opaque-light);
+                --input-bg: var(--system-gray-75-light);
+                --primary-color: var(--system-blue-light);
+                --primary-color-hover: var(--system-blue-light-hover);
+                --error-color: var(--system-red-light);
             }
-            @media (prefers-color-scheme: dark) { :root { --bg-color: var(--system-gray-50-dark); --content-bg: var(--system-gray-100-dark); --text-color: var(--system-gray-dark-100-dark); --secondary-text-color: var(--system-gray-light-75-dark); --tertiary-text-color: var(--system-gray-light-50-dark); --border-color: var(--system-separator-dark); --border-color-opaque: var(--system-separator-opaque-dark); --input-bg: var(--system-gray-75-dark); --primary-color: var(--system-blue-dark); --primary-color-hover: var(--system-blue-dark-hover); --error-color: var(--system-red-dark); } }
-            @media (prefers-color-scheme: light) { :root { --bg-color: var(--system-gray-100-light); --content-bg: var(--system-gray-50-light); --text-color: var(--system-gray-dark-100-light); --secondary-text-color: var(--system-gray-light-75-light); --tertiary-text-color: var(--system-gray-light-50-light); --border-color: var(--system-separator-light); --border-color-opaque: var(--system-separator-opaque-light); --input-bg: var(--system-gray-75-light); --primary-color: var(--system-blue-light); --primary-color-hover: var(--system-blue-light-hover); --error-color: var(--system-red-light); } }
-            html { height: -webkit-fill-available; }
-            body { font-family: -apple-system, BlinkMacSystemFont, "SF Pro Text", "Segoe UI", Roboto, "Helvetica Neue", Arial, sans-serif; margin: 0; padding: 20px env(safe-area-inset-top) 20px env(safe-area-inset-bottom); background-color: var(--bg-color); color: var(--text-color); display: flex; justify-content: center; align-items: flex-start; min-height: 100vh; min-height: -webkit-fill-available; line-height: 1.45; }
-            .container { background-color: var(--content-bg); padding: 25px 30px 30px 30px; border-radius: 24px; box-shadow: 0 12px 28px rgba(0, 0, 0, 0.08), 0 2px 4px rgba(0,0,0,0.05); max-width: 580px; width: calc(100% - 40px); box-sizing: border-box; margin-top: 30px; }
-            h1 { font-size: 32px; font-weight: 700; text-align: center; margin-bottom: 8px; color: var(--text-color); letter-spacing: -0.5px; }
-            p.subtitle { font-size: 17px; color: var(--secondary-text-color); text-align: center; margin-bottom: 35px; font-weight: 400; }
-            .form-group { margin-bottom: 28px; }
-            label.input-label { display: block; font-weight: 500; margin-bottom: 10px; font-size: 15px; color: var(--secondary-text-color); padding-left: 5px; }
-            textarea#prompt-input { width: 100%; padding: 14px 18px; border: 1px solid var(--border-color-opaque); border-radius: 12px; font-size: 16px; background-color: var(--input-bg); color: var(--text-color); box-sizing: border-box; transition: border-color 0.2s ease, box-shadow 0.2s ease; font-family: inherit; resize: vertical; min-height: 120px; }
-            textarea#prompt-input:focus { border-color: var(--primary-color); box-shadow: 0 0 0 3px color-mix(in srgb, var(--primary-color) 25%, transparent); outline: none; }
-            button#generate-button { width: 100%; padding: 16px; background-color: var(--primary-color); color: white; border: none; border-radius: 12px; font-size: 17px; font-weight: 600; cursor: pointer; transition: background-color 0.2s ease, transform 0.1s ease; margin-top: 15px; }
-            button#generate-button:hover { background-color: var(--primary-color-hover); }
-            button#generate-button:active { transform: scale(0.98); }
-            button#generate-button:disabled { background-color: var(--tertiary-text-color); cursor: not-allowed; }
-            .output-section { margin-top: 35px; } .output-header { display: flex; justify-content: space-between; align-items: center; margin-bottom: 10px; } label#output-label { font-weight: 500; font-size: 15px; color: var(--secondary-text-color); padding-left: 5px; }
-            button#copy-button { background-color: transparent; border: none; color: var(--primary-color); font-size: 14px; font-weight: 500; cursor: pointer; padding: 5px 8px; border-radius: 6px; transition: background-color 0.2s ease, color 0.2s ease; display: none; } button#copy-button:hover { background-color: color-mix(in srgb, var(--primary-color) 15%, transparent); } button#copy-button:active { background-color: color-mix(in srgb, var(--primary-color) 25%, transparent); } button#copy-button.copied { color: #34c759; }
-            #output-container { background-color: var(--input-bg); padding: 18px 20px; border-radius: 12px; min-height: 60px; border: 1px solid var(--border-color); word-wrap: break-word; font-size: 15px; color: var(--text-color); line-height: 1.5; transition: border-color 0.2s ease, background-color 0.2s ease; display: flex; align-items: center; justify-content: center; } #output-container a { color: var(--primary-color); text-decoration: none; font-weight: 500; } #output-container a:hover { text-decoration: underline; }
-            #output-container.loading::before { content: "Генерация и запуск сайта..."; display: block; text-align: center; font-style: italic; color: var(--secondary-text-color); animation: fadePulse 1.8s infinite ease-in-out; }
-            #output-container.error { color: var(--error-color); font-weight: 500; border-color: color-mix(in srgb, var(--error-color) 50%, transparent); background-color: color-mix(in srgb, var(--error-color) 10%, var(--input-bg)); justify-content: flex-start; }
-            @keyframes fadePulse { 0%, 100% { opacity: 0.6; } 50% { opacity: 1; } }
-        </style>
-    </head>
-    <body>
-        <div class="container">
-            <h1>EVA</h1>
-            <p class="subtitle">Генератор Python/Flask сайтов</p>
-            <form id="generate-form">
-                <div class="form-group">
-                    <label for="prompt-input" class="input-label">Опишите сайт, который вы хотите создать</label>
-                    <textarea id="prompt-input" name="prompt" rows="6" required placeholder="Например: создай сайт-визитку для фотографа. На главной странице должны быть галерея работ (3-4 фото-заглушки), раздел 'Обо мне' и контактная форма."></textarea>
-                </div>
-                <button type="submit" id="generate-button">Создать и запустить сайт</button>
-            </form>
-            <div class="output-section">
-                 <div class="output-header">
-                     <label id="output-label">Ссылка на запущенный сайт</label>
-                     <button id="copy-button">Копировать</button>
-                 </div>
-                <div id="output-container" aria-live="polite"></div>
+        }
+
+        html { height: -webkit-fill-available; }
+        body {
+            font-family: -apple-system, BlinkMacSystemFont, "SF Pro Text", "Segoe UI", Roboto, "Helvetica Neue", Arial, sans-serif;
+            margin: 0; padding: 20px env(safe-area-inset-top) 20px env(safe-area-inset-bottom);
+            background-color: var(--bg-color); color: var(--text-color);
+            display: flex; justify-content: center; align-items: flex-start;
+            min-height: 100vh; min-height: -webkit-fill-available;
+            line-height: 1.45; -webkit-font-smoothing: antialiased; -moz-osx-font-smoothing: grayscale;
+        }
+        .container {
+            background-color: var(--content-bg); padding: 25px 30px 30px 30px;
+            border-radius: 24px; box-shadow: 0 12px 28px rgba(0, 0, 0, 0.08), 0 2px 4px rgba(0,0,0,0.05);
+            max-width: 580px; width: calc(100% - 40px); box-sizing: border-box; margin-top: 30px;
+        }
+        h1 { font-size: 32px; font-weight: 700; text-align: center; margin-bottom: 8px; color: var(--text-color); letter-spacing: -0.5px; }
+        p.subtitle { font-size: 17px; color: var(--secondary-text-color); text-align: center; margin-bottom: 35px; font-weight: 400; }
+        .form-group { margin-bottom: 28px; }
+        label.input-label { display: block; font-weight: 500; margin-bottom: 10px; font-size: 15px; color: var(--secondary-text-color); padding-left: 5px; }
+        textarea#prompt-input {
+            width: 100%; padding: 14px 18px; border: 1px solid var(--border-color-opaque); border-radius: 12px;
+            font-size: 16px; background-color: var(--input-bg); color: var(--text-color);
+            box-sizing: border-box; transition: border-color 0.2s ease, box-shadow 0.2s ease;
+            font-family: inherit; resize: vertical; min-height: 120px;
+        }
+        textarea#prompt-input:focus {
+             border-color: var(--primary-color);
+             box-shadow: 0 0 0 3px color-mix(in srgb, var(--primary-color) 25%, transparent);
+             outline: none;
+        }
+        button#generate-button {
+            width: 100%; padding: 16px; background-color: var(--primary-color); color: white;
+            border: none; border-radius: 12px; font-size: 17px; font-weight: 600; cursor: pointer;
+            transition: background-color 0.2s ease, transform 0.1s ease; margin-top: 15px;
+        }
+        button#generate-button:hover { background-color: var(--primary-color-hover); }
+        button#generate-button:active { transform: scale(0.98); }
+        button#generate-button:disabled { background-color: var(--tertiary-text-color); cursor: not-allowed; }
+        .output-section { margin-top: 35px; }
+        .output-header { display: flex; justify-content: space-between; align-items: center; margin-bottom: 10px; }
+        label#output-label { font-weight: 500; font-size: 15px; color: var(--secondary-text-color); padding-left: 5px; }
+        button#copy-button {
+            background-color: transparent; border: none; color: var(--primary-color);
+            font-size: 14px; font-weight: 500; cursor: pointer; padding: 5px 8px;
+            border-radius: 6px; transition: background-color 0.2s ease, color 0.2s ease; display: none;
+        }
+        button#copy-button:hover { background-color: color-mix(in srgb, var(--primary-color) 15%, transparent); }
+        button#copy-button:active { background-color: color-mix(in srgb, var(--primary-color) 25%, transparent); }
+        button#copy-button.copied { color: #34c759; }
+        @media (prefers-color-scheme: dark) { button#copy-button.copied { color: #30d158; } }
+        #output-container {
+            background-color: var(--input-bg); padding: 18px 20px; border-radius: 12px;
+            min-height: 60px; border: 1px solid var(--border-color); word-wrap: break-word;
+            font-size: 15px; color: var(--text-color); line-height: 1.5;
+            transition: border-color 0.2s ease, background-color 0.2s ease;
+            display: flex; align-items: center; justify-content: center;
+        }
+        #output-container a { color: var(--primary-color); text-decoration: none; font-weight: 500; }
+        #output-container a:hover { text-decoration: underline; }
+        #output-container.loading::before {
+            content: "Генерация сайта..."; display: block; text-align: center;
+            font-style: italic; color: var(--secondary-text-color); animation: fadePulse 1.8s infinite ease-in-out;
+        }
+        #output-container.error {
+            color: var(--error-color); font-weight: 500;
+            border-color: color-mix(in srgb, var(--error-color) 50%, transparent);
+            background-color: color-mix(in srgb, var(--error-color) 10%, var(--input-bg));
+            justify-content: flex-start;
+        }
+        @keyframes fadePulse { 0%, 100% { opacity: 0.6; } 50% { opacity: 1; } }
+        @media (max-width: 620px) {
+            body { padding: 15px env(safe-area-inset-top) 15px env(safe-area-inset-bottom); align-items: flex-start; }
+            .container { padding: 20px 20px 25px 20px; margin-top: 15px; border-radius: 20px; width: calc(100% - 30px); }
+            h1 { font-size: 28px; }
+            p.subtitle { font-size: 16px; margin-bottom: 25px; }
+            .form-group { margin-bottom: 22px; }
+            textarea#prompt-input { padding: 12px 15px; min-height: 100px; }
+            button#generate-button { padding: 15px; font-size: 16px; }
+            #output-container { padding: 15px 18px; font-size: 14px; min-height: 50px; }
+            .output-section { margin-top: 30px; }
+        }
+    </style>
+</head>
+<body>
+    <div class="container">
+        <h1>EVA</h1>
+        <p class="subtitle">Генератор Python Flask сайтов</p>
+
+        <form id="generate-form">
+            <div class="form-group">
+                <label for="prompt-input" class="input-label">Опишите сайт, который вы хотите создать (на Python/Flask)</label>
+                <textarea id="prompt-input" name="prompt" rows="6" required placeholder="Например: создай простой одностраничный сайт-визитку на Flask. На главной странице должен быть заголовок 'Мой Сайт', текст 'Добро пожаловать!' и текущее время."></textarea>
             </div>
+            <button type="submit" id="generate-button">Создать сайт</button>
+        </form>
+
+        <div class="output-section">
+             <div class="output-header">
+                 <label id="output-label">Ссылка для запуска сайта</label>
+                 <button id="copy-button">Копировать</button>
+             </div>
+            <div id="output-container" aria-live="polite"></div>
         </div>
-        <script>
-            const form = document.getElementById('generate-form');
-            const outputContainer = document.getElementById('output-container');
-            const generateButton = document.getElementById('generate-button');
-            const copyButton = document.getElementById('copy-button');
-
-            form.addEventListener('submit', async (event) => {
-                event.preventDefault();
-                const formData = new FormData(form);
-                if (!formData.get('prompt').trim()) {
-                    showError("Пожалуйста, опишите сайт."); return;
-                }
-                generateButton.disabled = true; generateButton.textContent = 'Генерация...';
-                outputContainer.innerHTML = ''; outputContainer.classList.add('loading');
-                outputContainer.classList.remove('error'); copyButton.style.display = 'none';
-
-                try {
-                    const response = await fetch('/generate', { method: 'POST', body: formData });
-                    const result = await response.json();
-                    if (!response.ok) { throw new Error(result.error || `Ошибка сервера: ${response.status}`); }
-
-                    if (result.site_url) {
-                        const link = document.createElement('a');
-                        link.href = result.site_url;
-                        link.textContent = result.site_url;
-                        link.target = "_blank";
-                        outputContainer.innerHTML = '';
-                        outputContainer.appendChild(link);
-                        copyButton.style.display = 'block';
-                        copyButton.dataset.copyText = result.site_url;
-                    } else {
-                        showError(result.error || "Не удалось получить URL сайта.");
-                    }
-                } catch (error) {
-                    console.error("Fetch Error:", error);
-                    showError(`Ошибка: ${error.message}`);
-                } finally {
-                    generateButton.disabled = false; generateButton.textContent = 'Создать и запустить сайт';
-                    outputContainer.classList.remove('loading');
+    </div>
+
+    <script>
+        const form = document.getElementById('generate-form');
+        const promptInput = document.getElementById('prompt-input');
+        const outputContainer = document.getElementById('output-container');
+        const generateButton = document.getElementById('generate-button');
+        const copyButton = document.getElementById('copy-button');
+
+        form.addEventListener('submit', async (event) => {
+            event.preventDefault();
+            if (!promptInput.value.trim()) {
+                showError("Пожалуйста, опишите сайт, который вы хотите создать.");
+                return;
+            }
+            const formData = new FormData(form);
+            generateButton.disabled = true;
+            generateButton.textContent = 'Генерация...';
+            outputContainer.innerHTML = '';
+            outputContainer.classList.add('loading');
+            outputContainer.classList.remove('error');
+            copyButton.style.display = 'none';
+            copyButton.textContent = 'Копировать';
+            copyButton.classList.remove('copied');
+
+            try {
+                const response = await fetch('/generate', { method: 'POST', body: formData });
+                const result = await response.json();
+
+                if (!response.ok) {
+                    throw new Error(result.error || `Ошибка сервера: ${response.status}`);
                 }
-            });
 
-            copyButton.addEventListener('click', () => {
-                navigator.clipboard.writeText(copyButton.dataset.copyText).then(() => {
-                    copyButton.textContent = 'Скопировано!';
-                    setTimeout(() => { copyButton.textContent = 'Копировать'; }, 1500);
-                });
+                if (result.site_run_url) { // Изменено с site_url на site_run_url
+                    const link = document.createElement('a');
+                    link.href = result.site_run_url;
+                    link.textContent = "Запустить и открыть сгенерированный сайт";
+                    link.target = "_blank"; // Откроет в новой вкладке, что приведет к редиректу
+                    outputContainer.innerHTML = '';
+                    outputContainer.appendChild(link);
+                    copyButton.style.display = 'block';
+                    copyButton.dataset.copyText = window.location.origin + result.site_run_url;
+                } else if (result.error) {
+                    showError(result.error);
+                } else {
+                    showError("Не удалось получить ссылку на сайт. Ответ сервера не содержит URL.");
+                }
+            } catch (error) {
+                console.error("Fetch Error:", error);
+                showError(`Ошибка: ${error.message}`);
+            } finally {
+                generateButton.disabled = false;
+                generateButton.textContent = 'Создать сайт';
+                outputContainer.classList.remove('loading');
+            }
+        });
+
+        copyButton.addEventListener('click', () => {
+            const textToCopy = copyButton.dataset.copyText;
+            if (!textToCopy) return;
+            navigator.clipboard.writeText(textToCopy).then(() => {
+                copyButton.textContent = 'Скопировано!';
+                copyButton.classList.add('copied');
+                setTimeout(() => {
+                     copyButton.textContent = 'Копировать';
+                     copyButton.classList.remove('copied');
+                }, 1500);
+            }).catch(err => {
+                console.error('Ошибка копирования: ', err);
+                copyButton.textContent = 'Ошибка';
+                 setTimeout(() => { copyButton.textContent = 'Копировать'; }, 1500);
             });
+        });
+
+        function showError(message) {
+             outputContainer.innerHTML = ''; 
+             const errorMessageElement = document.createElement('span');
+             errorMessageElement.textContent = message;
+             outputContainer.appendChild(errorMessageElement);
+             outputContainer.classList.add('error');
+             outputContainer.classList.remove('loading');
+             copyButton.style.display = 'none';
+        }
+    </script>
+</body>
+</html>
+"""
 
-            function showError(message) {
-                 outputContainer.innerHTML = `<span>${message}</span>`;
-                 outputContainer.classList.add('error');
-                 outputContainer.classList.remove('loading');
-                 copyButton.style.display = 'none';
-            }
-        </script>
-    </body>
-    </html>
-    """
+# --- Маршруты Flask основного приложения ---
+@app.route('/')
+def index():
     return Response(html_template, mimetype='text/html')
 
 @app.route('/generate', methods=['POST'])
 def handle_generate():
-    user_prompt = request.form.get('prompt')
+    if 'prompt' not in request.form:
+        return jsonify({"error": "Текстовый запрос (промпт) не найден."}), 400
+
+    user_prompt = request.form['prompt']
     if not user_prompt or not user_prompt.strip():
-        return jsonify({"error": "Текстовый запрос не может быть пустым."}), 400
+         return jsonify({"error": "Текстовый запрос не может быть пустым."}), 400
 
     try:
-        # 1. Generate the Python code for the Flask app
-        flask_app_code = generate_flask_app_code_from_prompt(user_prompt)
+        python_code = generate_full_python_site_from_prompt(user_prompt)
         
-        # 2. Save the generated code to a .py file
-        site_id = str(uuid.uuid4())
-        filename = f"{site_id}.py"
-        filepath = os.path.join(GENERATED_SITES_DIR, filename)
+        if not python_code or not python_code.strip():
+            return jsonify({"error": "Сгенерированный Python-код пуст."}), 500
+
+        # Генерируем уникальное имя файла для .py сайта
+        site_filename = f"site_{uuid.uuid4().hex[:12]}.py"
+        local_filepath = os.path.join(SITES_CACHE_DIR, site_filename)
         
-        with open(filepath, "w", encoding="utf-8") as f:
-            f.write(flask_app_code)
+        with open(local_filepath, "w", encoding="utf-8") as f:
+            f.write(python_code)
+        logging.info(f"Сгенерированный сайт сохранен локально: {local_filepath}")
             
-        # 3. Find a free port and launch the new Flask app in a separate process
-        port = find_free_port()
-        
-        # Pass the HF token to the child process environment if needed
-        env = os.environ.copy()
-        if HF_TOKEN_WRITE:
-            env["HF_TOKEN"] = HF_TOKEN_WRITE
+        # Загружаем на Hugging Face
+        repo_filepath = f"sites/{site_filename}" # Путь в репозитории HF
+        if upload_file_to_hf(local_filepath, repo_filepath):
+            logging.info(f"Сайт {site_filename} успешно загружен на Hugging Face: {REPO_ID}/{repo_filepath}")
+        else:
+            # Не фатально, сайт все еще может работать из локального кэша, но не будет синхронизирован
+            logging.warning(f"Не удалось загрузить сайт {site_filename} на Hugging Face.")
 
-        process = subprocess.Popen(
-            [sys.executable, filepath, str(port)],
-            stdout=subprocess.PIPE,
-            stderr=subprocess.PIPE,
-            env=env
-        )
-        
-        # Give it a moment to start up or fail
-        time.sleep(3)
-        
-        # Check if the process has already exited with an error
-        if process.poll() is not None:
-             stderr_output = process.stderr.read().decode('utf-8', errors='ignore')
-             raise RuntimeError(f"Generated app failed to start. Error:\n{stderr_output}")
-
-        # 4. Store the running process and update state
-        running_sites[site_id] = {
-            'process': process,
-            'port': port,
-            'prompt': user_prompt,
-            'id': site_id
-        }
-        
-        # 5. Persist the state
-        sites_data['sites'][site_id] = {'port': port, 'prompt': user_prompt, 'id': site_id}
-        save_data()
-        
-        # 6. Return the URL to the user
-        host = request.host.split(':')[0]
-        site_url = f"http://{host}:{port}"
-        
-        logging.info(f"Successfully generated and launched site {site_id} on port {port}")
-        return jsonify({"site_url": site_url})
 
-    except (ValueError, RuntimeError) as e:
-        logging.error(f"Error during generation or launch: {e}", exc_info=True)
-        return jsonify({"error": str(e)}), 400
-    except Exception as e:
-        logging.error(f"Unexpected server error: {e}", exc_info=True)
-        return jsonify({"error": f"An unexpected internal error occurred: {e}"}), 500
+        # Ссылка для запуска сайта (которая приведет к редиректу на его порт)
+        site_run_url = url_for('run_generated_site', filename=site_filename, _external=False)
+        return jsonify({"site_run_url": site_run_url})
 
-# --- Main Application Start ---
-if __name__ == '__main__':
-    if not os.path.exists(GENERATED_SITES_DIR):
-        os.makedirs(GENERATED_SITES_DIR)
+    except ValueError as ve: # Ошибки валидации или от GenAI
+        logging.error(f"ValueError during site generation: {str(ve)}")
+        return jsonify({"error": str(ve)}), 400 # 400 Bad Request или 422 Unprocessable Entity
+    except Exception as e:
+        logging.error(f"Unexpected error during site generation: {e}", exc_info=True)
+        return jsonify({"error": f"Внутренняя ошибка сервера при генерации сайта: {e}"}), 500
 
-    logging.info("Application starting up...")
+@app.route('/run_site/<filename>')
+def run_generated_site(filename):
+    if not filename.endswith(".py"):
+        return "Неверное имя файла сайта.", 400
     
-    # Attempt to load state from HF on startup
-    load_data()
-    # Note: We are not auto-relaunching sites on startup to keep it simple,
-    # but the state is preserved in generator_data.json. An admin panel could show them.
-
-    if HF_TOKEN_WRITE:
-        backup_thread = threading.Thread(target=periodic_backup, daemon=True)
-        backup_thread.start()
-        logging.info("Periodic state backup thread started.")
+    logging.info(f"Запрос на запуск сайта: {filename}")
+    site_host, site_port = start_generated_site(filename)
+
+    if site_host and site_port:
+        # Формируем URL для редиректа. request.host содержит <host>:<port> основного приложения.
+        # Нам нужен только хост.
+        main_app_host = request.host.split(':')[0]
+        # Если основной сервер слушает на 0.0.0.0, браузер обычно обращается через localhost или конкретный IP.
+        # Используем main_app_host, чтобы редирект был на тот же хост, с которого пришел запрос.
+        # Если site_host это '127.0.0.1', а main_app_host это '0.0.0.0' или реальный IP,
+        # редирект на 127.0.0.1 может сработать только локально.
+        # Для простоты, если main_app_host '0.0.0.0', заменим на '127.0.0.1' для URL редиректа.
+        if main_app_host == '0.0.0.0':
+            redirect_host = '127.0.0.1'
+        else:
+            redirect_host = main_app_host
+
+        redirect_url = f"http://{redirect_host}:{site_port}/"
+        logging.info(f"Редирект на запущенный сайт: {redirect_url}")
+        return redirect(redirect_url, code=302)
     else:
-        logging.warning("Periodic backup will NOT run (HF_TOKEN for writing not set).")
+        logging.error(f"Не удалось запустить сайт {filename}.")
+        # Можно вернуть страницу с ошибкой
+        return f"Не удалось запустить сайт {filename}. Проверьте логи сервера.", 500
+
 
-    logging.info("Starting main generator Flask app on host 0.0.0.0 and port 7860")
-    app.run(host='0.0.0.0', port=7860, debug=False)
+if __name__ == '__main__':
+    # Проверка наличия HF_TOKEN_WRITE для загрузки на HF
+    if not HF_TOKEN_WRITE:
+        logging.warning("Переменная окружения HF_TOKEN (с правом записи) не установлена. "
+                        "Сгенерированные сайты не будут загружаться на Hugging Face.")
+    
+    # Порт для основного приложения-генератора
+    main_app_port = 7860
+    logging.info(f"Основное приложение генератора запускается на порту {main_app_port}")
+    app.run(host='0.0.0.0', port=main_app_port, debug=False)