Update app_lora1.py

app_lora1.py

@@ -163,32 +163,20 @@ def get_pipeline(script_name):
         return PIPELINES[script_name]
 
     log(f"🔧 Building pipeline from {script_name}")
-    code = SCRIPT_CODE[script_name]
 
-    # --- Extract lines after from_pretrained for logging ---
-    pipe_lines = []
-    found_pretrained = False
-    for line in code.splitlines():
-        stripped = line.strip()
-        if not found_pretrained and stripped.startswith("pipe = ZImagePipeline.from_pretrained"):
-            found_pretrained = True
-            pipe_lines.append(line)
-        elif found_pretrained:
-            # Capture subsequent pipe method calls (like layerwise casting)
-            if stripped.startswith("pipe.") or stripped == "":
-                pipe_lines.append(line)
-            else:
-                break
+    # --- Log full .py file ---
+    log("=== SCRIPT CONTENT START ===")
+    log(SCRIPT_CODE[script_name])
+    log("=== SCRIPT CONTENT END ===")
 
-    # Log each line
-    for line in pipe_lines:
-        if line.strip():
-            log(f"• {line.strip()}")
+    namespace = {
+        "__file__": script_name,
+        "__name__": "__main__",
+        "torch": torch,
+    }
 
-    # --- Execute the full script ---
-    namespace = {"torch": torch, "ZImagePipeline": ZImagePipeline}
     try:
-        exec(code, namespace)
+        exec(SCRIPT_CODE[script_name], namespace)
     except Exception as e:
         log(f"❌ Script failed: {script_name}")
         raise RuntimeError(f"Pipeline build failed for {script_name}") from e
@@ -201,8 +189,10 @@ def get_pipeline(script_name):
 
     PIPELINES[script_name] = namespace["pipe"]
     log(f"✅ Pipeline ready: {script_name}")
+
     return PIPELINES[script_name]
 
+
 def get_pipeline_fallback(script_name):
     if script_name in PIPELINES:
         return PIPELINES[script_name]