Update app.py

app.py

@@ -6,13 +6,35 @@ import asyncio
 import os
 from threading import RLock
 from datetime import datetime
+import gradio as gr
+from huggingface_hub import HfApi, whoami
+
+
+
+
+
+
+
 
 preSetPrompt = thePrompt
 negPreSetPrompt = "[deformed | disfigured], poorly drawn, [bad : wrong] anatomy, [extra | missing | floating | disconnected] limb, (mutated hands and fingers), blurry, text, fuzziness, asian, african, collage, composite, combined image"
-
 lock = RLock()
-
 HF_TOKEN = os.environ.get("HF_TOKEN") if os.environ.get("HF_TOKEN") else None # If private or gated models aren't used, ENV setting is unnecessary.
+# --- Step 2: Authenticate and fetch your models
+api = HfApi()
+user_info = whoami(token=HF_TOKEN)
+username = user_info["name"]
+# Get all models owned by the user
+models = api.list_models(author=username, token=HF_TOKEN)
+model_ids = [m.modelId for m in models]
+if not model_ids:
+    raise ValueError(f"No models found for user '{username}'")
+# --- Step 3: Build Gradio UI
+def handle_model_selection(selected_models):
+    if not selected_models:
+        return "No models selected."
+    return "✅ Selected models:\n" + "\n".join(selected_models)
+
 
 def get_current_time():
     now = datetime.now()
@@ -109,6 +131,12 @@ def gen_fn(model_str, prompt, nprompt="", height=0, width=0, steps=0, cfg=0, see
 
 
 with gr.Blocks(fill_width=True) as demo:
+    with gr.Row():
+        gr.Markdown(f"# ({username}) you are logged in")
+        model_selector = gr.CheckboxGroup(choices=model_ids,value=model_ids, label="your models",        interactive=True,    )
+        output_box = gr.Textbox(lines=10, label="Selected Models")
+        model_selector.change(fn=handle_model_selection, inputs=model_selector, outputs=output_box)
+
     with gr.Tab(str(num_models) + ' Models'):
         with gr.Column(scale=2):
             with gr.Group():