Testgen

Sleeping

App Files Files Community

Kgshop commited on Jun 22, 2025

Commit

72c3aff

verified ·

1 Parent(s): 484ce69

Update app.py

Browse files

Files changed (1) hide show

app.py +397 -393

app.py CHANGED Viewed

@@ -1,503 +1,507 @@
 import os
 import uuid
-import sys
-import subprocess
-import time
 import json
 import logging
 import threading
-from datetime import datetime
-import signal
 import atexit
-from flask import Flask, request, jsonify, render_template_string, redirect, url_for, flash
 import google.generativeai as genai
 from huggingface_hub import HfApi, hf_hub_download
 from huggingface_hub.utils import RepositoryNotFoundError, HfHubHTTPError
-from dotenv import load_dotenv
 import requests
-# --- 1. Конфигурация и Инициализация ---
 load_dotenv()
 app = Flask(__name__)
-app.secret_key = 'super_secret_key_for_flask_site_generator_12345'
-# Конфигурация для генератора сайтов
-API_KEY_INTERNAL = "AIzaSyArKidc4o0MwbaCFKStlb2q2AwNg6Pnqns" # Ваш ключ для Google AI
-GENERATED_SITES_DIR = 'generated_sites'
-# Конфигурация для Hugging Face Sync
-REPO_ID = "Kgshop/testsynk"  # !!! ИЗМЕНИТЕ НА ВАШ РЕПОЗИТОРИЙ !!!
-HF_TOKEN = os.getenv("HF_TOKEN")
-HF_TOKEN_READ = os.getenv("HF_TOKEN_READ") or HF_TOKEN
-DATA_FILE = 'sites_db.json'
 SYNC_FILES = [DATA_FILE]
-# Управление процессами
-running_sites = {} # { "site_id": subprocess_object }
-BASE_PORT = 7861
-# Настройка логирования
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
-if not os.path.exists(GENERATED_SITES_DIR):
-    os.makedirs(GENERATED_SITES_DIR)
-# --- 2. Функции для работы с Hugging Face (из Кода 2) ---
-def download_db_from_hf(specific_file=None, retries=3, delay=5):
-    if not HF_TOKEN_READ:
-        logging.warning("HF_TOKEN_READ not set. Download might fail for private repos.")
     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
     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})...")
                 hf_hub_download(
-                    repo_id=REPO_ID, filename=file_name, repo_type="dataset", token=HF_TOKEN_READ,
                     local_dir=".", local_dir_use_symlinks=False, force_download=True, resume_download=False
                 )
                 logging.info(f"Successfully downloaded {file_name}.")
-                success = True
-                break
             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 not os.path.exists(file_name):
-                        if file_name == DATA_FILE:
-                            with open(file_name, 'w', encoding='utf-8') as f:
-                                json.dump({'sites': {}}, f)
-                    success = True # Считаем успехом, так как создали пустой файл
-                    break
-                logging.error(f"HTTP error downloading {file_name}: {e}. Retrying...")
             except Exception as e:
-                logging.error(f"Unexpected error downloading {file_name}: {e}. Retrying...", exc_info=True)
             if attempt < retries: time.sleep(delay)
-        if not success:
-            all_successful = False
-            logging.error(f"Failed to download {file_name} after {retries + 1} attempts.")
-    return all_successful
 def upload_db_to_hf(specific_file=None):
-    if not HF_TOKEN:
         logging.warning("HF_TOKEN (for writing) not set. Skipping upload.")
         return
     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,
                     commit_message=f"Sync {file_name} {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}"
                 )
-                logging.info(f"File {file_name} successfully uploaded.")
-            else:
-                logging.warning(f"File {file_name} not found locally, skipping upload.")
     except Exception as e:
-        logging.error(f"Error during Hugging Face upload: {e}", exc_info=True)
 def periodic_backup():
     while True:
         time.sleep(1800) # 30 минут
         logging.info("Starting periodic backup...")
-        upload_db_to_hf()
 def load_data():
-    if not os.path.exists(DATA_FILE):
-        logging.warning(f"{DATA_FILE} not found locally. Attempting to download.")
-        download_db_from_hf(DATA_FILE)
-    try:
-        with open(DATA_FILE, 'r', encoding='utf-8') as f:
-            data = json.load(f)
-        if 'sites' not in data: data['sites'] = {}
-        return data
-    except (FileNotFoundError, json.JSONDecodeError):
-        logging.error(f"Failed to load or parse {DATA_FILE}. Using empty structure.")
-        return {'sites': {}}
 def save_data(data):
     try:
         with open(DATA_FILE, 'w', encoding='utf-8') as f:
