Merge branch 'develop' of gitee.com:hezhaohui123/gen2d into feat/AssertGeneration

Signed-off-by: 何朝晖 <hezhaohui0807@163.com>

backend/internal/config/config.go

@@ -14,6 +14,7 @@ type Config struct {
 	JWT      JWTConfig      `mapstructure:"jwt"`
 	LLM      LLMConfig      `mapstructure:"llm"`
 	ImageGen ImageGenConfig `mapstructure:"image_gen"`
+	Qiniu    QiniuConfig    `mapstructure:"qiniu"`
 }
 
 // ServerConfig HTTP 服务配置。
@@ -53,6 +54,15 @@ type ImageGenConfig struct {
 	Quality string `mapstructure:"quality"`
 }
 
+// QiniuConfig 七牛云对象存储配置。
+type QiniuConfig struct {
+	AccessKey string `mapstructure:"access_key"`
+	SecretKey string `mapstructure:"secret_key"`
+	Bucket    string `mapstructure:"bucket"`
+	CDNHost   string `mapstructure:"cdn_host"`
+	UseHTTPS  bool   `mapstructure:"use_https"`
+}
+
 // Load 从 YAML 配置文件和环境变量加载配置。
 // 优先级：环境变量 > YAML 文件 > 默认值。
 func Load() *Config {
@@ -107,6 +117,15 @@ func setDefaults(v *viper.Viper) {
 	v.SetDefault("image_gen.width", 1024)
 	v.SetDefault("image_gen.height", 1024)
 	v.SetDefault("image_gen.quality", "low")
+	v.SetDefault("image_gen.num_images", 1)
+	v.SetDefault("image_gen.steps", 30)
+	v.SetDefault("image_gen.cfg_scale", 7.0)
+
+	v.SetDefault("qiniu.access_key", "")
+	v.SetDefault("qiniu.secret_key", "")
+	v.SetDefault("qiniu.bucket", "")
+	v.SetDefault("qiniu.cdn_host", "")
+	v.SetDefault("qiniu.use_https", true)
 }
 
 func bindEnvVars(v *viper.Viper) {
@@ -129,4 +148,13 @@ func bindEnvVars(v *viper.Viper) {
 	v.BindEnv("image_gen.width", "GEN2D_IMAGE_WIDTH")
 	v.BindEnv("image_gen.height", "GEN2D_IMAGE_HEIGHT")
 	v.BindEnv("image_gen.quality", "GEN2D_IMAGE_QUALITY")
+	v.BindEnv("image_gen.num_images", "GEN2D_IMAGE_NUM_IMAGES")
+	v.BindEnv("image_gen.steps", "GEN2D_IMAGE_STEPS")
+	v.BindEnv("image_gen.cfg_scale", "GEN2D_IMAGE_CFG_SCALE")
+
+	v.BindEnv("qiniu.access_key", "GEN2D_QINIU_ACCESS_KEY")
+	v.BindEnv("qiniu.secret_key", "GEN2D_QINIU_SECRET_KEY")
+	v.BindEnv("qiniu.bucket", "GEN2D_QINIU_BUCKET")
+	v.BindEnv("qiniu.cdn_host", "GEN2D_QINIU_CDN_HOST")
+	v.BindEnv("qiniu.use_https", "GEN2D_QINIU_USE_HTTPS")
 }