50 lines
1.2 KiB
C++
50 lines
1.2 KiB
C++
#pragma once
|
|
|
|
#include <QDialog>
|
|
|
|
class QLabel;
|
|
class QStackedWidget;
|
|
|
|
class BlackholeResolveDialog final : public QDialog {
|
|
Q_OBJECT
|
|
public:
|
|
enum class Algorithm {
|
|
CopyBackgroundRegion,
|
|
UseOriginalBackground,
|
|
ModelInpaint,
|
|
};
|
|
|
|
explicit BlackholeResolveDialog(const QString& blackholeName, QWidget* parent = nullptr);
|
|
|
|
Algorithm selectedAlgorithm() const { return m_selectedAlgorithm; }
|
|
QString promptText() const;
|
|
|
|
/// 模型补全（LaMa 等）内置正向提示：中国画去人物、补背景纹理，无人物残留。
|
|
[[nodiscard]] static QString defaultModelInpaintPrompt();
|
|
|
|
private:
|
|
void buildSelectPage();
|
|
void buildDetailPage();
|
|
void enterAlgorithmPage(Algorithm algo);
|
|
|
|
private:
|
|
QString m_blackholeName;
|
|
Algorithm m_selectedAlgorithm = Algorithm::UseOriginalBackground;
|
|
|
|
QStackedWidget* m_pages = nullptr;
|
|
|
|
QWidget* m_pageSelect = nullptr;
|
|
QWidget* m_pageDetail = nullptr;
|
|
|
|
QLabel* m_detailTitle = nullptr;
|
|
QStackedWidget* m_algoDetails = nullptr;
|
|
|
|
QWidget* m_copyDetail = nullptr;
|
|
|
|
QWidget* m_originalDetail = nullptr;
|
|
|
|
QWidget* m_modelDetail = nullptr;
|
|
class QPlainTextEdit* m_promptEdit = nullptr;
|
|
};
|
|
|