|
|
@@ -74,6 +74,10 @@ public class MidjourneyController {
|
|
|
|
|
|
private final MidjourneyUserSettingsService midjourneyUserSettingsService;
|
|
|
|
|
|
+ private static final String PROMPT = "--v 6 ";
|
|
|
+ private static final String SETTINGS = "{\"version\":\"--v 6\",\"remix\":false,\"raw\":false,\"variation\":\"HIGH\",\"mode\":\"relax\",\"stylize\":\"MED\"}";
|
|
|
+
|
|
|
+
|
|
|
public MidjourneyUser getUser(){
|
|
|
String userToken = request.getHeader("user-token");
|
|
|
return midjourneyService.getUser(userToken);
|
|
|
@@ -483,7 +487,15 @@ public class MidjourneyController {
|
|
|
public Result<MidjourneyUserSettings> getSettings(){
|
|
|
MidjourneyUser user = getUser();
|
|
|
|
|
|
- return GatewayResponse.SUCCESS.newBuilder().toResult(midjourneyUserSettingsService.getById(user.getSettings().getId()));
|
|
|
+ MidjourneyUserSettings midjourneyUserSettings = midjourneyUserSettingsService.getById(user.getSettings().getId());
|
|
|
+ if (midjourneyUserSettings == null){
|
|
|
+ midjourneyUserSettings = new MidjourneyUserSettings();
|
|
|
+ midjourneyUserSettings.setSettings(SETTINGS);
|
|
|
+ midjourneyUserSettings.setPrompt(PROMPT);
|
|
|
+ midjourneyUserSettings.setUserId(user.getId());
|
|
|
+ midjourneyUserSettingsService.save(midjourneyUserSettings);
|
|
|
+ }
|
|
|
+ return GatewayResponse.SUCCESS.newBuilder().toResult(midjourneyUserSettings);
|
|
|
}
|
|
|
|
|
|
/**
|