Update app.py

app.py

@@ -449,40 +449,56 @@ def check_custom_model(link):
         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>
-            '''
+    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):
+            if not existing_item_index:
                 new_item = {
                     "image": image,
                     "title": title,
                     "repo": repo,
                     "weights": path,
-                    "trigger_word": trigger_word
+                    "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
+
+            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, ""
+            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, ""