🏖️ Summer Update: Retro cleanup, QOL, optimized JPEG output
This commit is contained in:
mrkmntal
parent
7c6a3f7bc4
commit
cb3ccd80c0
6 changed files with 421 additions and 392 deletions
386
index.php
386
index.php
|
|
@ -2,22 +2,8 @@
|
|||
ini_set('max_execution_time', '300');
|
||||
header('Content-Type: text/html; charset=iso-8859-1');
|
||||
|
||||
$response = "";
|
||||
$generated_image = "";
|
||||
$prompt = "";
|
||||
$negative_prompt = "";
|
||||
$width = 768;
|
||||
$height = 768;
|
||||
$steps = 35;
|
||||
$cfg_scale = 3;
|
||||
$sampler = "DPM++ 3M SDE";
|
||||
$error_message = "";
|
||||
$success_message = "";
|
||||
|
||||
// File to store saved prompts
|
||||
$saved_prompts_file = 'saved-prompts.json';
|
||||
|
||||
// Load saved prompts
|
||||
function loadSavedPrompts() {
|
||||
global $saved_prompts_file;
|
||||
if (file_exists($saved_prompts_file)) {
|
||||
|
|
@ -27,366 +13,102 @@ function loadSavedPrompts() {
|
|||
return array();
|
||||
}
|
||||
|
||||
// Save a new prompt
|
||||
function savePrompt($prompt_text, $negative_prompt_text, $prompt_name) {
|
||||
global $saved_prompts_file;
|
||||
$prompts = loadSavedPrompts();
|
||||
$prompts[$prompt_name] = array(
|
||||
'prompt' => $prompt_text,
|
||||
'negative_prompt' => $negative_prompt_text
|
||||
);
|
||||
file_put_contents($saved_prompts_file, json_encode($prompts, JSON_PRETTY_PRINT));
|
||||
}
|
||||
|
||||
$saved_prompts = loadSavedPrompts();
|
||||
|
||||
// List of available samplers
|
||||
$samplers = array(
|
||||
"Euler",
|
||||
"Euler a",
|
||||
"Heun",
|
||||
"DPM2",
|
||||
"DPM2 a",
|
||||
"DPM++ 2S a",
|
||||
"DPM++ 2M",
|
||||
"DPM++ SDE",
|
||||
"DPM++ 2M SDE",
|
||||
"DPM++ 3M SDE",
|
||||
"DPM fast",
|
||||
"DPM adaptive",
|
||||
"LMS",
|
||||
"DPM++ SDE Karras",
|
||||
"DPM++ 2M SDE Karras",
|
||||
"DPM++ 3M SDE Karras",
|
||||
"DDIM",
|
||||
"PLMS"
|
||||
);
|
||||
$prompt = '';
|
||||
$negative_prompt = '';
|
||||
$width = 768;
|
||||
$height = 768;
|
||||
$steps = 35;
|
||||
$cfg_scale = 3;
|
||||
$sampler = "Euler a";
|
||||
|
||||
if ($_SERVER["REQUEST_METHOD"] == "POST") {
|
||||
// All requests to the server to save or execute prompts will be POST requests to itself
|
||||
// Handle prompt saving
|
||||
if (isset($_POST['save_prompt']) && isset($_POST['prompt_name'])) {
|
||||
$prompt_to_save = trim($_POST['prompt']);
|
||||
$negative_prompt_to_save = trim($_POST['negative_prompt']);
|
||||
$prompt_name = trim($_POST['prompt_name']);
|
||||
|
||||
if (!empty($prompt_to_save) && !empty($prompt_name)) {
|
||||
savePrompt($prompt_to_save, $negative_prompt_to_save, $prompt_name);
|
||||
$success_message = "Prompt saved successfully!";
|
||||
$saved_prompts = loadSavedPrompts(); // Reload prompts
|
||||
} else {
|
||||
$error_message = "Both prompt and prompt name are required to save.";
|
||||
}
|
||||
}
|
||||
// Handle image generation
|
||||
else {
|
||||
$prompt = isset($_POST['prompt']) ? $_POST['prompt'] : '';
|
||||
$negative_prompt = isset($_POST['negative_prompt']) ? $_POST['negative_prompt'] : '';
|
||||
$width = isset($_POST['width']) ? (int)$_POST['width'] : 512;
|
||||
$height = isset($_POST['height']) ? (int)$_POST['height'] : 512;
|
||||
$steps = isset($_POST['steps']) ? (int)$_POST['steps'] : 20;
|
||||
$cfg_scale = isset($_POST['cfg_scale']) ? (float)$_POST['cfg_scale'] : 7.5;
|
||||
$sampler = isset($_POST['sampler']) ? $_POST['sampler'] : 'Euler';
|
||||
|
||||
// Check for angle brackets in the prompt
|
||||
if (strpos($prompt, '<') !== false || strpos($prompt, '>') !== false) {
|
||||
$error_message = "Error: Angle brackets are not allowed in the prompt. Please use square brackets [lora-name:weight] for LORAs.";
|
||||
}
|
||||
// Only proceed if there's no error
|
||||
elseif (!empty($prompt)) {
|
||||
// Process the prompt to handle LORA syntax
|
||||
$processed_prompt = preg_replace('/\[([\w\s\-]+?):([\d\.]+)\]/', '<lora:$1:$2>', $prompt);
|
||||
|
||||
$data = array(
|
||||
'prompt' => $processed_prompt,
|
||||
'negative_prompt' => $negative_prompt,
|
||||
'steps' => $steps,
|
||||
'width' => $width,
|
||||
'height' => $height,
|
||||
'cfg_scale' => $cfg_scale,
|
||||
'sampler_name' => $sampler,
|
||||
'override_settings' => new stdClass(),
|
||||
'override_settings_restore_afterwards' => true
|
||||
);
|
||||
|
||||
// Replace 127.0.0.1 with your Stable diffusion server's IP if running externally
|
||||
$ch = curl_init('http://127.0.0.1:7860/sdapi/v1/txt2img');
|
||||
curl_setopt($ch, CURLOPT_RETURNTRANSFER, true);
|
||||
curl_setopt($ch, CURLOPT_POST, true);
|
||||
curl_setopt($ch, CURLOPT_POSTFIELDS, json_encode($data));
|
||||
curl_setopt($ch, CURLOPT_HTTPHEADER, array('Content-Type: application/json'));
|
||||
|
||||
$result = curl_exec($ch);
|
||||
|
||||
if (curl_errno($ch)) {
|
||||
$response = 'Error: ' . curl_error($ch);
|
||||
} else {
|
||||
$decoded = json_decode($result, true);
|
||||
if (isset($decoded['images'][0])) {
|
||||
$image_data = base64_decode($decoded['images'][0]);
|
||||
|
||||
// Save original PNG temporarily
|
||||
$temp_png = 'temp_' . time() . '_' . bin2hex(random_bytes(8)) . '.png';
|
||||
file_put_contents($temp_png, $image_data);
|
||||
|
||||
// Create output JPG filename
|
||||
$generated_image = 'generated_' . time() . '_' . bin2hex(random_bytes(8)) . '.jpg';
|
||||
|
||||
// Build ImageMagick convert command with optimization
|
||||
$quality = 85; // Balanced quality setting
|
||||
$cmd = "convert \"$temp_png\" -strip -interlace Plane -gaussian-blur 0.05 -quality $quality ";
|
||||
$cmd .= "-sampling-factor 4:2:0 \"$generated_image\" 2>&1";
|
||||
|
||||
// Execute conversion
|
||||
$output = [];
|
||||
$returnVar = 0;
|
||||
exec($cmd, $output, $returnVar);
|
||||
|
||||
if ($returnVar === 0 && file_exists($generated_image)) {
|
||||
// Get file sizes for comparison
|
||||
$originalSize = filesize($temp_png);
|
||||
$convertedSize = filesize($generated_image);
|
||||
|
||||
// Clean up temporary PNG file
|
||||
if (file_exists($temp_png)) {
|
||||
unlink($temp_png);
|
||||
}
|
||||
|
||||
// Clean up old generated files
|
||||
foreach (glob("generated_*.png") as $file) {
|
||||
if ($file != $generated_image && (time() - filemtime($file)) > 3600) {
|
||||
unlink($file);
|
||||
}
|
||||
}
|
||||
|
||||
// Store size information for display
|
||||
$size_info = array(
|
||||
'original' => round($originalSize / 1024, 2),
|
||||
'converted' => round($convertedSize / 1024, 2),
|
||||
'reduction' => round((($originalSize - $convertedSize) / $originalSize) * 100, 2)
|
||||
);
|
||||
} else {
|
||||
// If conversion fails, keep the original PNG
|
||||
if (file_exists($temp_png)) {
|
||||
$generated_image = $temp_png;
|
||||
}
|
||||
$error_message = "Image conversion failed. Using original PNG format.";
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
curl_close($ch);
|
||||
}
|
||||
}
|
||||
} else if ($_SERVER["REQUEST_METHOD"] == "GET" && isset($_GET['load_prompt'])) {
|
||||
// For loading prompts we use GET method to send the client's choice of prompt to load
|
||||
// Load saved prompt into textarea
|
||||
$saved_prompts = loadSavedPrompts();
|
||||
if ($_SERVER["REQUEST_METHOD"] == "GET" && isset($_GET['load_prompt'])) {
|
||||
if (isset($saved_prompts[$_GET['load_prompt']])) {
|
||||
// Check if the saved prompt is in the new format (array) or old format (string)
|
||||
if (is_array($saved_prompts[$_GET['load_prompt']])) {
|
||||
$prompt = $saved_prompts[$_GET['load_prompt']]['prompt'];
|
||||
$negative_prompt = $saved_prompts[$_GET['load_prompt']]['negative_prompt'];
|
||||
} else {
|
||||
// Handle old format
|
||||
$prompt = $saved_prompts[$_GET['load_prompt']];
|
||||
$negative_prompt = '';
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
$samplers = array(
|
||||
"Euler", "Euler a", "Heun", "DPM2", "DPM2 a", "DPM++ 2S a", "DPM++ 2M",
|
||||
"DPM++ SDE", "DPM++ 2M SDE", "DPM++ 3M SDE", "DPM fast", "DPM adaptive",
|
||||
"LMS", "DPM++ SDE Karras", "DPM++ 2M SDE Karras", "DPM++ 3M SDE Karras",
|
||||
"DDIM", "PLMS"
|
||||
);
|
||||
?>
|
||||
<!DOCTYPE HTML PUBLIC "-//W3C//DTD HTML 4.01 Transitional//EN" "http://www.w3.org/TR/html4/loose.dtd">
|
||||
<!DOCTYPE HTML>
|
||||
<html>
|
||||
<head>
|
||||
<title>Stable Diffusion Image Generator</title>
|
||||
<meta http-equiv="Content-Type" content="text/html; charset=iso-8859-1">
|
||||
<style type="text/css">
|
||||
.lora-help {
|
||||
background-color: #f8f8f8;
|
||||
padding: 10px;
|
||||
margin: 10px 0;
|
||||
border: 1px solid #ddd;
|
||||
}
|
||||
.form-group {
|
||||
margin-bottom: 15px;
|
||||
}
|
||||
textarea#prompt {
|
||||
width: 100%;
|
||||
height: 100px;
|
||||
padding: 8px;
|
||||
font-family: Arial, sans-serif;
|
||||
font-size: 14px;
|
||||
border: 1px solid #ddd;
|
||||
}
|
||||
textarea#negative_prompt {
|
||||
width: 100%;
|
||||
height: 100px;
|
||||
padding: 8px;
|
||||
font-family: Arial, sans-serif;
|
||||
font-size: 14px;
|
||||
border: 1px solid #ddd;
|
||||
background-color: #fff3f3;
|
||||
}
|
||||
.error-message {
|
||||
color: #dc3545;
|
||||
background-color: #f8d7da;
|
||||
border: 1px solid #f5c6cb;
|
||||
padding: 10px;
|
||||
margin: 10px 0;
|
||||
}
|
||||
.success-message {
|
||||
color: #28a745;
|
||||
background-color: #d4edda;
|
||||
border: 1px solid #c3e6cb;
|
||||
padding: 10px;
|
||||
margin: 10px 0;
|
||||
}
|
||||
.saved-prompts {
|
||||
margin: 20px 0;
|
||||
padding: 15px;
|
||||
background-color: #f8f8f8;
|
||||
border: 1px solid #ddd;
|
||||
}
|
||||
.save-prompt-form {
|
||||
margin: 20px 0;
|
||||
padding: 15px;
|
||||
background-color: #f8f8f8;
|
||||
border: 1px solid #ddd;
|
||||
}
|
||||
</style>
|
||||
</head>
|
||||
<body>
|
||||
<?php if (isset($_GET['save_success'])): ?>
|
||||
<div style="color:green;">Prompt saved successfully.</div>
|
||||
<?php elseif (isset($_GET['save_error'])): ?>
|
||||
<div style="color:red;">Error: Prompt and name are required.</div>
|
||||
<?php endif; ?>
|
||||
|
||||
<h1>Stable Diffusion Image Generator</h1>
|
||||
|
||||
<div class="lora-help">
|
||||
<h3>How to use LORAs:</h3>
|
||||
<p>Include LORAs in your prompt using this syntax: <code>[lora-name:weight]</code></p>
|
||||
<p>Example: <code>a beautiful landscape by [my-artist-lora:0.8]</code></p>
|
||||
<p>Weight values typically range from 0.1 to 1.0</p>
|
||||
<p><strong>Note:</strong> Angle brackets (< >) are not allowed in prompts. Use square brackets instead.</p>
|
||||
<div style="border:1px solid #ccc;padding:10px;margin-bottom:10px;">
|
||||
<strong>How to use LORAs:</strong>
|
||||
<p>Use square bracket format: <code>[lora-name:weight]</code></p>
|
||||
<p>Example: <code>a detailed castle, [my-style-lora:0.7]</code></p>
|
||||
<p><b>Note:</b> Angle brackets are not allowed.</p>
|
||||
</div>
|
||||
|
||||
<?php if (!empty($error_message)): ?>
|
||||
<div class="error-message">
|
||||
<?php echo htmlspecialchars($error_message); ?>
|
||||
</div>
|
||||
<?php endif; ?>
|
||||
|
||||
<?php if (!empty($success_message)): ?>
|
||||
<div class="success-message">
|
||||
<?php echo htmlspecialchars($success_message); ?>
|
||||
</div>
|
||||
<?php endif; ?>
|
||||
|
||||
<div class="saved-prompts">
|
||||
<h3>Saved Prompts</h3>
|
||||
<form method="get" action="<?php echo htmlspecialchars($_SERVER["PHP_SELF"]); ?>">
|
||||
<select name="load_prompt">
|
||||
<option value="">Select a saved prompt</option>
|
||||
<?php foreach ($saved_prompts as $name => $saved_prompt): ?>
|
||||
<option value="<?php echo htmlspecialchars($name); ?>"
|
||||
title="Prompt: <?php echo htmlspecialchars(is_array($saved_prompt) ? $saved_prompt['prompt'] : $saved_prompt); ?> Negative: <?php echo htmlspecialchars(is_array($saved_prompt) ? $saved_prompt['negative_prompt'] : ''); ?>">
|
||||
<div>
|
||||
<form method="get" action="index.php">
|
||||
<label for="load_prompt">Load Saved Prompt:</label>
|
||||
<select name="load_prompt" id="load_prompt">
|
||||
<option value="">-- Choose --</option>
|
||||
<?php foreach ($saved_prompts as $name => $data): ?>
|
||||
<option value="<?php echo htmlspecialchars($name); ?>">
|
||||
<?php echo htmlspecialchars($name); ?>
|
||||
</option>
|
||||
<?php endforeach; ?>
|
||||
</select>
|
||||
<input type="submit" value="Load Prompt">
|
||||
<input type="submit" value="Load">
|
||||
</form>
|
||||
</div>
|
||||
|
||||
<form method="post" action="<?php echo htmlspecialchars($_SERVER["PHP_SELF"]); ?>">
|
||||
<div class="form-group">
|
||||
<label for="prompt">Enter your prompt (including LORA syntax if needed):</label><br>
|
||||
<textarea name="prompt" id="prompt"><?php echo htmlspecialchars($prompt); ?></textarea>
|
||||
</div>
|
||||
<form method="post" action="loading.php">
|
||||
<p><label for="prompt">Prompt:</label><br>
|
||||
<textarea name="prompt" id="prompt" rows="5" cols="60"><?php echo htmlspecialchars($prompt); ?></textarea></p>
|
||||
|
||||
<div class="form-group">
|
||||
<label for="negative_prompt">Enter your negative prompt (things to avoid in the image):</label><br>
|
||||
<textarea name="negative_prompt" id="negative_prompt"><?php echo htmlspecialchars($negative_prompt); ?></textarea>
|
||||
</div>
|
||||
<p><label for="negative_prompt">Negative Prompt:</label><br>
|
||||
<textarea name="negative_prompt" id="negative_prompt" rows="5" cols="60"><?php echo htmlspecialchars($negative_prompt); ?></textarea></p>
|
||||
|
||||
<div class="form-group">
|
||||
<label for="width">Width:</label>
|
||||
<input type="number" name="width" id="width" value="<?php echo $width; ?>" min="64" max="2048" step="64">
|
||||
</div>
|
||||
<p>
|
||||
Width: <input type="number" name="width" value="<?php echo $width; ?>" min="64" max="2048" step="64">
|
||||
Height: <input type="number" name="height" value="<?php echo $height; ?>" min="64" max="2048" step="64">
|
||||
</p>
|
||||
|
||||
<div class="form-group">
|
||||
<label for="height">Height:</label>
|
||||
<input type="number" name="height" id="height" value="<?php echo $height; ?>" min="64" max="2048" step="64">
|
||||
</div>
|
||||
<p>
|
||||
Steps: <input type="number" name="steps" value="<?php echo $steps; ?>" min="1" max="150">
|
||||
CFG Scale: <input type="number" name="cfg_scale" value="<?php echo $cfg_scale; ?>" min="1" max="30" step="0.5">
|
||||
</p>
|
||||
|
||||
<div class="form-group">
|
||||
<label for="steps">Steps:</label>
|
||||
<input type="number" name="steps" id="steps" value="<?php echo $steps; ?>" min="1" max="150">
|
||||
</div>
|
||||
|
||||
<div class="form-group">
|
||||
<label for="cfg_scale">CFG Scale:</label>
|
||||
<input type="number" name="cfg_scale" id="cfg_scale" value="<?php echo $cfg_scale; ?>" min="1" max="30" step="0.5">
|
||||
</div>
|
||||
|
||||
<div class="form-group">
|
||||
<label for="sampler">Sampler:</label>
|
||||
<select name="sampler" id="sampler">
|
||||
<p>
|
||||
Sampler:
|
||||
<select name="sampler">
|
||||
<?php foreach ($samplers as $s): ?>
|
||||
<option value="<?php echo htmlspecialchars($s); ?>" <?php echo $s === $sampler ? 'selected' : ''; ?>>
|
||||
<?php echo htmlspecialchars($s); ?>
|
||||
</option>
|
||||
<?php endforeach; ?>
|
||||
</select>
|
||||
</div>
|
||||
</p>
|
||||
|
||||
<div>
|
||||
<input type="submit" value="Generate Image">
|
||||
</div>
|
||||
<p><input type="submit" value="Generate Image"></p>
|
||||
</form>
|
||||
|
||||
<div class="save-prompt-form">
|
||||
<h3>Save Current Prompt</h3>
|
||||
<form method="post" action="<?php echo htmlspecialchars($_SERVER["PHP_SELF"]); ?>">
|
||||
<div class="form-group">
|
||||
<label for="prompt_name">Prompt Name:</label>
|
||||
<input type="text" name="prompt_name" id="prompt_name" required>
|
||||
</div>
|
||||
<input type="hidden" name="prompt" value="<?php echo htmlspecialchars($prompt); ?>">
|
||||
<input type="hidden" name="negative_prompt" value="<?php echo htmlspecialchars($negative_prompt); ?>">
|
||||
<input type="submit" name="save_prompt" value="Save Prompt">
|
||||
</form>
|
||||
</div>
|
||||
|
||||
<?php if (!empty($generated_image)): ?>
|
||||
<div class="result">
|
||||
<h2>Generated Image:</h2>
|
||||
<img width="250px" src="<?php echo htmlspecialchars($generated_image); ?>" alt="Generated image"/>
|
||||
<br/>
|
||||
<a download href="<?php echo htmlspecialchars($generated_image); ?>">Download Generated Image</a>
|
||||
<?php if (isset($size_info)): ?>
|
||||
<div class="size-info">
|
||||
<p>
|
||||
Original size: <?php echo $size_info['original']; ?> KB<br>
|
||||
Converted size: <?php echo $size_info['converted']; ?> KB<br>
|
||||
Size reduction: <?php echo $size_info['reduction']; ?>%
|
||||
</p>
|
||||
</div>
|
||||
<?php endif; ?>
|
||||
<div class="parameters">
|
||||
<strong>Parameters used:</strong><br>
|
||||
Prompt: <?php echo htmlspecialchars($prompt); ?><br>
|
||||
Negative Prompt: <?php echo htmlspecialchars($negative_prompt); ?><br>
|
||||
Processed Prompt: <?php echo htmlspecialchars($processed_prompt); ?><br>
|
||||
Size: <?php echo $width; ?>x<?php echo $height; ?><br>
|
||||
Steps: <?php echo $steps; ?><br>
|
||||
CFG Scale: <?php echo $cfg_scale; ?><br>
|
||||
Sampler: <?php echo htmlspecialchars($sampler); ?>
|
||||
</div>
|
||||
</div>
|
||||
<?php endif; ?>
|
||||
|
||||
<?php if (!empty($response)): ?>
|
||||
<div class="result">
|
||||
<h2>Error:</h2>
|
||||
<p><?php echo htmlspecialchars($response); ?></p>
|
||||
</div>
|
||||
<?php endif; ?>
|
||||
</body>
|
||||
</html>
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue