created demo options

app.py

@@ -10,47 +10,35 @@ from huggingface_hub import list_models
 
 from diffusers import StableDiffusionPipeline
 
-#submissions_list = list_models(filter=["dreambooth-hackathon", category], full=True)
-#spaces_pipeline_load = [submission.id for submission in submissions_list ]
-#for ids in spaces_pipeline_load:
-#    mydict[ids] = StableDiffusionPipeline.from_pretrained(ids) #('ashiqabdulkhader/shiba-dog')
-    #f"pipeline{ids.split('//')[-1]}" = StableDiffusionPipeline.from_pretrained(ids) #('ashiqabdulkhader/shiba-dog')
-
-#pipeline = StableDiffusionPipeline.from_pretrained("ashiqabdulkhader/shiba-dog") #('pharma/sugar-glider')
-#image = pipeline().images[0]
-#image
-#https://huggingface.co/ashiqabdulkhader/shiba-dog
-def filter_species(species):
-    return gr.Dropdown.update(choices=species_map[species], value=species_map[species][1]), gr.update(visible=True)
-
-    # %% app.ipynb 1
-def make_clickable_demo(model_name, prompt): #link=None):
-    #if link is None:
-    #    link = "https://huggingface.co/" + model_name
-    # Remove user from model name
-    prompt = "a photo of " + ' '.join(model_name.split('/')[-1].split['-']) + str(prompt)
-        
-    return gr.Button.update()
-    
-def make_clickable_model(model_name,prompt, link=None):
-    if link is None:
-        link = "https://huggingface.co/" + model_name
+# %% app.ipynb 1
+def get_model_list(category):
+    submissions_list = list_models(filter=["dreambooth-hackathon", category], full=True)
+    spaces_pipeline_load = [submission.id for submission in submissions_list ]
+    return gr.Dropdown.update(choices=spaces_pipeline_load , value=spaces_pipeline_load[5])
+
 
-    #adding functionality for demo
-    user_model_nm = model_name.split('/')[-1]
-    print(f"user and model name is - {user_model_nm}")
+def get_initial_prompt(model_nm):
+    #a photo of peterj/shbrcky-dog 
+    user_model_nm = model_nm.split('/')[-1]
     if '-' in user_model_nm:
-        model_nm = " ".join(user_model_nm.split('-'))
+        prompt = " ".join(user_model_nm.split('-'))
     else:
-        model_nm = user_model_nm
-    print(f"model name is - {model_nm}")
-    prompt = "a photo of " + model_nm + str(prompt)
+        prompt = user_model_nm
+    return gr.Textbox.update(value="a photo of " + prompt + " ")
+
+def make_demo(model_name, prompt): #link=None):
+    #prompt = "a photo of " + ' '.join(model_name.split('/')[-1].split['-']) + str(prompt)
     pipeline = StableDiffusionPipeline.from_pretrained(model_name) #("ashiqabdulkhader/shiba-dog") #('pharma/sugar-glider')
-    image_demo = pipeline(prompt).images[0]
-    
-    # Remove user from model name
-    return image_out.Update(value=image_demo, label=model_name.split("/")[-1])  #f'<a target="_blank" href="{link}">{model_name.split("/")[-1]}</a>'
+    image_demo = pipeline(prompt).images[0]    
+    return image_demo #gr.Button.update()
+
 
+def make_clickable_model(model_name, link=None):
+    if link is None:
+        link = "https://huggingface.co/" + model_name
+    # Remove user from model name
+    return f'<a target="_blank" href="{link}">{model_name.split("/")[-1]}</a>'
+    
 def make_clickable_user(user_id):
     link = "https://huggingface.co/" + user_id
     return f'<a  target="_blank" href="{link}">{user_id}</a>'
@@ -63,15 +51,20 @@ def get_submissions(category, prompt):
     for submission in submissions:
         # user, model, likes
         user_id = submission.id.split("/")[0]