-            json.dump(data, f, ensure_ascii=False, indent=4)
-        logging.info(f"Data saved to {DATA_FILE}. Uploading to HF.")
-        upload_db_to_hf(specific_file=DATA_FILE)
     except Exception as e:
-        logging.error(f"Error saving data: {e}", exc_info=True)
-# --- 3. Функции Управления Сайтами и Процессами ---
-def find_available_port(start_port):
-    data = load_data()
-    used_ports = {site['port'] for site in data['sites'].values() if 'port' in site and site.get('status') == 'running'}
-    port = start_port
-    while port in used_ports:
-        port += 1
-    return port
-def start_site_process(site_id, filepath):
-    """Запускает сайт в отдельном процессе и возвращает объект процесса и порт."""
-    port = find_available_port(BASE_PORT)
-    env = os.environ.copy()
-    env['PORT'] = str(port)
-    try:
-        process = subprocess.Popen(
-            [sys.executable, filepath],
-            env=env,
-            stdout=subprocess.PIPE, # Можно перенаправить в лог-файлы
-            stderr=subprocess.PIPE
-        )
-        time.sleep(2) # Даем время на запуск
-        # Проверяем, жив ли процесс
-        if process.poll() is None:
-            running_sites[site_id] = process
-            return process, port
-        else:
-            stdout, stderr = process.communicate()
-            logging.error(f"Failed to start site {site_id}. Exit code: {process.returncode}")
-            logging.error(f"Stderr: {stderr.decode('utf-8', errors='ignore')}")
-            return None, None
-    except Exception as e:
-        logging.error(f"Exception while starting site {site_id}: {e}", exc_info=True)
-        return None, None
-def stop_site_process(site_id):
-    """Останавливает процесс сайта."""
-    data = load_data()
-    site_info = data['sites'].get(site_id)
-    if site_id in running_sites:
-        process = running_sites[site_id]
-        logging.info(f"Stopping running process for site {site_id} with PID {process.pid}")
-        process.terminate()
-        try:
-            process.wait(timeout=5)
-        except subprocess.TimeoutExpired:
-            process.kill()
-        del running_sites[site_id]
-        return True
-    # Если процесс не в памяти, но в БД есть PID (после перезапуска)
-    if site_info and 'pid' in site_info and site_info.get('status') == 'running':
-        pid = site_info['pid']
-        logging.warning(f"Process for site {site_id} not in memory. Attempting to kill by PID {pid}")
-        try:
-            os.kill(pid, signal.SIGTERM)
-            return True
-        except ProcessLookupError:
-            logging.warning(f"Process with PID {pid} not found.")
-            return True # Процесса и так нет
-        except Exception as e:
-            logging.error(f"Failed to kill process with PID {pid}: {e}")
-            return False
-    return True
-def cleanup_processes():
-    """Останавливает все дочерние процессы при выходе."""
-    logging.info("Shutting down. Terminating all running site processes...")
-    for site_id, process in list(running_sites.items()):
-        logging.info(f"Terminating site {site_id} (PID: {process.pid})")
-        process.terminate()
-    logging.info("Cleanup complete.")
-atexit.register(cleanup_processes)
-# --- 4. Генерация Кода с Помощью ИИ (доработано) ---
-def generate_website_code_from_prompt(user_prompt):
     try:
         genai.configure(api_key=API_KEY_INTERNAL)
     except Exception as e:
-        raise ValueError(f"Ошибка конфигурации Google AI: {e}")
     if not user_prompt or not user_prompt.strip():
