Update app.py

app.py

@@ -1,4 +1,4 @@
-# app.py (с авто-инициализацией)
+# app.py (исправленная версия)
 import gradio as gr
 import os
 import time
@@ -19,12 +19,32 @@ def create_zip_archive(output_dir):
                 zipf.write(file_path, file_path.relative_to(output_dir))
     return str(zip_path)
 
+def get_initial_status():
+    """Get initial system status"""
+    system_info = separator_app.get_system_info()
+    if system_info['initialized']:
+        return "✅ Система готова к работе!"
+    elif separator_app.initializing:
+        return "🔄 Система инициализируется..."
+    elif separator_app.init_error:
+        return f"❌ Ошибка инициализации: {separator_app.init_error}"
+    else:
+        return "⚪ Статус неизвестен"
+
 def separate_audio_interface(audio_file, progress=gr.Progress()):
     """Gradio interface function for audio separation"""
     if not audio_file:
         return "❌ Пожалуйста, загрузите аудиофайл", None
     
     try:
+        # Check if system is ready
+        if not separator_app.initialized and separator_app.initializing:
+            return "🔄 Система все еще инициализируется. Пожалуйста, подождите...", None
+        elif not separator_app.initialized and separator_app.init_error:
+            return f"❌ Система не готова: {separator_app.init_error}", None
+        elif not separator_app.initialized:
+            return "❌ Система не инициализирована", None
+
         # Save uploaded file
         audio_path = AUDIO_DIR / f"input_{int(time.time())}{Path(audio_file).suffix}"
         with open(audio_path, "wb") as f:
@@ -51,19 +71,6 @@ def separate_audio_interface(audio_file, progress=gr.Progress()):
     except Exception as e:
         return f"❌ Ошибка обработки: {str(e)}", None
 
-def get_system_status():
-    """Get current system status"""
-    system_info = separator_app.get_system_info()
-    
-    if system_info['initialized']:
-        return "✅ Система готова к работе!", "success"
-    elif separator_app.initializing:
-        return "🔄 Система инициализируется...", "warning"
-    elif separator_app.init_error:
-        return f"❌ Ошибка инициализации: {separator_app.init_error}", "error"
-    else:
-        return "⚪ Статус неизвестен", "neutral"
-
 def get_example_files():
     """Get example files for demo"""
     examples_dir = Path("examples")
@@ -99,7 +106,7 @@ with gr.Blocks(
     # System status display
     with gr.Row():
         status_display = gr.HTML(
-            value="<div class='status-warning'>🔄 Система инициализируется...</div>",
+            value=f"<div class='status-warning'>{get_initial_status()}</div>",
             label="Статус системы"
         )
     
@@ -177,14 +184,6 @@ with gr.Blocks(
     - Архив содержит 5 отдельных дорожек для профессиональной работы
     """)
     
-    # JavaScript для обновления статуса
-    demo.load(
-        fn=get_system_status,
-        inputs=[],
-        outputs=[status_display],
-        every=2  # Обновлять каждые 2 секунды
-    )
-    
     # Event handlers
     process_btn.click(
         fn=separate_audio_interface,
@@ -197,6 +196,11 @@ if __name__ == "__main__":
     print("🚀 Запуск Music Source Separator...")
     print("🔄 Начата автоматическая инициализация моделей...")
     
+    # Check system status after a short delay to show initialization progress
+    time.sleep(2)
+    initial_status = get_initial_status()
+    print(f"Статус системы: {initial_status}")
+    
     demo.launch(
         server_name="0.0.0.0",
         server_port=7860,