Upload app.py with huggingface_hub

app.py

@@ -237,16 +237,17 @@ def resolve_scheduler_name(scheduler_name):
     logger.warning(f"Unknown scheduler: {scheduler_name}, using default: {DEFAULT_SCHEDULER}")
     return DEFAULT_SCHEDULER
 
-def switch_scheduler(pipeline, scheduler_name):
+def get_scheduler(base_scheduler, scheduler_name):
     """
-    Переключает scheduler в pipeline
+    Создает новый scheduler на основе базового
+    Thread-safe: создает новый объект вместо изменения существующего
     
     Args:
-        pipeline: DiffusionPipeline или QwenImageImg2ImgPipeline
+        base_scheduler: Базовый scheduler из pipeline
         scheduler_name: Имя scheduler или его алиас
     
     Returns:
-        pipeline с новым scheduler
+        Новый scheduler объект
     """
     # Разрешаем алиас
     scheduler_name = resolve_scheduler_name(scheduler_name)
@@ -260,20 +261,20 @@ def switch_scheduler(pipeline, scheduler_name):
     scheduler_config = scheduler_info["config"]
     
     try:
-        # Создаем новый scheduler с конфигом текущего + дополнительными параметрами
+        # Создаем НОВЫЙ scheduler с конфигом базового + дополнительными параметрами
+        # Это thread-safe подход - не меняем глобальный pipeline
         new_scheduler = scheduler_class.from_config(
-            pipeline.scheduler.config,
+            base_scheduler.config,
             **scheduler_config
         )
         
-        pipeline.scheduler = new_scheduler
-        logger.info(f"  Switched to scheduler: {scheduler_name} ({scheduler_info['alias']})")
+        logger.info(f"  Created scheduler: {scheduler_name} ({scheduler_info['alias']})")
+        return new_scheduler
         
     except Exception as e:
-        logger.error(f"  [X] Error switching scheduler: {e}")
-        logger.info(f"  Keeping current scheduler")
-    
-    return pipeline
+        logger.error(f"  [X] Error creating scheduler: {e}")
+        logger.warning(f"  Using base scheduler instead")
+        return base_scheduler
 
 # =================================================================
 # HELPER FUNCTIONS
@@ -449,9 +450,6 @@ def generate_text2img(
     logger.info("TEXT-TO-IMAGE GENERATION")
     logger.info("=" * 60)
     
-    # Switch scheduler if needed
-    switch_scheduler(pipe_txt2img, scheduler_name)
-    
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
     
@@ -462,7 +460,16 @@ def generate_text2img(
     logger.info(f"  Scheduler: {scheduler_name}")
     logger.info(f"  LoRA: {lora_name} (scale: {lora_scale})")
     
+    # Сохраняем оригинальный scheduler ДО try блока (для finally)
+    original_scheduler = pipe_txt2img.scheduler
+    
     try:
+        # Создаем scheduler (thread-safe)
+        current_scheduler = get_scheduler(pipe_txt2img.scheduler, scheduler_name)
+        
+        # Устанавливаем новый scheduler
+        pipe_txt2img.scheduler = current_scheduler
+        
         # Загружаем LoRA если выбрана
         trigger_word = None
         if lora_name != "None":
@@ -496,6 +503,9 @@ def generate_text2img(
     except Exception as e:
         logger.error(f"  [X] Error: {e}")
         raise
+    finally:
+        # Восстанавливаем оригинальный scheduler (thread-safety)
+        pipe_txt2img.scheduler = original_scheduler
 
 @gpu_decorator(duration=180)
 def generate_img2img(
@@ -518,9 +528,6 @@ def generate_img2img(
     logger.info("IMAGE-TO-IMAGE GENERATION")
     logger.info("=" * 60)
     
-    # Switch scheduler if needed
-    switch_scheduler(pipe_img2img, scheduler_name)
-    
     if input_image is None:
         raise gr.Error("Please upload an input image")
     
@@ -537,10 +544,19 @@ def generate_img2img(
     logger.info(f"  Scheduler: {scheduler_name}")
     logger.info(f"  LoRA: {lora_name}")
     
+    # Сохраняем оригинальный scheduler ДО try блока (для finally)
+    original_scheduler = pipe_img2img.scheduler if pipe_img2img else None
+    
     try:
         if pipe_img2img is None:
             raise gr.Error("Image2Image pipeline not available")
         
+        # Создаем scheduler (thread-safe)
+        current_scheduler = get_scheduler(pipe_img2img.scheduler, scheduler_name)
+        
+        # Устанавливаем новый scheduler
+        pipe_img2img.scheduler = current_scheduler
+        
         # Загружаем LoRA если выбрана
         trigger_word = None
         if lora_name != "None":
@@ -573,6 +589,10 @@ def generate_img2img(
     except Exception as e:
         logger.error(f"  [X] Error: {e}")
         raise
+    finally:
+        # Восстанавливаем оригинальный scheduler (thread-safety)
+        if original_scheduler is not None and pipe_img2img is not None:
+            pipe_img2img.scheduler = original_scheduler
 
 # ControlNet функция убрана - не используется