Update app.py

app.py

@@ -1,95 +1,111 @@
 # app.py
+# app.py
 import gradio as gr
-from audio_analyzer import AudioAnalyzer
-from asset_generator import AssetGenerator
-from renderer import VideoRenderer
+import time
+
+# --- ИЗМЕНЕНИЕ: "Ленивое" хранилище для наших тяжелых объектов ---
+# Используем простой класс, чтобы хранить объекты после их первой загрузки.
+# Это гарантирует, что модель Stable Diffusion загрузится в память только один раз.
+class AppState:
+    audio_analyzer = None
+    asset_generator = None
 
-# --- Глобальные объекты ---
-# Инициализируем тяжелые классы один раз
-audio_analyzer = AudioAnalyzer()
-asset_generator = AssetGenerator()
+APP_STATE = AppState()
 
 # --- Главная функция-оркестратор ---
 def generate_visualization(
-    audio_file, artist, title, theme,
+    audio_file, user_prompt, style,
     enable_particles, enable_spectrum, enable_camera,
     duration, progress=gr.Progress(track_tqdm=True)
 ):
-    if not audio_file:
-        raise gr.Error("Пожалуйста, загрузите аудиофайл.")
-
-    # 1. Анализ аудио
-    progress(0.1, desc="Шаг 1/4: Глубокий анализ аудио...")
-    audio_features = audio_analyzer.analyze(audio_file)
-
-    # 2. Загрузка AI-модели для графики (если еще не загружена)
-    progress(0.3, desc="Шаг 2/4: Загрузка AI-модели для графики...")
-    asset_generator.load_model()
-
-    # 3. Создание фоновой текстуры
-    progress(0.5, desc="Шаг 3/4: Генерация уникальной AI-текстуры...")
-    base_texture = asset_generator.create_background_texture(theme, audio_features)
-
-    # 4. Рендеринг видео
-    progress(0.7, desc="Шаг 4/4: Рендеринг видео с эффектами...")
+    # --- ГЛАВНОЕ ИЗМЕНЕНИЕ: Все импорты и инициализация происходят здесь ---
+    # Это гарантирует, что приложение запустится без проблем с памятью.
+    from audio_analyzer import AudioAnalyzer
+    from asset_generator import AssetGenerator
+    from renderer import VideoRenderer
+
+    # Ленивая инициализация: создаем объекты только если их еще нет
+    if APP_STATE.audio_analyzer is None:
+        progress(0, desc="Первый запуск: инициализация анализатора...")
+        APP_STATE.audio_analyzer = AudioAnalyzer()
+    
+    if APP_STATE.asset_generator is None:
+        progress(0, desc="Первый запуск: инициализация AI-художника...")
+        APP_STATE.asset_generator = AssetGenerator()
+        # Загружаем тяжелую модель Stable Diffusion в память. Это займет время.
+        progress(0.1, desc="Первый запуск: загрузка модели Stable Diffusion (до 2-3 минут)...")
+        APP_STATE.asset_generator.load_model()
+    
+    # Дальнейшая логика остается той же
+    if not audio_file: raise gr.Error("Пожалуйста, загрузите аудиофайл.")
+    if not user_prompt: raise gr.Error("Пожалуйста, опишите вашу идею в поле для промпта.")
+    
+    progress(0.3, desc="Шаг 1/3: Анализ ритма и энергии...")
+    audio_features = APP_STATE.audio_analyzer.analyze(audio_file)
+    
+    progress(0.5, desc="Шаг 2/3: ИИ-художник рисует вашу идею...")
+    base_texture = APP_STATE.asset_generator.create_background_texture(style, user_prompt)
+    
+    progress(0.7, desc="Шаг 3/3: Анимация и синхронизация с музыкой...")
     user_settings = {
-        'artist': artist,
-        'title': title,
-        'enable_particles': enable_particles,
-        'enable_spectrum': enable_spectrum,
-        'enable_camera_effect': enable_camera,
+        'artist': '', 'title': '',
+        'enable_particles': enable_particles, 
+        'enable_spectrum': enable_spectrum, 
+        'enable_camera_effect': enable_camera, 
         'duration': duration
     }
-
+    # Renderer легкий, его можно создавать каждый раз
     renderer = VideoRenderer(1920, 1080, 30, base_texture, audio_features, user_settings)
     final_video_path = renderer.render(audio_file)
-
+    
     progress(1.0, desc="Готово!")
-    return final_video_path, base_texture, audio_features
-
-
-# --- Интерфейс Gradio ---
-with gr.Blocks(theme=gr.themes.Soft(), title="Ultimate AI Music Visualizer") as demo:
-    gr.Markdown("# 🎵 Ultimate AI Music Visualizer (Архитектура v3.1)")
-    gr.Markdown("Создание профес��иональной аудио-визуализации с помощью нескольких нейронных сетей на Gradio " + gr.__version__)
+    return final_video_path, base_texture
 
+# --- Интерфейс Gradio (без изменений) ---
+with gr.Blocks(theme=gr.themes.Soft(), title="Creative AI Music Visualizer") as demo:
+    gr.Markdown("# 🎨 Creative AI Music Visualizer v5.3 (Bulletproof Fix)")
+    gr.Markdown("**Вы — режиссёр. ИИ — ваш инструмент.**")
+    
     with gr.Row():
         with gr.Column(scale=1):
             gr.Markdown("### 1. Загрузите трек")
-            audio_input = gr.Audio(label="Аудиофайл", type="filepath")
-
-            gr.Markdown("### 2. Информация о треке")
-            artist_input = gr.Textbox(label="Имя артиста (для бегущей строки)")
-            title_input = gr.Textbox(label="Название трека (для бегущей строки)")
-
-            gr.Markdown("### 3. Визуальный стиль")
-            theme_input = gr.Dropdown(
-                label="Основная тема",
-                choices=["Cosmic Symphony", "Cyberpunk Night", "Liquid Metal Dreams", "Organic Universe", "Fantasy Landscape"],
-                value="Cosmic Symphony"
+            audio_input = gr.Audio(label="Аудиофайл (для анализа ритма и синхронизации)", type="filepath")
+            
+            gr.Markdown("### 2. Опишите вашу идею")
+            style_input = gr.Dropdown(
+                label="Общий стиль",
+                choices=["Cinematic 8K Wallpaper", "Dark Fantasy Art", "Surrealism", "Symbolic Masterpiece", "Abstract Painting"],
+                value="Cinematic 8K Wallpaper"
             )
-
-            gr.Markdown("### 4. Настройки эффектов")
-            with gr.Accordion("Включить/выключить модули", open=True):
+            prompt_input = gr.Textbox(
+                label="Ваш творческий промпт", 
+                lines=5, 
+                placeholder="Опишите здесь вашу метафору, например: 'потресканные песочные часы с черным песком внутри, на черном фоне с тяжелыми цепями, которые медленно качаются'..."
+            )
+            
+            gr.Markdown("### 3. Настройки анимации")
+            with gr.Accordion("Включить/выключить эффекты", open=True):
                 particles_toggle = gr.Checkbox(label="Эффект частиц", value=True)
-                spectrum_toggle = gr.Checkbox(label="Эквалайзер (Спектр)", value=True)
-                camera_toggle = gr.Checkbox(label="AI-эффект камеры (зум на битах)", value=True)
-
-            gr.Markdown("### 5. Настройки рендера")
-            duration_slider = gr.Slider(label="Максимальная длительность (секунды)", minimum=10, maximum=180, value=30, step=1)
-
-            generate_btn = gr.Button("🚀 Сгенерировать Визуализацию!", variant="primary")
-
+                spectrum_toggle = gr.Checkbox(label="Эквалайзер", value=False)
+                camera_toggle = gr.Checkbox(label="Эффект камеры (зум на битах)", value=True)
+            
+            duration_slider = gr.Slider(label="Длительность видео (сек)", minimum=10, maximum=60, value=20, step=1)
+            
+            generate_btn = gr.Button("✨ Воплотить идею!", variant="primary")
+            
         with gr.Column(scale=2):
-            output_video = gr.Video(label="Итоговое видео")
-            output_image = gr.Image(label="Сгенерированная AI-текстура")
-            output_json = gr.JSON(label="Результаты анализа аудио")
+            output_video = gr.Video(label="Ваше видео")
+            output_image = gr.Image(label="Концепт-арт, созданный по вашему промпту")
 
     generate_btn.click(
         fn=generate_visualization,
         inputs=[
-            audio_input, artist_input, title_input, theme_input,
+            audio_input, prompt_input, style_input,
             particles_toggle, spectrum_toggle, camera_toggle, duration_slider
         ],
-        outputs=[output_video, output_image, output_json]
-    )
+        outputs=[output_video, output_image]
+    )
+
+# --- Запуск приложения (без изменений) ---
+if __name__ == "__main__":
+    demo.launch()