stable-diffusion-settings.hpp

Go to the documentation of this file.

#ifndef STABLE_DIFFUSION_SETTINGS_HPP
#define STABLE_DIFFUSION_SETTINGS_HPP
 
#include <QDialog>
#include <QList>
#include <QString>
#include <QStringList>
 
class QCheckBox;
class QComboBox;
class QDoubleSpinBox;
class QLineEdit;
class QListWidget;
class QPushButton;
class QSpinBox;
 
struct StableDiffusionConfiguration {
    bool enabled = false;
    QString model_file;
    QString upscale_model_file;
    QStringList lora_files;
    QList<double> lora_multipliers;
    QString prompt;
    QString negative_prompt;
    QString server_executable = "sd-server";
    QString server_arguments;
    int server_port = 1234;
    int width = 576;
    int height = 320;
    int upscale_working_width = 0;
    int upscale_working_height = 0;
    int steps = 12;
    double strength = 0.35;
    double cfg_scale = 5.0;
    int seed = 1234;
    QString sampler = "euler_a";
    QString scheduler = "discrete";
    bool upscale = false;
    bool upscale_only = false;
};
 
class StableDiffusionSettingsDialog : public QDialog {
    Q_OBJECT
 
  public:
    explicit StableDiffusionSettingsDialog(QWidget *parent = nullptr);
    [[nodiscard]] StableDiffusionConfiguration configuration() const;
 
  signals:
    void settingsApplied();
 
  private slots:
    void browse_model();
    void add_lora_models();
    void remove_lora_models();
    void select_lora_model();
    void update_lora_multiplier(double multiplier);
    void browse_upscale_model();
    void browse_server();
    void apply_settings();
    void accept_settings();
 
  private:
    [[nodiscard]] bool validate_settings();
    void load_ui_state();
    void save_ui_state();
    void update_enabled_state();
    void add_lora_item(const QString &filename, double multiplier);
    void update_lora_item_text(int row);
 
    QCheckBox *enable_check_box = nullptr;
    QLineEdit *model_file_edit = nullptr;
    QPushButton *browse_model_button = nullptr;
    QLineEdit *prompt_edit = nullptr;
    QLineEdit *negative_prompt_edit = nullptr;
    QListWidget *lora_list_widget = nullptr;
    QPushButton *add_lora_button = nullptr;
    QPushButton *remove_lora_button = nullptr;
    QDoubleSpinBox *lora_multiplier_spin_box = nullptr;
    QLineEdit *server_edit = nullptr;
    QLineEdit *server_arguments_edit = nullptr;
    QPushButton *browse_server_button = nullptr;
    QSpinBox *server_port_spin_box = nullptr;
    QComboBox *resolution_combo_box = nullptr;
    QSpinBox *steps_spin_box = nullptr;
    QDoubleSpinBox *strength_spin_box = nullptr;
    QDoubleSpinBox *cfg_scale_spin_box = nullptr;
    QSpinBox *seed_spin_box = nullptr;
    QComboBox *sampler_combo_box = nullptr;
    QComboBox *scheduler_combo_box = nullptr;
    QCheckBox *upscale_check_box = nullptr;
    QCheckBox *upscale_only_check_box = nullptr;
    QCheckBox *server_upscale_check_box = nullptr;
    QLineEdit *upscale_model_edit = nullptr;
    QPushButton *browse_upscale_model_button = nullptr;
    QLineEdit *upscale_working_resolution_edit = nullptr;
};
 
#endif