Update app.py

app.py

@@ -6,19 +6,6 @@ import random
 import numpy as np
 from typing import Iterable
 
-# Patch: HfFolder removed in huggingface_hub>=0.24 but needed by gradio 4.40
-import huggingface_hub as _hfh
-if not hasattr(_hfh, "HfFolder"):
-    class _HfFolder:
-        @staticmethod
-        def get_token():
-            return os.environ.get("HF_TOKEN", None)
-        @staticmethod
-        def save_token(token): pass
-        @staticmethod
-        def delete_token(): pass
-    _hfh.HfFolder = _HfFolder
-
 import torch
 from PIL import Image
 import gradio as gr
@@ -286,8 +273,11 @@ def generate(
     prompt, neg_prompt, cfg, steps, selected_index,
     randomize_seed, seed, aspect, lora_scale, speed_mode,
     history_state,
+    oauth_token: gr.OAuthToken | None = None,
     progress=gr.Progress(track_tqdm=True),
 ):
+    if oauth_token is None:
+        raise gr.Error("Please sign in with your HuggingFace account to generate images.")
     if selected_index is None:
         raise gr.Error("Select a LoRA from the gallery first.")
     if not prompt.strip():
@@ -540,6 +530,10 @@ with gr.Blocks(theme=drex_theme, css=CSS, js=JS_INIT, delete_cache=(300, 300), t
     </div>
     """)
 
+    # ── LOGIN ──
+    with gr.Row():
+        gr.LoginButton(scale=0)
+
     # ── THEME BAR ──
     with gr.Row():
         theme_selector = gr.HTML("""
@@ -710,6 +704,7 @@ with gr.Blocks(theme=drex_theme, css=CSS, js=JS_INIT, delete_cache=(300, 300), t
         ],
         outputs=[result, seed, meta_panel, history_html, history_state],
     )
+    # Note: oauth_token injected automatically by Gradio from LoginButton
 
     custom_lora_input.input(
         add_custom_lora,
@@ -730,4 +725,4 @@ with gr.Blocks(theme=drex_theme, css=CSS, js=JS_INIT, delete_cache=(300, 300), t
 # ── LAUNCH ────────────────────────────────────────────────────────────────────
 
 app.queue()
-app.launch(show_error=True, share=False, server_name="0.0.0.0")
+app.launch()