@@ -142,10 +141,20 @@ export function showLoraModal(lora) {
-
${lora.usage_tips || 'Save usage tips here..'}
-
+
+
+
+
+
+
+ ${renderPresetTags(parsePresets(lora.usage_tips))}
+
${renderTriggerWords(escapedWords)}
@@ -195,25 +204,57 @@ function setupEditableFields() {
}
});
});
-}
-// Add these functions to handle saving the editable fields
-window.saveUsageTips = async function(filePath) {
- const content = document.querySelector('.usage-tips-content').textContent;
- try {
- await saveModelMetadata(filePath, { usage_tips: content });
+ const presetSelector = document.getElementById('preset-selector');
+ const presetValue = document.getElementById('preset-value');
+ const addPresetBtn = document.querySelector('.add-preset-btn');
+ const presetTags = document.querySelector('.preset-tags');
- // Update the corresponding lora card's dataset
- const loraCard = document.querySelector(`.lora-card[data-filepath="${filePath}"]`);
- if (loraCard) {
- loraCard.dataset.usage_tips = content;
+ presetSelector.addEventListener('change', function() {
+ const selected = this.value;
+ if (selected) {
+ presetValue.style.display = 'inline-block';
+ presetValue.min = selected.includes('strength') ? 0 : 1;
+ presetValue.max = selected.includes('strength') ? 1 : 12;
+ presetValue.step = selected.includes('strength') ? 0.01 : 1;
+ if (selected === 'clip_skip') {
+ presetValue.type = 'number';
+ presetValue.step = 1;
+ }
+ } else {
+ presetValue.style.display = 'none';
}
+ });
- showToast('Usage tips saved successfully', 'success');
- } catch (error) {
- showToast('Failed to save usage tips', 'error');
- }
-};
+ addPresetBtn.addEventListener('click', async function() {
+ const key = presetSelector.value;
+ const value = presetValue.value;
+
+ if (!key || !value) return;
+
+ const filePath = document.querySelector('.modal-content')
+ .querySelector('.file-path').textContent +
+ document.querySelector('.modal-content')
+ .querySelector('#file-name').textContent + '.safetensors';
+
+ const loraCard = document.querySelector(`.lora-card[data-filepath="${filePath}"]`);
+ const currentPresets = parsePresets(loraCard.dataset.usage_tips);
+
+ currentPresets[key] = parseFloat(value);
+ const newPresetsJson = JSON.stringify(currentPresets);
+
+ await saveModelMetadata(filePath, {
+ usage_tips: newPresetsJson
+ });
+
+ loraCard.dataset.usage_tips = newPresetsJson;
+ presetTags.innerHTML = renderPresetTags(currentPresets);
+
+ presetSelector.value = '';
+ presetValue.value = '';
+ presetValue.style.display = 'none';
+ });
+}
window.saveNotes = async function(filePath) {
const content = document.querySelector('.notes-content').textContent;
@@ -225,7 +266,7 @@ window.saveNotes = async function(filePath) {
if (loraCard) {
loraCard.dataset.notes = content;
}
-
+
showToast('Notes saved successfully', 'success');
} catch (error) {
showToast('Failed to save notes', 'error');
@@ -426,4 +467,47 @@ export function scrollToTop(button) {
behavior: 'smooth'
});
}
-}
\ No newline at end of file
+}
+
+function parsePresets(usageTips) {
+ if (!usageTips || usageTips === 'Save usage tips here..') return {};
+ try {
+ return JSON.parse(usageTips);
+ } catch {
+ return {};
+ }
+}
+
+function renderPresetTags(presets) {
+ return Object.entries(presets).map(([key, value]) => `
+
+ ${formatPresetKey(key)}: ${value}
+
+
+ `).join('');
+}
+
+function formatPresetKey(key) {
+ return key.split('_').map(word =>
+ word.charAt(0).toUpperCase() + word.slice(1)
+ ).join(' ');
+}
+
+window.removePreset = async function(key) {
+ const filePath = document.querySelector('.modal-content')
+ .querySelector('.file-path').textContent +
+ document.querySelector('.modal-content')
+ .querySelector('#file-name').textContent + '.safetensors';
+ const loraCard = document.querySelector(`.lora-card[data-filepath="${filePath}"]`);
+ const currentPresets = parsePresets(loraCard.dataset.usage_tips);
+
+ delete currentPresets[key];
+ const newPresetsJson = JSON.stringify(currentPresets);
+
+ await saveModelMetadata(filePath, {
+ usage_tips: newPresetsJson
+ });
+
+ loraCard.dataset.usage_tips = newPresetsJson;
+ document.querySelector('.preset-tags').innerHTML = renderPresetTags(currentPresets);
+};
\ No newline at end of file
diff --git a/utils/models.py b/utils/models.py
index 22af5f5b..0a39e065 100644
--- a/utils/models.py
+++ b/utils/models.py
@@ -15,7 +15,7 @@ class LoraMetadata:
sha256: str # SHA256 hash of the file
base_model: str # Base model (SD1.5/SD2.1/SDXL/etc.)
preview_url: str # Preview image URL
- usage_tips: str = "" # Usage tips for the model
+ usage_tips: str = "{}" # Usage tips for the model, json string
notes: str = "" # Additional notes
from_civitai: bool = True # Whether the lora is from Civitai
civitai: Optional[Dict] = None # Civitai API data if available