Update app.py

app.py

@@ -126,10 +126,6 @@ div#warning-duplicate .actions a {
     display: inline-block;
     margin-right: 10px;
 }
-#prompt{padding: 0 0 1em 0}
-#prompt input{width: calc(100% - 160px);border-top-right-radius: 0px;border-bottom-right-radius: 0px;}
-#run_button{position: absolute;margin-top: 25.8px;right: 0;margin-right: 0.75em;border-bottom-left-radius: 0px;border-top-left-radius: 0px}
-#prompt_area .form{border:0}
 """
 
 with gr.Blocks(css=css) as demo:
@@ -192,11 +188,27 @@ with gr.Blocks(css=css) as demo:
                 )
             
             with gr.Column():
+
+                if not is_shared_ui:
+
+                    your_username = api.whoami()["name"]
+                    my_models = api.list_models(author=your_username, filter=["diffusers", "stable-diffusion-xl", 'lora'])
+                    model_names = [item.modelId for item in my_models]
+                    
+                    #print(model_names)           
+                    
+                    lora_2_id = gr.Dropdown(
+                        label = "LoRa 2 ID",
+                        choices = model_names,
+                        allow_custom_value = True
+                        #placeholder = "username/model_id"
+                    )
                 
-                lora_2_id = gr.Textbox(
-                    label = "LoRa 2 ID",
-                    placeholder = "username/model_id"
-                )
+                else:
+                    lora_2_id = gr.Textbox(
+                        label = "LoRa 2 ID",
+                        placeholder = "username/model_id"
+                    )
 
                 lora_2_sfts = gr.Dropdown(
                     label = "Safetensors file",
@@ -206,17 +218,17 @@ with gr.Blocks(css=css) as demo:
         load_models_btn = gr.Button("Load models and .safetensors")
 
         # PART 2 • INFERENCE
-        with gr.Box():
-            with gr.Row(elem_id="prompt-area"):
+        with gr.Column():
+            with gr.Row():
             
                 prompt = gr.Textbox(
                     label = "Your prompt",
-                    show_label = False,
+                    show_label = True,
                     info = "Use your trigger words into a coherent prompt",
                     placeholder = "e.g: a triggerWordOne portrait in triggerWord2 style"
                 )
 
-                run_btn = gr.Button("Run", elem_id="run_button")
+            run_btn = gr.Button("Run", elem_id="run_button")
         
             output_image = gr.Image(
                 label = "Output"