Update app.py

app.py

@@ -17,6 +17,7 @@ def get_samples():
     prompt_list = prompt_df.sample(n = 10)["Prompt"].map(lambda x: x).values.tolist()
     return prompt_list
 
+'''
 def get_params(request: gr.Request):
     params = request.query_params
     ip = request.client.host
@@ -32,6 +33,16 @@ def get_params(request: gr.Request):
                                    info = "default prompt: {}".format(DEFAULT_PROMPT)
                                    )
     #return list_models, get_samples()
+'''
+
+def update_models():
+    client = InferenceClient()
+    models = client.list_deployed_models()
+    list_models = models["text-to-image"]
+    return gr.Dropdown.update(choices=list_models) 
+
+def update_prompts():
+    return gr.Dropdown.update(choices=get_samples())
     
 client = InferenceClient()
 models = client.list_deployed_models()
@@ -234,8 +245,8 @@ with gr.Blocks(css=css) as demo:
     
     with gr.Row(elem_id="prompt-container"):
         with gr.Column():
-            btn_refresh = gr.Button(value="Click Refresh to get current deployed models and newly Prompt candidates")
-            btn_refresh.click(None, js="window.location.reload()")
+            btn_refresh = gr.Button(value="Click to get current deployed models and newly Prompt candidates")
+            #btn_refresh.click(None, js="window.location.reload()")
             current_model = gr.Dropdown(label="Current Model", choices=list_models, value=DEFAULT_MODEL,
                                        info = "default model: {}".format(DEFAULT_MODEL)
                                        )
@@ -272,6 +283,8 @@ with gr.Blocks(css=css) as demo:
             
     text_button.click(generate_txt2img, inputs=[current_model, text_prompt, negative_prompt, image_style], outputs=image_output)
     select_button.click(generate_txt2img, inputs=[current_model, select_prompt, negative_prompt, image_style], outputs=image_output)
+    btn_refresh.click(update_models, None, current_model)
+    btn_refresh.click(update_prompts, None, select_prompt)
     
     #demo.load(get_params, None, [current_model, select_prompt])