+        model_nm = submission.id.split("/")[-1]
+        if '-' in model_nm:
+            model_nm = " ".join(model_nm.split('-'))
+        #button_html = get_button()    
         leaderboard_models.append(
             (
                 make_clickable_user(user_id),
                 make_clickable_model(submission.id, prompt),
                 submission.likes,
+                #button_html #'a photo of ' + model_nm + " "
             )
         )
 
-    df = pd.DataFrame(data=leaderboard_models, columns=["User", "Model", "Likes"])
+    df = pd.DataFrame(data=leaderboard_models, columns=["User", "Model", "Likes", ])
     df.sort_values(by=["Likes"], ascending=False, inplace=True)
     df.insert(0, "Rank", list(range(1, len(df) + 1)))
     return df
@@ -92,12 +85,15 @@ with block:
     )
     with gr.Row():
         prompt_in = gr.Textbox(label="Type in a Prompt. This will be suffixed to 'a photo of <model name>', so prompt accordingly -", value="")
-        #button_in = gr.Button(label = "Generate Image using this model")
+        with gr.Column():
+            gr.Dropdown(label="Choose a dreamboooth model", value)
+            button_in = gr.Button(label = "Generate Image using this model")
+        image_out = gr.Image()
     with gr.Tabs():
         with gr.TabItem("Animal 🐨"):
             with gr.Row():
                 animal_data = gr.components.Dataframe(
-                    type="pandas", datatype=["number", "markdown", "markdown", "number"]
+                    type="pandas", datatype=["number", "markdown", "markdown", "number","str"], interactive = True
                 )
             with gr.Row():
                 data_run = gr.Button("Refresh")
@@ -109,7 +105,7 @@ with block:
         with gr.TabItem("Science 🔬"):
             with gr.Row():
                 science_data = gr.components.Dataframe(
-                    type="pandas", datatype=["number", "markdown", "markdown", "number"]
+                    type="pandas", datatype=["number", "markdown", "markdown", "number", "str"], interactive = True
                 )
             with gr.Row():
                 data_run = gr.Button("Refresh")
@@ -119,7 +115,7 @@ with block:
         with gr.TabItem("Food 🍔"):
             with gr.Row():
                 food_data = gr.components.Dataframe(
-                    type="pandas", datatype=["number", "markdown", "markdown", "number"]
+                    type="pandas", datatype=["number", "markdown", "markdown", "number", "str"], interactive = True
                 )
             with gr.Row():
                 data_run = gr.Button("Refresh")
@@ -129,7 +125,7 @@ with block:
         with gr.TabItem("Landscape 🏔"):
             with gr.Row():
                 landscape_data = gr.components.Dataframe(
-                    type="pandas", datatype=["number", "markdown", "markdown", "number"]
+                    type="pandas", datatype=["number", "markdown", "markdown", "number", "str"], interactive = True
                 )
             with gr.Row():
                 data_run = gr.Button("Refresh")
@@ -141,7 +137,7 @@ with block:
         with gr.TabItem("Wilcard 🔥"):
             with gr.Row():
                 wildcard_data = gr.components.Dataframe(
-                    type="pandas", datatype=["number", "markdown", "markdown", "number"]
+                    type="pandas", datatype=["number", "markdown", "markdown", "number", "str"], interactive = True
                 )
             with gr.Row():
                 data_run = gr.Button("Refresh")
@@ -151,11 +147,10 @@ with block:
                     outputs=wildcard_data,
                 )
     
-    with gr.Row() : #as your_model_demo :
-        image_out = gr.Image()
-        
-    #button_in.click(make_clickable_demo, prompt_in, your_model_demo)
-
+    theme.change(get_model_list, theme, model_list )  
+    model_list.change(get_initial_prompt, model_list, prompt_in )
+    button_in.click(make_demo, [model_list, prompt_in], image_out)
+    
     block.load(get_submissions, inputs=[gr.Variable("animal"), prompt_in], outputs=animal_data)
     block.load(get_submissions, inputs=[gr.Variable("science"), prompt_in], outputs=science_data)
     block.load(get_submissions, inputs=[gr.Variable("food"), prompt_in], outputs=food_data)