Update app.py

app.py

@@ -1,41 +1,67 @@
 import gradio as gr
-from huggingface_hub import list_models
+import os
 
+from huggingface_hub import list_liked_repos
+from diffusers import DiffusionPipeline
 
-def hello(profile: gr.OAuthProfile | None) -> str:
-    # ^ expect a gr.OAuthProfile object as input to get the user's profile
-    # if the user is not logged in, profile will be None
-    if profile is None:
-        return "I don't know you."
-    return f"Hello {profile.name}"
+def image_mod(prompt: str, model: str, image_0: gr.Image, image_1: gr.Image) -> list[gr.Image]:
+    images = [image_0, image_1]
+    for i, diffusion_model in enumerate(model):
+        pipeline = DiffusionPipeline.from_pretrained(pretrained_model_name_or_path=diffusion_model, device_map="auto")
+        images[i] = pipeline(prompt).images[0]
+    return images
 
+def get_liked_models(username: str) -> list:
+    likes = list_liked_repos(username)
+    # filter if the models are diffusion models
+    return likes.models
 
-def list_private_models(profile: gr.OAuthProfile | None, oauth_token: gr.OAuthToken | None) -> str:
-    # ^ expect a gr.OAuthToken object as input to get the user's token
-    # if the user is not logged in, oauth_token will be None
-    if oauth_token is None:
-        return "Please log in to list private models."
-    models = [
-        f"{model.id} ({'private' if model.private else 'public'})"
-        for model in list_models(author=profile.username, token=oauth_token.token)
-    ]
-    return "Models:\n\n" + "\n - ".join(models) + "."
+
+def get_dropdown_for_huggingface(profile: gr.OAuthProfile):
+    return gr.Dropdown(
+        choices=get_liked_models(profile.username),
+        multiselect=True,
+        max_choices=2,
+        label="Select a Model from Hugginface",
+        info="Select up to two models to compare. If you want to see more models populated in the dropdown, just like a model on huggingface! The dropdown is refreshed every minute.",
+        every=60,
+        visible=True
+    )
 
 
+def change_model(choice): 
+    images = [gr.Image(visible=False), gr.Image(visible=False)]
+    for i, model in enumerate(choice):
+        images[i] = gr.Image(label=model, visible=True, interactive=False)
+    
+    return images
+
 with gr.Blocks() as demo:
     gr.Markdown(
-        "# Gradio OAuth Space"
-        "\n\nThis Space is a demo for the **Sign in with Hugging Face** feature. "
-        "Duplicate this Space to get started."
-        "\n\nFor more details, check out:"
-        "\n- https://www.gradio.app/guides/sharing-your-app#o-auth-login-via-hugging-face"
-        "\n- https://huggingface.co/docs/hub/spaces-oauth"
-    )
+    """
+    **Comparing Stable Diffusion Models**
+
+    See https://huggingface.co/docs/diffusers/main/en/api/diffusion_pipeline
+
+
+    """)
     gr.LoginButton()
-    # ^ add a login button to the Space
-    m1 = gr.Markdown()
-    m2 = gr.Markdown()
-    demo.load(hello, inputs=None, outputs=m1)
-    demo.load(list_private_models, inputs=None, outputs=m2)
+    dropdown = gr.Dropdown(visible=False)
+    demo.load(get_dropdown_for_huggingface, inputs=None, outputs=dropdown)
+    
+    textbox = gr.Textbox(label="Prompt for Image Generation")
+    
+    with gr.Row() as row:
+        image_0 = gr.Image(visible=False)
+        image_1 = gr.Image(visible=False)
+        dropdown.change(
+            fn=change_model,
+            inputs=dropdown,
+            outputs=[image_0, image_1]
+        )
+    
+    textbox.submit(
+        image_mod, [textbox, dropdown, image_0, image_1], [image_0, image_1]
+    )
 
 demo.launch()