-        raise ValueError("Промпт не может быть пустым.")
     system_instruction = (
-        "You are an expert Python Flask web developer. Your mission is to generate a complete, single-file Python Flask web application based on the user's request.\n"
-        "The generated script MUST adhere to the following strict rules:\n"
-        "1.  **Single File:** The entire application, including Flask routes and HTML templates, must be in one single `.py` file.\n"
-        "2.  **Flask Framework:** The application must use the Flask framework. It should start with `from flask import Flask, render_template_string, ...`.\n"
-        "3.  **Inline Templates:** All HTML templates must be embedded directly into the Python script as multiline string variables. For example: `HOME_PAGE_TEMPLATE = '''<!DOCTYPE html>...'''`. Use `render_template_string()` to render them.\n"
-        "4.  **No External Files:** The script must NOT attempt to read from or write to any local files (like .css, .js, .json). All data should be pre-populated in-memory (e.g., as a list of dictionaries). All CSS and JS must be inside the HTML templates within `<style>` and `<script>` tags.\n"
-        "5.  **Port from Environment:** The application must get its port number from an environment variable named `PORT`. The code to do this is crucial: `port = int(os.environ.get('PORT', 5000))`.\n"
-        "6.  **Startup Command:** The final executable line of the script MUST be `if __name__ == '__main__': app.run(host='0.0.0.0', port=port)`. This is non-negotiable.\n"
-        "7.  **Dependencies:** Assume only Flask is installed (`pip install Flask`). Do not use any other external libraries that are not part of the Python standard library.\n"
-        "8.  **Output Format:** Your output must be ONLY the raw Python code. Do not wrap it in markdown like ```python or add any explanatory text before or after the code block.\n"
-        "\n"
-        "Example of a valid structure:\n"
-        "```python\n"
-        "import os\n"
-        "from flask import Flask, render_template_string\n\n"
-        "app = Flask(__name__)\n\n"
-        "PRODUCTS_DATA = [{'id': 1, 'name': 'Toy Car'}, {'id': 2, 'name': 'Plush Bear'}]\n\n"
-        "HOME_TEMPLATE = '''\n"
-        "<!DOCTYPE html><html>...<body>{% for product in products %}...{% endfor %}</body></html>'''\n\n"
-        "@app.route('/')\n"
-        "def home():\n"
-        "    return render_template_string(HOME_TEMPLATE, products=PRODUCTS_DATA)\n\n"
-        "if __name__ == '__main__':\n"
-        "    port = int(os.environ.get('PORT', 5000))\n"
-        "    app.run(host='0.0.0.0', port=port)\n"
-        "```"
     )
     full_prompt = f"{system_instruction}\n\nUser request: \"{user_prompt}\""
     try:
-        model = genai.GenerativeModel('gemini-1.5-flash-latest') # Используем современную модель
         response = model.generate_content(full_prompt)
-        generated_code = response.text
-        # Очистка от markdown, если модель его добавила
-        if generated_code.strip().startswith("```python"):
-            generated_code = generated_code.strip()[9:]
-            if generated_code.strip().endswith("```"):
-                generated_code = generated_code.strip()[:-3]
-        if not generated_code.strip():
-            raise ValueError("Модель вернула пустой результат.")
-        return generated_code.strip()
     except Exception as e:
-        logging.error(f"Error generating content with GenAI: {e}", exc_info=True)
-        raise ValueError(f"Ошибка при генерации кода сайта: {e}")
-# --- 5. Flask Routes ---
-ADMIN_TEMPLATE = """
-<!DOCTYPE html>
-<html lang="ru">
-<head>
-    <meta charset="UTF-8">
-    <meta name="viewport" content="width=device-width, initial-scale=1.0">
-    <title>Панель Управления Сайтами</title>
-    <style>
-        body { font-family: -apple-system, BlinkMacSystemFont, "Segoe UI", Roboto, Helvetica, Arial, sans-serif; background-color: #f0f2f5; color: #1c1c1e; margin: 0; padding: 20px; }
-        .container { max-width: 900px; margin: 20px auto; background: #fff; border-radius: 12px; box-shadow: 0 4px 12px rgba(0,0,0,0.1); }
-        .header { padding: 20px 25px; border-bottom: 1px solid #e5e5e5; }
-        .header h1 { margin: 0; font-size: 24px; }
-        .content { padding: 25px; }
-        .form-section { margin-bottom: 30px; padding: 20px; background: #f7f7f7; border-radius: 8px; }
-        textarea { width: 100%; padding: 12px; border-radius: 6px; border: 1px solid #ccc; font-size: 16px; min-height: 80px; box-sizing: border-box; }
-        button { background-color: #007aff; color: white; border: none; padding: 12px 20px; font-size: 16px; font-weight: 500; border-radius: 8px; cursor: pointer; transition: background-color 0.2s; }
-        button:hover { background-color: #005ecf; }
-        button:disabled { background-color: #aaa; cursor: not-allowed; }
-        .site-list { margin-top: 20px; }
-        .site-item { display: grid; grid-template-columns: 1fr auto; gap: 15px; align-items: center; padding: 15px; border: 1px solid #e5e5e5; border-radius: 8px; margin-bottom: 10px; background: #fff; }
-        .site-info p { margin: 0 0 8px; }
-        .site-info p:last-child { margin-bottom: 0; }
-        .site-info .prompt { font-style: italic; color: #666; font-size: 14px; }
-        .site-actions { display: flex; gap: 10px; flex-wrap: wrap; }
-        .site-actions .btn { padding: 8px 12px; font-size: 14px; text-decoration: none; border-radius: 6px; color: white; display: inline-block; }
-        .btn-start { background-color: #34c759; } .btn-start:hover { background-color: #2ea34a; }
-        .btn-stop { background-color: #ff3b30; } .btn-stop:hover { background-color: #d9332a; }
-        .btn-delete { background-color: #8e8e93; } .btn-delete:hover { background-color: #6c6c70; }
-        .status { padding: 4px 8px; border-radius: 12px; font-size: 12px; font-weight: 500; }
-        .status.running { background-color: #d4edda; color: #155724; }
-        .status.stopped { background-color: #f8d7da; color: #721c24; }
-        .flash { padding: 15px; margin-bottom: 20px; border-radius: 8px; }
-        .flash.success { background-color: #d4edda; color: #155724; border: 1px solid #c3e6cb; }
-        .flash.error { background-color: #f8d7da; color: #721c24; border: 1px solid #f5c6cb; }
-    </style>
-</head>
-<body>
-    <div class="container">
-        <div class="header"><h1>Панель Управления Сайтами</h1></div>
-        <div class="content">
-            {% with messages = get_flashed_messages(with_categories=true) %}
-                {% if messages %}
-                    {% for category, message in messages %}
-                        <div class="flash {{ category }}">{{ message }}</div>
-                    {% endfor %}
-                {% endif %}
-            {% endwith %}
-            <div class="form-section">
-                <h2>Создать новый сайт</h2>
-                <form action="{{ url_for('handle_generate') }}" method="POST">
-                    <textarea name="prompt" required placeholder="Например: создай сайт-визитку для фотографа Анны. На сайте должны быть разделы 'Главная', 'Портфолио' с 6 примерами работ и 'Контакты' с формой обратной связи."></textarea><br><br>
-                    <button type="submit">Сгенерировать и запустить</button>
-                </form>
-            </div>
-            <h2>Управление Сайтами</h2>
-            <div class="site-list">
-                {% if sites %}
-                    {% for site_id, site in sites.items()|sort(attribute='1.created_at', reverse=True) %}
-                    <div class="site-item">
-                        <div class="site-info">
-                            <p><strong>ID:</strong> {{ site_id }}</p>
-                            <p class="prompt"><strong>Промпт:</strong> {{ site.prompt }}</p>
-                            <p><strong>Статус:</strong> <span class="status {{ 'running' if site.status == 'running' else 'stopped' }}">{{ site.status }}</span>
-                            {% if site.status == 'running' and site.port %}
-                                | <strong>Порт:</strong> {{ site.port }} | <a href="http://{{ request.host.split(':')[0] }}:{{ site.port }}" target="_blank">Открыть сайт →</a>
-                            {% endif %}
-                            </p>
-                        </div>
-                        <div class="site-actions">
-                            {% if site.status == 'stopped' %}
-                            <a href="{{ url_for('start_site', site_id=site_id) }}" class="btn btn-start">Запустить</a>
-                            {% else %}
-                            <a href="{{ url_for('stop_site', site_id=site_id) }}" class="btn btn-stop">Остановить</a>
-                            {% endif %}
-                            <a href="{{ url_for('delete_site', site_id=site_id) }}" class="btn btn-delete" onclick="return confirm('Вы уверены, что хотите удалить этот сайт и все его файлы?');">Удалить</a>
-                        </div>
-                    </div>
-                    {% endfor %}
-                {% else %}
-                    <p>Пока не создано ни одного сайта.</p>
-                {% endif %}
-            </div>
-        </div>
-    </div>
-</body>
-</html>
-"""
 @app.route('/')
 def index():
-    data = load_data()
-    return render_template_string(ADMIN_TEMPLATE, sites=data.get('sites', {}))
 @app.route('/generate', methods=['POST'])
 def handle_generate():
-    user_prompt = request.form.get('prompt', '').strip()
     if not user_prompt:
-        flash("Промпт не может быть пустым.", 'error')
-        return redirect(url_for('index'))
     try:
-        python_code = generate_website_code_from_prompt(user_prompt)
-        site_id = uuid.uuid4().hex[:8]
-        filename = f"{site_id}.py"
-        filepath = os.path.join(GENERATED_SITES_DIR, filename)
         with open(filepath, "w", encoding="utf-8") as f:
             f.write(python_code)
