Update app.py

app.py

@@ -276,15 +276,12 @@ def generate(
     oauth_token: gr.OAuthToken | None = None,
     progress=gr.Progress(track_tqdm=True),
 ):
-    # --- INTHA VARIGALAI MATTUM PUTHUSA SERTHIDUNGA ---
     try:
-        # Palaiya lora weight-ai remove panna
         pipe.unload_lora_weights()
-        # GPU memory-ai free panna
         torch.cuda.empty_cache()
     except Exception as e:
         print(f"Unload error (ithu paravaillai): {e}")
-    # ---------------------------------------------
+   
     if oauth_token is None:
         raise gr.Error("Please sign in with your HuggingFace account to generate images.")
     if selected_index is None:
@@ -294,10 +291,8 @@ def generate(
     trigger   = lora["trigger_word"]
     prompt_in = f"{trigger} {prompt}" if trigger else prompt
 
-    # ZeroGPU: LoRA loading must happen inside GPU scope
     pipe.to("cuda")
-    pipe.unload_lora_weights()
-
+  
     if speed_mode == "Fast · 8 steps":
         pipe.load_lora_weights(LIGHTNING_REPO, weight_name=LIGHTNING_WEIGHT, adapter_name="lightning")
         pipe.load_lora_weights(lora["repo"], weight_name=lora.get("weights"), low_cpu_mem_usage=True, adapter_name="style")
@@ -309,21 +304,31 @@ def generate(
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
 
-    w, h = aspect_to_wh(aspect)
-
+    w, h = aspect_to_wh(aspect_ratio)
     generator = torch.Generator(device="cuda").manual_seed(seed)
+    
     image = pipe(
         prompt=prompt_in,
         negative_prompt=neg_prompt,
         num_inference_steps=steps,
-        true_cfg_scale=cfg,
-        width=w,
+        guidance_scale=cfg_scale,
         height=h,
         generator=generator,
     ).images[0]
-
+    
     pipe.to("cpu")
     torch.cuda.empty_cache()
+    
+    new_meta = build_metadata_html(lora["title"], seed, steps, cfg_scale, aspect_ratio)
+    
+    import datetime
+    history_entry = {
+        "prompt": prompt,
+        "model": lora["title"],
+        "time": datetime.datetime.now().strftime("%H:%M")
+    }
+    history_state.append(history_entry)
+    new_history_html = format_history_html(history_state)
 
     return image, seed, meta_html, history_html, history_state
 
@@ -522,14 +527,13 @@ with gr.Blocks(title="D-REX Studio") as app:
     selected_index = gr.State(value=None)
     history_state  = gr.State(value=[])
 
-    # ── HEADER ──
     # ── HEADER ──
     gr.HTML("""<div id="drex-header"><div class="drex-logo"><div class="drex-mark"><svg viewBox="0 0 16 16"><path d="M8 0L1 4v8l7 4 7-4V4L8 0zm0 2.4L13 5.5 8 8.6 3 5.5 8 2.4z M2.5 6.8l4.7 2.7v5.1l-4.7-2.7V6.8zm6.3 7.8V9.5l4.7-2.7v5.1l-4.7 2.7z"/></svg></div><div><h1>D-REX</h1><div class="drex-sub">LoRA Studio - Qwen-Image</div></div></div></div>""")
 
-    # ── LOGIN ──
+    # ── THEME BAR & LOGIN ──
     with gr.Row():
         gr.LoginButton(scale=0)
-
+        
     # ── THEME BAR ──
     with gr.Row():
         theme_selector = gr.HTML("""