Add 4x upscale button for any History image via PiD-Image-Upscaler API

Click an image in the History gallery, then "Upscale 4x (PiD)": the image is
sent to prithivMLmods/PiD-Image-Upscaler's /upscaler_run endpoint, so the model
runs on that Space's hardware instead of ours. With the visitor's OAuth token
(hf_oauth) the ZeroGPU quota for that call is billed to them; without login the
call goes out anonymously.

Co-Authored-By: Claude Opus 5 (1M context) <noreply@anthropic.com>

README.md

@@ -7,6 +7,9 @@ sdk: gradio
 sdk_version: 5.49.1
 app_file: app.py
 pinned: true
+hf_oauth: true
+hf_oauth_scopes:
+  - inference-api
 short_description: Explore Lichorosario Qwen-Image-2512 LoRAs
 tags:
 - anycoder

app.py

@@ -38,7 +38,9 @@ import re
 import math
 import numpy as np
 import traceback
+import tempfile
 from prompt_rewrite import rewrite
+from gradio_client import Client, handle_file
 import hashlib
 from functools import partial
 
@@ -191,6 +193,39 @@ def update_history(new_images, history):
 def clear_history():
     """Devuelve una lista vacía para limpiar la galería de historial."""
     return []
+
+
+PID_UPSCALER_SPACE = "prithivMLmods/PiD-Image-Upscaler"
+
+
+def select_history_image(history, evt: gr.SelectData):
+    """Guarda la imagen del historial que se acaba de clickear."""
+    if not history or evt.index is None or evt.index >= len(history):
+        return None
+    return history[evt.index][0]
+
+
+def upscale_history_image(image, oauth_token: gr.OAuthToken | None = None):
+    """Escala 4x la imagen elegida del historial llamando a la API del space
+    prithivMLmods/PiD-Image-Upscaler: el modelo corre en el hardware de ese
+    space, no en el nuestro. Con el token OAuth del visitante la cuota ZeroGPU
+    de esa llamada se le factura a él; sin login sale anónima."""
+    if image is None:
+        raise gr.Error("Click an image in the History gallery first.")
+
+    input_path = os.path.join(tempfile.mkdtemp(), "upscale_input.png")
+    image.save(input_path)
+
+    try:
+        client = Client(PID_UPSCALER_SPACE, hf_token=getattr(oauth_token, "token", None))
+        _, slider = client.predict(handle_file(input_path), "", api_name="/upscaler_run")
+    except Exception as e:
+        raise gr.Error(f"PiD upscaler failed: {e}")
+
+    # El endpoint remoto devuelve el gr.update(...) sin resolver: un dict
+    # {"value": [original, upscalada]}, no la tupla ya lista.
+    original, upscaled = slider["value"]
+    return gr.update(visible=True, value=(original, upscaled))
 ### FIN DE LA MODIFICACIÓN 1 ###
 
 
@@ -884,7 +919,11 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css, delete_cache=(60, 60)) as app:
         <h3 style=\"margin-top: -10px\">LoRA🦜 ChoquinLabs Explorer</h3>""",
         elem_id="title",
     )
-    
+
+    # Login opcional: si el visitante entra con su cuenta, el upscale 4x se
+    # cobra a su cupo de ZeroGPU en vez de salir anónimo (cupo por IP).
+    gr.LoginButton(size="sm")
+
     selected_index = gr.State(None)
     
     with gr.Row():
@@ -977,6 +1016,13 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css, delete_cache=(60, 60)) as app:
                     height="auto",
                     interactive=False
                 )
+
+                selected_history_image = gr.State(None)
+                upscale_button = gr.Button("🔍 Upscale 4x (PiD)", size="sm")
+                upscaled_slider = gr.ImageSlider(
+                    label="Original ↔ PiD 4x upscale",
+                    visible=False
+                )
             ### FIN DE LA MODIFICACIÓN 2 ###
             
             with gr.Row():
@@ -1112,6 +1158,19 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css, delete_cache=(60, 60)) as app:
         inputs=None,
         outputs=[history_state, history_gallery]
     )
+
+    # Upscale 4x de cualquier imagen del historial, vía API del space PiD
+    history_gallery.select(
+        fn=select_history_image,
+        inputs=[history_state],
+        outputs=[selected_history_image]
+    )
+
+    upscale_button.click(
+        fn=upscale_history_image,
+        inputs=[selected_history_image],
+        outputs=[upscaled_slider]
+    )
     ### FIN DE LA MODIFICACIÓN 3 ###
 
     aspect_ratio.change(