-        process, port = start_site_process(site_id, filepath)
-        if process and port:
-            data = load_data()
-            data['sites'][site_id] = {
-                'prompt': user_prompt,
-                'filename': filename,
-                'status': 'running',
-                'port': port,
-                'pid': process.pid,
-                'created_at': datetime.now().isoformat()
-            }
-            save_data(data)
-            flash(f"Сайт '{site_id}' успешно сгенерирован и запущен на порту {port}.", 'success')
-        else:
-            os.remove(filepath) # Удаляем неудачный файл
-            flash(f"Не удалось запустить сгенерированный сайт '{site_id}'. Проверьте логи.", 'error')
-    except (ValueError, Exception) as e:
-        flash(f"Ошибка при генерации сайта: {e}", 'error')
-    return redirect(url_for('index'))
-@app.route('/start/<site_id>')
-def start_site(site_id):
-    data = load_data()
-    site_info = data['sites'].get(site_id)
-    if not site_info:
-        flash(f"Сайт с ID {site_id} не найден.", 'error')
-        return redirect(url_for('index'))
-    if site_info['status'] == 'running':
-        flash(f"Сайт {site_id} уже запущен.", 'error')
-        return redirect(url_for('index'))
-    filepath = os.path.join(GENERATED_SITES_DIR, site_info['filename'])
-    if not os.path.exists(filepath):
-        flash(f"Файл для сайта {site_id} не найден. Возможно, он был удален.", 'error')
-        return redirect(url_for('index'))
-    process, port = start_site_process(site_id, filepath)
-    if process and port:
-        site_info.update({'status': 'running', 'port': port, 'pid': process.pid})
-        data['sites'][site_id] = site_info
-        save_data(data)
-        flash(f"Сайт {site_id} успешно запущен на порту {port}.", 'success')
-    else:
-        flash(f"Не удалось запустить сайт {site_id}.", 'error')
-    return redirect(url_for('index'))
-@app.route('/stop/<site_id>')
-def stop_site(site_id):
-    if stop_site_process(site_id):
-        data = load_data()
-        if site_id in data['sites']:
-            data['sites'][site_id]['status'] = 'stopped'
-            data['sites'][site_id].pop('port', None)
-            data['sites'][site_id].pop('pid', None)
-            save_data(data)
-            flash(f"Сайт {site_id} успешно остановлен.", 'success')
-        else:
-            flash(f"Сайт {site_id} не найден в базе данных, но процесс (если был) остановлен.", 'error')
-    else:
-        flash(f"Не удалось остановить сайт {site_id}.", 'error')
-    return redirect(url_for('index'))
-@app.route('/delete/<site_id>')
-def delete_site(site_id):
-    data = load_data()
-    site_info = data['sites'].get(site_id)
-    if not site_info:
-        flash(f"Сайт {site_id} не найден.", 'error')
-        return redirect(url_for('index'))
-    # Сначала остановить, если запущен
-    if site_info['status'] == 'running':
-        stop_site_process(site_id)
-    # Удалить файл
-    filepath = os.path.join(GENERATED_SITES_DIR, site_info['filename'])
-    if os.path.exists(filepath):
-        os.remove(filepath)
-    # Удалить из БД
-    del data['sites'][site_id]
-    save_data(data)
-    flash(f"Сайт {site_id} и его файлы были успешно удалены.", 'success')
-    return redirect(url_for('index'))
-# --- 6. Запуск Приложения ---
 if __name__ == '__main__':
     logging.info("Application starting up...")
-    download_db_from_hf()
-    # При старте синхронизируем состояние (все сайты считаются остановленными)
-    db_data = load_data()
-    for site_id in db_data['sites']:
-        db_data['sites'][site_id]['status'] = 'stopped'
-        db_data['sites'][site_id].pop('port', None)
-        db_data['sites'][site_id].pop('pid', None)
-    save_data(db_data)
-    if HF_TOKEN:
         backup_thread = threading.Thread(target=periodic_backup, daemon=True)
         backup_thread.start()
         logging.info("Periodic backup thread started.")
     else:
         logging.warning("Periodic backup will NOT run (HF_TOKEN for writing not set).")
-    app.run(host='0.0.0.0', port=7860, debug=False)

 import os
 import uuid
 import json
 import logging
 import threading
+import time
+import socket
+import subprocess
 import atexit
+from datetime import datetime
+from flask import Flask, request, jsonify, Response, render_template_string
 import google.generativeai as genai
 from huggingface_hub import HfApi, hf_hub_download
 from huggingface_hub.utils import RepositoryNotFoundError, HfHubHTTPError
 import requests
+from dotenv import load_dotenv
+# --- Конфигурация и инициализация ---
 load_dotenv()
 app = Flask(__name__)
+# --- Константы из Кода 1 ---
+API_KEY_INTERNAL = "AIzaSyArKidc4o0MwbaCFKStlb2q2AwNg6Pnqns" # Ваш ключ Google AI
+GENERATED_APPS_DIR = 'generated_apps'
+# --- Константы из Кода 2 (адаптированные) ---
+DATA_FILE = 'running_apps.json'
 SYNC_FILES = [DATA_FILE]
