Update app.py

app.py

@@ -83,12 +83,14 @@ div#share-btn-container > div {flex-direction: row;background: black;align-items
 #post_gen_info{margin-top: .5em}
 #thumbs_up_clicked{background:green}
 #thumbs_down_clicked{background:red}
+.elem_lora a{color: var(--body-text-color)}
 '''
 
 original_pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16)
 
 @spaces.GPU
 def merge_and_run(prompt, negative_prompt, shuffled_items, lora_1_scale=0.5, lora_2_scale=0.5, seed=-1, progress=gr.Progress(track_tqdm=True)):
+  print("Run this?")
   repo_id_1 = shuffled_items[0]['repo']
   repo_id_2 = shuffled_items[1]['repo']
   print("Loading state dicts...")
@@ -117,7 +119,11 @@ def merge_and_run(prompt, negative_prompt, shuffled_items, lora_1_scale=0.5, lor
 def get_description(item):
       trigger_word = item["trigger_word"]
       return f"Trigger: `{trigger_word}`" if trigger_word else "No trigger, applied automatically", trigger_word
-    
+
+
+def truncate_string(s, max_length=30):
+    return s[:max_length - 3] + "..." if len(s) > max_length else s
+
 def shuffle_images():
     compatible_items = [item for item in sdxl_loras if item['is_compatible']]
     random.shuffle(compatible_items)
@@ -129,8 +135,8 @@ def shuffle_images():
     description_1, trigger_word_1 = get_description(two_shuffled_items[0])
     description_2, trigger_word_2 = get_description(two_shuffled_items[1])
 
-    lora_1_link = f"[{two_shuffled_items[0]['repo']}](https://huggingface.co/{two_shuffled_items[0]['repo']}) ✨"
-    lora_2_link = f"[{two_shuffled_items[1]['repo']}](https://huggingface.co/{two_shuffled_items[1]['repo']}) ✨"
+    lora_1_link = f"[{truncate_string(two_shuffled_items[0]['repo'])}](https://huggingface.co/{two_shuffled_items[0]['repo']}) ✨"
+    lora_2_link = f"[{truncate_string(two_shuffled_items[1]['repo'])}](https://huggingface.co/{two_shuffled_items[1]['repo']}) ✨"
     prompt_description_1 = gr.update(value=description_1, visible=True)
     prompt_description_2 = gr.update(value=description_2, visible=True)
     prompt = gr.update(value=f"{trigger_word_1} {trigger_word_2}")
@@ -160,14 +166,14 @@ with gr.Blocks(css=css) as demo:
     with gr.Column(min_width=10, scale=16, elem_classes="plus_column"):
         with gr.Row():
             with gr.Column(min_width=10, scale=4, elem_classes="random_column"):
-              lora_1_link = gr.Markdown()
+              lora_1_link = gr.Markdown(elem_classes="title_lora")
               lora_1 = gr.Image(interactive=False, show_label=False, height=150, elem_classes="selected_random", elem_id="randomLoRA_1", show_share_button=False, show_download_button=False)
               lora_1_id = gr.Textbox(visible=False, elem_id="random_lora_1_id")
               lora_1_prompt = gr.Markdown(visible=False)
             with gr.Column(min_width=10, scale=1, elem_classes="plus_column"):
               plus = gr.HTML("+", elem_classes="plus_button")
             with gr.Column(min_width=10, scale=4, elem_classes="random_column"):
-              lora_2_link = gr.Markdown()
+              lora_2_link = gr.Markdown(elem_classes="title_lora")
               lora_2 = gr.Image(interactive=False, show_label=False, height=150, elem_classes="selected_random", elem_id="randomLoRA_2", show_share_button=False, show_download_button=False)
               lora_2_id = gr.Textbox(visible=False, elem_id="random_lora_2_id")
               lora_2_prompt = gr.Markdown(visible=False)