feat: improve generation stability and prompt optimization

Author: ShellMonster

backend/go.mod

@@ -7,18 +7,15 @@ require (
 	github.com/disintegration/imaging v1.6.2
 	github.com/gin-gonic/gin v1.11.0
 	github.com/google/uuid v1.6.0
+	github.com/mattn/go-sqlite3 v1.14.22
 	github.com/mazrean/formstream v1.1.3
-	github.com/openai/openai-go/v3 v3.15.0
 	github.com/spf13/viper v1.21.0
-	google.golang.org/genai v1.40.0
+	golang.org/x/sync v0.17.0
 	gorm.io/driver/sqlite v1.6.0
 	gorm.io/gorm v1.31.1
 )
 
 require (
-	cloud.google.com/go v0.116.0 // indirect
-	cloud.google.com/go/auth v0.9.3 // indirect
-	cloud.google.com/go/compute/metadata v0.5.0 // indirect
 	github.com/bytedance/sonic v1.14.0 // indirect
 	github.com/bytedance/sonic/loader v0.3.0 // indirect
 	github.com/cloudwego/base64x v0.1.6 // indirect
@@ -31,18 +28,12 @@ require (
 	github.com/go-viper/mapstructure/v2 v2.4.0 // indirect
 	github.com/goccy/go-json v0.10.2 // indirect
 	github.com/goccy/go-yaml v1.18.0 // indirect
-	github.com/golang/groupcache v0.0.0-20210331224755-41bb18bfe9da // indirect
-	github.com/google/go-cmp v0.7.0 // indirect
-	github.com/google/s2a-go v0.1.8 // indirect
-	github.com/googleapis/enterprise-certificate-proxy v0.3.4 // indirect
-	github.com/gorilla/websocket v1.5.3 // indirect
 	github.com/jinzhu/inflection v1.0.0 // indirect
 	github.com/jinzhu/now v1.1.5 // indirect
 	github.com/json-iterator/go v1.1.12 // indirect
 	github.com/klauspost/cpuid/v2 v2.3.0 // indirect
 	github.com/leodido/go-urn v1.4.0 // indirect
 	github.com/mattn/go-isatty v0.0.20 // indirect
-	github.com/mattn/go-sqlite3 v1.14.22 // indirect
 	github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd // indirect
 	github.com/modern-go/reflect2 v1.0.2 // indirect
 	github.com/pelletier/go-toml/v2 v2.2.4 // indirect
@@ -54,26 +45,18 @@ require (
 	github.com/spf13/cast v1.10.0 // indirect
 	github.com/spf13/pflag v1.0.10 // indirect
 	github.com/subosito/gotenv v1.6.0 // indirect
-	github.com/tidwall/gjson v1.18.0 // indirect
-	github.com/tidwall/match v1.1.1 // indirect
-	github.com/tidwall/pretty v1.2.1 // indirect
-	github.com/tidwall/sjson v1.2.5 // indirect
 	github.com/twitchyliquid64/golang-asm v0.15.1 // indirect
 	github.com/ugorji/go/codec v1.3.0 // indirect
-	go.opencensus.io v0.24.0 // indirect
 	go.uber.org/mock v0.6.0 // indirect
 	go.yaml.in/yaml/v3 v3.0.4 // indirect
 	golang.org/x/arch v0.20.0 // indirect
 	golang.org/x/crypto v0.41.0 // indirect
 	golang.org/x/image v0.0.0-20191009234506-e7c1f5e7dbb8 // indirect
 	golang.org/x/mod v0.27.0 // indirect
 	golang.org/x/net v0.43.0 // indirect
-	golang.org/x/sync v0.17.0 // indirect
 	golang.org/x/sys v0.35.0 // indirect
 	golang.org/x/text v0.28.0 // indirect
 	golang.org/x/time v0.6.0 // indirect
 	golang.org/x/tools v0.36.0 // indirect
-	google.golang.org/genproto/googleapis/rpc v0.0.0-20240903143218-8af14fe29dc1 // indirect
-	google.golang.org/grpc v1.66.2 // indirect
 	google.golang.org/protobuf v1.36.9 // indirect
 )

backend/go.sum

@@ -11,12 +11,16 @@ import (
 	"strings"
 	"time"
 
+	"image-gen-service/internal/config"
 	"image-gen-service/internal/model"
 
 	"github.com/gin-gonic/gin"
 )
 
-const maxExportRemoteSize = 50 * 1024 * 1024
+const (
+	defaultExportRemoteFetchTimeoutSeconds = 120
+	defaultExportRemoteMaxFileMB           = 512
+)
 
 type exportImagesRequest struct {
 	ImageIDs    []string `json:"imageIds"`
@@ -176,7 +180,16 @@ func ExportImagesHandler(c *gin.Context) {
 }
 
 func writeRemoteFile(writer io.Writer, source string) error {
-	resp, err := http.Get(source)
+	timeoutSeconds := config.GlobalConfig.Exports.RemoteFetchTimeoutSeconds
+	if timeoutSeconds <= 0 {
+		timeoutSeconds = defaultExportRemoteFetchTimeoutSeconds
+	}
+
+	client := &http.Client{
+		Timeout: time.Duration(timeoutSeconds) * time.Second,
+	}
+
+	resp, err := client.Get(source)
 	if err != nil {
 		return err
 	}
@@ -186,13 +199,19 @@ func writeRemoteFile(writer io.Writer, source string) error {
 		return fmt.Errorf("http status %d", resp.StatusCode)
 	}
 
-	reader := io.LimitReader(resp.Body, maxExportRemoteSize+1)
+	maxFileMB := config.GlobalConfig.Exports.RemoteMaxFileMB
+	if maxFileMB <= 0 {
+		maxFileMB = defaultExportRemoteMaxFileMB
+	}
+
+	maxBytes := int64(maxFileMB) * 1024 * 1024
+	reader := io.LimitReader(resp.Body, maxBytes+1)
 	written, err := io.Copy(writer, reader)
 	if err != nil {
 		return err
 	}
-	if written > maxExportRemoteSize {
-		return fmt.Errorf("remote file exceeds %d bytes", maxExportRemoteSize)
+	if written > maxBytes {
+		return fmt.Errorf("remote file exceeds %d bytes", maxBytes)
 	}
 	return nil
 }