+REPO_ID = "Kgshop/testsynk" # <-- ВАШ НОВЫЙ РЕПОЗИТОРИЙ
+HF_TOKEN_WRITE = os.getenv("HF_TOKEN")
+HF_TOKEN_READ = os.getenv("HF_TOKEN_READ")
+DOWNLOAD_RETRIES = 3
+DOWNLOAD_DELAY = 5
+# --- Глобальные переменные для управления состоянием ---
+RUNNING_APPS = {} # { "uuid": {"process": Popen_object, "port": 12345, "prompt": "..."} }
+# --- Настройка логирования ---
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
+# --- Создание необходимых директорий ---
+if not os.path.exists(GENERATED_APPS_DIR):
+    os.makedirs(GENERATED_APPS_DIR)
+# --- 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 - Генератор Сайтов</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;
+        }
+        @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;
+            -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 сайтов на базе ИИ</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="Например: создай одностраничный сайт-портфолио для веб-дизайнера Алии. Нужны секции 'Обо мне', 'Работы' и 'Контакты'. Используй современный минималистичный дизайн с темной темой."></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>
+    </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}`); }
+                if (result.site_url) {
+                    const link = document.createElement('a');
+                    link.href = result.site_url;
+                    link.textContent = "Открыть сгенерированное приложение";
+                    link.target = "_blank";
+                    outputContainer.innerHTML = '';
+                    outputContainer.appendChild(link);
+                    copyButton.style.display = 'block';
+                    copyButton.dataset.copyText = window.location.origin + result.site_url;
+                } else {
+                    showError(result.error || "Не удалось получить ссылку на сайт. Ответ сервера не содержит URL.");
+                }
+            } catch (error) {
+                console.error("Fetch Error:", error);
+                showError(`Ошибка: ${error.message}`);
+                copyButton.style.display = 'none';
+            } 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>
+"""
+# --- Функции для работы с Hugging Face (из Кода 2) ---
+def download_db_from_hf(specific_file=None, retries=DOWNLOAD_RETRIES, delay=DOWNLOAD_DELAY):
+    token_to_use = HF_TOKEN_READ if HF_TOKEN_READ else HF_TOKEN_WRITE
+    if not token_to_use:
+        logging.warning("HF_TOKEN_READ/HF_TOKEN_WRITE not set. Download might fail for private repos.")
     files_to_download = [specific_file] if specific_file else SYNC_FILES
     logging.info(f"Attempting download for {files_to_download} from {REPO_ID}...")
     for file_name in files_to_download:
         for attempt in range(retries + 1):
             try:
                 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}.")
+                return True
             except HfHubHTTPError as e:
                 if e.response.status_code == 404:
+                    logging.warning(f"File {file_name} not found in repo {REPO_ID}. Skipping.")
+                    return False
+                logging.error(f"HTTP error downloading {file_name} (Attempt {attempt + 1}): {e}. Retrying...")
             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)
+    logging.error(f"Failed to download {file_name} after {retries + 1} attempts.")
+    return False
 def upload_db_to_hf(specific_file=None):
+    if not HF_TOKEN_WRITE:
         logging.warning("HF_TOKEN (for writing) not set. Skipping upload.")
         return
     try:
         api = HfApi()
         files_to_upload = [specific_file] if specific_file else SYNC_FILES
         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 {file_name} {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}"
                 )
+                logging.info(f"File {file_name} successfully uploaded to Hugging Face.")
     except Exception as e:
+        logging.error(f"Error uploading file to Hugging Face: {e}", exc_info=True)
 def periodic_backup():
     while True:
         time.sleep(1800) # 30 минут
         logging.info("Starting periodic backup...")
+        save_data(RUNNING_APPS)
+        logging.info("Periodic backup finished.")
+# --- Функции управления данными (адаптированные из Кода 2) ---
 def load_data():
+    if os.path.exists(DATA_FILE):
+        try:
+            with open(DATA_FILE, 'r', encoding='utf-8') as f:
+                return json.load(f)
+        except json.JSONDecodeError:
+            logging.error(f"Could not decode {DATA_FILE}. It might be corrupted.")
+            return {}
+    return {}
 def save_data(data):
+    # Сохраняем только метаданные, а не объекты Popen
+    data_to_save = {}
+    for site_uuid, details in data.items():
+        data_to_save[site_uuid] = {
+            "port": details.get("port"),
+            "prompt": details.get("prompt"),
+            "pid": details.get("process").pid if details.get("process") else None,
+            "filepath": details.get("filepath")
+        }
     try:
         with open(DATA_FILE, 'w', encoding='utf-8') as f:
+            json.dump(data_to_save, f, ensure_ascii=False, indent=4)
+        logging.info(f"Data successfully saved to {DATA_FILE}")
+        # Запускаем загрузку в HF в отдельном потоке, чтобы не блокировать основной
+        threading.Thread(target=upload_db_to_hf, args=(DATA_FILE,)).start()
     except Exception as e:
+        logging.error(f"Error saving data to {DATA_FILE}: {e}", exc_info=True)
+# --- Основная логика генерации и управления приложениями ---
+def generate_app_code_from_prompt(user_prompt):
+    """
+    Генерирует код Python Flask приложения на основе запроса пользователя.
+    """
     try:
         genai.configure(api_key=API_KEY_INTERNAL)
     except Exception as e:
+        logging.error(f"Error configuring GenAI: {e}")
+        raise ValueError(f"Не удалось настроить Google AI: {e}")
     if not user_prompt or not user_prompt.strip():
+        raise ValueError("Текстовый запрос (промпт) не может быть пустым.")
+    # !ВАЖНО! Новый системный промпт для генерации Python кода
     system_instruction = (
+        "You are an expert Python Flask 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 Python script.\n"
+        "Follow these rules strictly:\n"
+        "1.  The script must import all necessary libraries, like `Flask`, `render_template_string`, and `os`.\n"
+        "2.  All HTML, CSS, and JavaScript must be embedded within Python multiline strings.\n"
+        "3.  Use `render_template_string` to serve the HTML content.\n"
+        "4.  The application MUST get its port from an environment variable named 'PORT'. Use `os.environ.get('PORT', 5000)`.\n"
+        "5.  The application must run on host '0.0.0.0' to be accessible within a containerized environment.\n"
+        "6.  The script must be directly runnable using `if __name__ == '__main__':`.\n"
+        "7.  Do NOT use any external files, databases, or libraries that are not standard (like flask, os, json).\n"
+        "8.  The generated code must be visually appealing and functional according to the user's request.\n"
+        "9.  CRITICAL: Directly output ONLY the raw Python code. Do not include any explanatory text, markdown formatting (like ```python), or anything else before `import` or after the last line of code."
     )
     full_prompt = f"{system_instruction}\n\nUser request: \"{user_prompt}\""
     try:
