Fix searching for civitai loras

Author: Metal079

src/app/home/add-loras/add-loras.component.html

@@ -22,6 +22,7 @@
       pInputText 
       [(ngModel)]="searchField" 
       placeholder="Search LoRA by Name" 
+      (keyup.enter)="search()"
     />
 
     <input 
@@ -30,6 +31,7 @@
       pInputText 
       [(ngModel)]="searchField" 
       placeholder="Enter LoRA Model ID" 
+      (keyup.enter)="search()"
     />
 
     <input 
@@ -38,8 +40,16 @@
       pInputText 
       [(ngModel)]="searchField" 
       placeholder="Enter LoRA Creator's CivitAI Username"
+      (keyup.enter)="search()"
     />
   </div>
+
+  <div class="p-field d-flex justify-content-end align-items-center" style="margin-top: 8px;">
+    <span class="d-flex align-items-center">
+      Show NSFW results?
+      <p-inputSwitch [(ngModel)]="showNSFWLoras" [class]="'ms-2'" (ngModelChange)="onShowNsfwChanged()"></p-inputSwitch>
+    </span>
+  </div>
   <div>
     <button 
     pButton 

src/app/home/add-loras/add-loras.component.ts

@@ -1,4 +1,4 @@
-import { Component } from '@angular/core';
+import { Component, OnInit } from '@angular/core';
 import { DynamicDialogRef } from 'primeng/dynamicdialog';
 import { DynamicDialogConfig } from 'primeng/dynamicdialog';
 import { StableDiffusionService } from 'src/app/stable-diffusion.service';
@@ -17,6 +17,8 @@ export class AddLorasComponent {
   loraFile: File | null = null;
   searchField: string = ''; // New field for search field
 
+  showNSFWLoras: boolean = false;
+
   searchResults: any[] = []; // Will hold the API search results
   selectedLoRA: any = null;  // Holds the selected LoRA from the table
   showImageDialog: boolean = false; // Track the state of the image dialog
@@ -32,6 +34,21 @@ export class AddLorasComponent {
     , private sharedService: SharedService
   ) { }
 
+  ngOnInit(): void {
+    const fromParent = this.config?.data?.showNSFWLoras;
+    if (typeof fromParent === 'boolean') {
+      this.showNSFWLoras = fromParent;
+      return;
+    }
+
+    const stored = localStorage.getItem('showNSFWLoras');
+    if (stored != null) this.showNSFWLoras = stored === 'true';
+  }
+
+  onShowNsfwChanged(): void {
+    localStorage.setItem('showNSFWLoras', this.showNSFWLoras.toString());
+  }
+
   selectOption(option: string) {
     this.selectedSearchOption = option;
     this.searchResults = [];
@@ -47,6 +64,8 @@ export class AddLorasComponent {
   }
 
   search() {
+    if (this.isLoading || !this.searchField) return;
+
     if (this.selectedSearchOption === 'username') {
       this.searchByUsername();
     } else if (this.selectedSearchOption === 'lora_name') {
@@ -58,7 +77,7 @@ export class AddLorasComponent {
 
   searchByUsername() {
     this.isLoading = true; // Start loading
-    this.stableDiffusionService.searchByUser(this.searchField).subscribe({
+    this.stableDiffusionService.searchByUser(this.searchField, this.showNSFWLoras).subscribe({
       next: (response: any) => {
         console.log('civitAi user search done');
         this.searchResults = response;
@@ -80,7 +99,7 @@ export class AddLorasComponent {
 
   searchByLoRAName() {
     this.isLoading = true; // Start loading
-    this.stableDiffusionService.searchByQuery(this.searchField).subscribe({
+    this.stableDiffusionService.searchByQuery(this.searchField, this.showNSFWLoras).subscribe({
       next: (response: any) => {
         console.log('civitAi query search done');
         this.searchResults = response;
@@ -103,7 +122,7 @@ export class AddLorasComponent {
   // New method to search by Model ID
   searchByModelId() {
     this.isLoading = true; // Start loading
-    this.stableDiffusionService.searchByID(this.searchField).subscribe({
+    this.stableDiffusionService.searchByID(this.searchField, this.showNSFWLoras).subscribe({
       next: (response: any) => {
         this.searchResults = response;
 
@@ -159,7 +178,7 @@ export class AddLorasComponent {
         }
       });
     }
-    this.ref.close();
+    this.ref.close({ showNSFWLoras: this.showNSFWLoras });
   }
 
   selectLora(lora: any) {
@@ -174,7 +193,7 @@ export class AddLorasComponent {
 
 
   close() {
-    this.ref.close();
+    this.ref.close({ showNSFWLoras: this.showNSFWLoras });
   }
 
   showError(error: any) {

src/app/home/options/options.component.ts

@@ -1914,9 +1914,21 @@ export class OptionsComponent implements OnInit {
     }
 
     // Open the dialog with the dynamic width
-    this.dialogService.open(AddLorasComponent, {
+    const dialogRef = this.dialogService.open(AddLorasComponent, {
       header: 'Request Loras To Be Added!',
-      width: dialogWidth
+      width: dialogWidth,
+      data: {
+        showNSFWLoras: this.showNSFWLoras,
+      },
+    });
+
+    dialogRef.onClose.subscribe((result: any) => {
+      const next = result?.showNSFWLoras;
+      if (typeof next === 'boolean' && next !== this.showNSFWLoras) {
+        this.showNSFWLoras = next;
+        localStorage.setItem('showNSFWLoras', this.showNSFWLoras.toString());
+        this.filterLoras();
+      }
     });
   }
 

src/app/stable-diffusion.service.ts

@@ -138,18 +138,18 @@ export class StableDiffusionService {
   //#endregion
 
   //#region CivitAI API calls
-  searchByQuery(query: string): Observable<any> {
-    const url = `${this.apiBaseUrl}/search_civitAi_loras_by_query/${query}`;  // note the trailing slash
+  searchByQuery(query: string, showNsfw: boolean = false): Observable<any> {
+    const url = `${this.apiBaseUrl}/search_civitAi_loras_by_query/${encodeURIComponent(query)}?show_nsfw=${showNsfw}`;
     return this.http.get(url);
   }
 
-  searchByID(id: string): Observable<any> {
-    const url = `${this.apiBaseUrl}/search_civitAi_loras_by_id/${id}`;  // note the trailing slash
+  searchByID(id: string, showNsfw: boolean = false): Observable<any> {
+    const url = `${this.apiBaseUrl}/search_civitAi_loras_by_id/${encodeURIComponent(id)}?show_nsfw=${showNsfw}`;
     return this.http.get(url);
   }
 
-  searchByUser(username: string): Observable<any> {
-    const url = `${this.apiBaseUrl}/search_civitAi_loras_by_user/${username}`;  // note the trailing slash
+  searchByUser(username: string, showNsfw: boolean = false): Observable<any> {
+    const url = `${this.apiBaseUrl}/search_civitAi_loras_by_user/${encodeURIComponent(username)}?show_nsfw=${showNsfw}`;
     return this.http.get(url);
   }