Update app.py

app.py

@@ -440,60 +440,56 @@ def get_huggingface_safetensors(link):
               raise Exception(f"You didn't include a link neither a valid Hugging Face repository with a *.safetensors LoRA")
             return split_link[1], link, safetensors_name, trigger_word, image_url
 
-#def check_custom_model(link):
-#    if(link.startswith("https://")):
-#        if(link.startswith("https://huggingface.co") or link.startswith("https://www.huggingface.co")):
-#            link_split = link.split("huggingface.co/")
-#            return get_huggingface_safetensors(link_split[1])
-#    else: 
-#        return get_huggingface_safetensors(link)
-
-# def add_custom_lora(custom_lora):
-#     global loras
-#     if(custom_lora):
-#         try:
-#             title, repo, path, trigger_word, image = check_custom_model(custom_lora)
-#             print(f"Modelo Externo: {repo}")
-#             card = f'''
-#             <div class="custom_lora_card">
-#               <span>Loaded custom LoRA:</span>
-#               <div class="card_internal">
-#                 <img src="{image}" />
-#                 <div>
-#                     <h3>{title}</h3>
-#                    <small>{"Usando: <code><b>"+trigger_word+"</code></b> como palavra-chave" if trigger_word else "Não encontramos a palavra-chave, se tiver, coloque-a no prompt."}<br></small>
-#                 </div>
-#               </div>
-#            </div>
-#            '''
-#            existing_item_index = next((index for (index, item) in enumerate(loras) if item['repo'] == repo), None)
-#            if(not existing_item_index):
-#                new_item = {
-#                    "image": image,
-#                    "title": title,
-#                    "repo": repo,
-#                    "weights": path,
-#                    "trigger_word": trigger_word
-#                }
-#                print(new_item)
-#                existing_item_index = len(loras)
-#                loras.append(new_item)
-#        
-#            return gr.update(visible=True, value=card), gr.update(visible=True), gr.Gallery(selected_index=None), f"Custom: {path}", existing_item_index, trigger_word
-#        except Exception as e:
-#            gr.Warning(f"Modelo Inválido: ou o link está errado ou não é um FLUX")
-#            return gr.update(visible=True, value=f"Modelo Inválido: ou o link está errado ou não é um FLUX"), gr.update(visible=False), gr.update(), "", None, ""
-#    else:
-#        return gr.update(visible=False), gr.update(visible=False), gr.update(), "", None, ""
-
-# def remove_custom_lora():
-#    return gr.update(visible=False), gr.update(visible=False), gr.update(), "", None, ""
-
-run_lora.zerogpu = True
-
-#import gradio as gr
+def check_custom_model(link):
+    if(link.startswith("https://")):
+        if(link.startswith("https://huggingface.co") or link.startswith("https://www.huggingface.co")):
+            link_split = link.split("huggingface.co/")
+            return get_huggingface_safetensors(link_split[1])
+    else: 
+        return get_huggingface_safetensors(link)
+
+def add_custom_lora(custom_lora):
+     global loras
+     if(custom_lora):
+         try:
+             title, repo, path, trigger_word, image = check_custom_model(custom_lora)
+             print(f"Modelo Externo: {repo}")
+             card = f'''
+             <div class="custom_lora_card">
+               <span>Loaded custom LoRA:</span>
+               <div class="card_internal">
+                 <img src="{image}" />
+                 <div>
+                     <h3>{title}</h3>
+                    <small>{"Usando: <code><b>"+trigger_word+"</code></b> como palavra-chave" if trigger_word else "Não encontramos a palavra-chave, se tiver, coloque-a no prompt."}<br></small>
+                 </div>
+               </div>
+            </div>
+            '''
+            existing_item_index = next((index for (index, item) in enumerate(loras) if item['repo'] == repo), None)
+            if(not existing_item_index):
+                new_item = {
+                    "image": image,
+                    "title": title,
+                    "repo": repo,
+                    "weights": path,
+                    "trigger_word": trigger_word
+                }
+                print(new_item)
+                existing_item_index = len(loras)
+                loras.append(new_item)
+        
+            return gr.update(visible=True, value=card), gr.update(visible=True), gr.Gallery(selected_index=None), f"Custom: {path}", existing_item_index, trigger_word
+        except Exception as e:
+            gr.Warning(f"Modelo Inválido: ou o link está errado ou não é um FLUX")
+            return gr.update(visible=True, value=f"Modelo Inválido: ou o link está errado ou não é um FLUX"), gr.update(visible=False), gr.update(), "", None, ""
+    else:
+        return gr.update(visible=False), gr.update(visible=False), gr.update(), "", None, ""
 
+def remove_custom_lora():
+     return gr.update(visible=False), gr.update(visible=False), gr.update(), "", None, ""
 
+run_lora.zerogpu = True
 
 
 collos = gr.themes.Soft(
@@ -550,11 +546,11 @@ with gr.Blocks(theme=collos, delete_cache=(60, 60)) as app:
                 columns=3,
                 show_share_button=False
             )
-#            with gr.Group():
-#                custom_lora = gr.Textbox(label="Selecione um Modelo Externo", placeholder="black-forest-labs/FLUX.1-dev")
-#                gr.Markdown("[Cheque a lista de modelos do Huggingface](https://huggingface.co/models?other=base_model:adapter:black-forest-labs/FLUX.1-dev)", elem_id="lora_list")
-#            custom_lora_info = gr.HTML(visible=False)
-#            custom_lora_button = gr.Button("Remova modelo Externo", visible=False)
+            with gr.Group():
+                custom_lora = gr.Textbox(label="Selecione um Modelo Externo", placeholder="black-forest-labs/FLUX.1-dev")
+                gr.Markdown("[Cheque a lista de modelos do Huggingface](https://huggingface.co/models?other=base_model:adapter:black-forest-labs/FLUX.1-dev)", elem_id="lora_list")
+            custom_lora_info = gr.HTML(visible=False)
+            custom_lora_button = gr.Button("Remova modelo Externo", visible=False)
             with gr.Group():
                         with gr.Group():
                             gr.Text(value="Dica: Sempre digite a Palavra-chave referente ao modelo que são: José Alves, Júlio, Pedro e Clóvis, com os devidos acentos.", elem_id="padded_text")
@@ -587,15 +583,15 @@ with gr.Blocks(theme=collos, delete_cache=(60, 60)) as app:
         inputs=[width, height],
         outputs=[prompt, selected_info, selected_index, width, height]
     )
-#    custom_lora.input(
-#        add_custom_lora,
-#        inputs=[custom_lora],
-#        outputs=[custom_lora_info, custom_lora_button, gallery, selected_info, selected_index, prompt]
-#    )
-#    custom_lora_button.click(
-#        remove_custom_lora,
-#        outputs=[custom_lora_info, custom_lora_button, gallery, selected_info, selected_index, custom_lora]
-#    )
+    custom_lora.input(
+        add_custom_lora,
+        inputs=[custom_lora],
+        outputs=[custom_lora_info, custom_lora_button, gallery, selected_info, selected_index, prompt]
+   )
+    custom_lora_button.click(
+        remove_custom_lora,
+        outputs=[custom_lora_info, custom_lora_button, gallery, selected_info, selected_index, custom_lora]
+    )
     gr.on(
         triggers=[generate_button.click, prompt.submit],
         fn=run_lora,