+        model = genai.GenerativeModel('gemini-1.5-flash-latest')
         response = model.generate_content(full_prompt)
+        generated_text = response.text
+        if not generated_text or not generated_text.strip().startswith("import"):
+            raise ValueError("Модель сгенерировала некорректный или пустой код.")
+        # Очистка от возможных артефактов
+        if generated_text.strip().startswith("```python"):
+            generated_text = generated_text.strip()[9:]
+            if generated_text.strip().endswith("```"):
+                generated_text = generated_text.strip()[:-3]
+        return generated_text.strip()
     except Exception as e:
+        logging.error(f"Error generating content with GenAI: {e}")
+        if "API key not valid" in str(e):
+             raise ValueError("Внутренняя ошибка конфигурации API.")
+        raise ValueError(f"Ошибка при генерации Python-кода сайта: {e}")
+def find_free_port():
+    """Находит свободный TCP порт."""
+    with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
+        s.bind(('', 0))
+        return s.getsockname()[1]
+def start_app_process(site_uuid, filepath, prompt):
+    """Запускает сгенерированное приложение в отдельном процессе."""
+    port = find_free_port()
+    env = os.environ.copy()
+    env["PORT"] = str(port)
+    try:
+        process = subprocess.Popen(
+            ['python', filepath],
+            env=env,
+            stdout=open(f'{GENERATED_APPS_DIR}/{site_uuid}.log', 'w'),
+            stderr=subprocess.STDOUT
+        )
+        time.sleep(2) # Даем время на запуск
+        if process.poll() is not None: # Проверяем, не завершился ли процесс с ошибкой
+            raise RuntimeError(f"Процесс для сайта {site_uuid} не запустился. Смотрите лог {site_uuid}.log")
+        RUNNING_APPS[site_uuid] = {
+            "process": process,
+            "port": port,
+            "prompt": prompt,
+            "filepath": filepath,
+        }
+        logging.info(f"Successfully started app {site_uuid} on port {port} with PID {process.pid}")
+        return True
+    except Exception as e:
+        logging.error(f"Failed to start app {site_uuid}: {e}", exc_info=True)
+        return False
+# --- Маршруты Flask ---
 @app.route('/')
 def index():
