refactor: simplify garment model initialization and reloading logic

app.py

@@ -170,30 +170,18 @@ def train_garment(video_path, garment_name):
 # --- Inference Logic ---
 
 def init_processor(garment_name):
-    global frame_processor, current_garment_id
+    # global frame_processor, current_garment_id # Avoid global reliance in helper
     if garment_name is None:
-        return
+        return None
         
-    print(f"Loading garment: {garment_name}")
-    # Initialize if needed
-    if frame_processor is None:
-        # Create processor with list. We just need to give it the name we want to load.
-        # It handles loading from ./checkpoints or ./rtv_ckpts
-        # We need to construct a list where the index matches.
-        # For simplicity, we just re-instantiate or hold a list of 1.
-        # But FrameProcessor holds a list. Let's make it hold just the current one for simplicity.
-        frame_processor = FrameProcessor([garment_name], ckpt_dir='./checkpoints') 
-        # Note: FrameProcessor expects 'rtv_ckpts' for pretrained? 
-        # Check source: it uses opt.checkpoints_dir which defaults to rtv_ckpts in make_pix2pix_model
-        # But we passed ckpt_dir='./checkpoints'.
-        # We need to make sure pretrained and new ones are found.
-        # Simpler approach: symlink rtv_ckpts content to checkpoints 
-    else:
-        # Re-init for simplicity as ID mapping is complex dynamically
-         frame_processor = FrameProcessor([garment_name], ckpt_dir='./checkpoints') # or './rtv_ckpts'
-         
+    print(f"Loading garment: {garment_name}", flush=True)
+    # Initialize 
+    # Always create new for now to ensure we have the right one
+    processor = FrameProcessor([garment_name], ckpt_dir='./checkpoints') 
+    
     # Trigger load
-    frame_processor.switch_to_target_garment(0)
+    processor.switch_to_target_garment(0)
+    return processor
 
 @spaces.GPU
 def process_frame(image, garment_name, enable_tryon):
@@ -204,10 +192,17 @@ def process_frame(image, garment_name, enable_tryon):
         return image
     
     global frame_processor
-    # Check if we need to load/reload
-    # This is a basic check. Real app needs better state management.
-    if frame_processor is None or frame_processor.garment_name_list[0] != garment_name:
-        try:
+    
+    try:
+        # Check if we need to load/reload
+        # We need to treat frame_processor as potentially stale or None
+        should_reload = False
+        if frame_processor is None:
+            should_reload = True
+        elif frame_processor.garment_name_list[0] != garment_name:
+             should_reload = True
+             
+        if should_reload:
              # Link Pretrained to checkpoints if needed
             if os.path.exists(f"rtv_ckpts/{garment_name}") and not os.path.exists(f"checkpoints/{garment_name}"):
                 if not os.path.exists("checkpoints"): os.makedirs("checkpoints")
@@ -216,10 +211,10 @@ def process_frame(image, garment_name, enable_tryon):
                      import shutil
                      shutil.copytree(f"rtv_ckpts/{garment_name}", f"checkpoints/{garment_name}")
 
-            init_processor(garment_name)
-        except Exception as e:
-            print(f"Error loading model: {e}")
-            return image
+            frame_processor = init_processor(garment_name)
+    except Exception as e:
+        print(f"Error loading model: {e}", flush=True)
+        return image
 
     # Convert to RGB (Gradio is RGB, OpenCV is BGR)
     # RTV expects BGR usually? checking rtl_demo...