+    return Response(html_template, mimetype='text/html')
+@app.route('/app/<site_uuid>/', defaults={'path': ''})
+@app.route('/app/<site_uuid>/<path:path>', methods=['GET', 'POST', 'PUT', 'DELETE'])
+def proxy_to_generated_app(site_uuid, path):
+    """Проксирует запросы к сгенерированным приложениям."""
+    if site_uuid not in RUNNING_APPS:
+        return "Приложение не найдено или не запущено.", 404
+    port = RUNNING_APPS[site_uuid]['port']
+    target_url = f"http://127.0.0.1:{port}/{path}"
+    try:
+        resp = requests.request(
+            method=request.method,
+            url=target_url,
+            headers={key: value for (key, value) in request.headers if key != 'Host'},
+            data=request.get_data(),
+            cookies=request.cookies,
+            allow_redirects=False,
+            stream=True,
+            params=request.args
+        )
+        headers = [(name, value) for (name, value) in resp.raw.headers.items()]
+        return Response(resp.iter_content(chunk_size=1024), resp.status_code, headers)
+    except requests.exceptions.RequestException as e:
+        logging.error(f"Proxy error for app {site_uuid}: {e}")
+        return f"Ошибка при подключении к приложению {site_uuid}.", 502
 @app.route('/generate', methods=['POST'])
 def handle_generate():
+    user_prompt = request.form.get('prompt')
     if not user_prompt:
+        return jsonify({"error": "Текстовый запрос (промпт) не найден."}), 400
     try:
+        python_code = generate_app_code_from_prompt(user_prompt)
+        site_uuid = str(uuid.uuid4())
+        filename = f"{site_uuid}.py"
+        filepath = os.path.join(GENERATED_APPS_DIR, filename)
         with open(filepath, "w", encoding="utf-8") as f:
             f.write(python_code)
+        logging.info(f"Generated Python code saved to {filepath}")
+        if not start_app_process(site_uuid, filepath, user_prompt):
+             return jsonify({"error": f"Сгенерированный код не удалось запустить. Проверьте лог {site_uuid}.log"}), 500
+        save_data(RUNNING_APPS) # Сохраняем новое состояние
+        site_url = f"/app/{site_uuid}/"
+        return jsonify({"site_url": site_url})
+    except ValueError as ve:
+        return jsonify({"error": str(ve)}), 400
+    except Exception as e:
+        logging.error(f"Unexpected error during site generation: {e}", exc_info=True)
+        return jsonify({"error": f"Внутренняя ошибка сервера: {e}"}), 500
+# --- Функции запуска и очистки ---
+def cleanup_processes():
+    """Останавливает все дочерние процессы при выходе."""
+    logging.info("Shutting down... Terminating all running app processes.")
+    for site_uuid, details in RUNNING_APPS.items():
+        if details.get("process"):
+            logging.info(f"Terminating process for site {site_uuid} (PID: {details['process'].pid})")
+            details["process"].terminate()
+            details["process"].wait() # Ждем завершения
+    logging.info("All child processes terminated.")
+def restart_running_apps(saved_apps):
+    """Перезапускает приложения, которые были активны до перезагрузки."""
+    logging.info("Restarting previously running applications...")
+    for site_uuid, details in saved_apps.items():
+        filepath = details.get("filepath")
+        prompt = details.get("prompt")
+        if filepath and os.path.exists(filepath):
+            logging.info(f"Attempting to restart app {site_uuid} from {filepath}")
+            start_app_process(site_uuid, filepath, prompt)
+        else:
+            logging.warning(f"Could not restart app {site_uuid}, file {filepath} not found.")
+    if saved_apps:
+        save_data(RUNNING_APPS) # Сохраняем обновленные PIDы и порты
 if __name__ == '__main__':
+    atexit.register(cleanup_processes)
     logging.info("Application starting up...")
+    # 1. Скачиваем последнее состояние с HF
+    if download_db_from_hf(DATA_FILE):
+        saved_state = load_data()
+        if saved_state:
+            # 2. Перезапускаем ранее активные приложения
+            restart_running_apps(saved_state)
+    # 3. Запускаем фоновый бэкап, если есть токен
+    if HF_TOKEN_WRITE:
         backup_thread = threading.Thread(target=periodic_backup, daemon=True)
         backup_thread.start()
         logging.info("Periodic backup thread started.")
     else:
         logging.warning("Periodic backup will NOT run (HF_TOKEN for writing not set).")
+    # 4. Запускаем основное приложение
+    port = int(os.environ.get('PORT', 7860))
+    logging.info(f"Starting main Flask app on host 0.0.0.0 and port {port}")
+    app.run(host='0.0.0.0', port=port, debug=False)