refactor: army

controllers/ai.go

@@ -35,6 +35,7 @@ const (
 	audienceTypeYoungAdult         = "young_adult"
 	audienceTypeMiddleAged         = "middle_aged"
 	audienceTypeElderly            = "elderly"
+	audienceTypeArmy               = "army"
 	sourceUpload                   = "upload"
 	sourceCloud                    = "cloud"
 	sourceWechat                   = "wechat"
@@ -56,6 +57,8 @@ func resolveAudienceLabel(audienceType string) (string, bool) {
 		return "中年人", true
 	case audienceTypeElderly:
 		return "老年人", true
+	case audienceTypeArmy:
+		return "18-25岁军人", true
 	default:
 		return "", false
 	}
@@ -346,7 +349,7 @@ func callAIForAnalysis(prompt string) (*aiAnalysisResult, error) {
 // @Param step_data formData file false "步数数据CSV文件(analysis_type为heart_rate_with_steps时必填)"
 // @Param teaching_plan formData file false "教案DOCX文件(teaching_plan_source为upload/wechat时必填)"
 // @Param analysis_type formData string false "分析类型: heart_rate_only(默认) | heart_rate_with_steps"
-// @Param audience_type formData string false "适用人群: kindergarten(默认) | primary_school | middle_school | university | young_adult | middle_aged | elderly"
+// @Param audience_type formData string false "适用人群: kindergarten(默认) | primary_school | middle_school | university | young_adult | middle_aged | elderly | army"
 // @Param audience_detail formData string false "精确班级或年级，例如: 小班、中班、大班、小学一年级"
 // @Param teaching_plan_source formData string false "教案来源: upload(默认) | cloud | wechat"
 // @Param regionid formData string false "区域ID"
@@ -382,7 +385,7 @@ func (tc *TrainingController) AnalyzeByAI(c *gin.Context) {
 	}
 	audienceLabel, ok := resolveAudienceDescription(audienceType, audienceDetail)
 	if !ok {
-		c.JSON(http.StatusBadRequest, gin.H{"error": "invalid audience_type, expected kindergarten, primary_school, middle_school, university, young_adult, middle_aged or elderly"})
+		c.JSON(http.StatusBadRequest, gin.H{"error": "invalid audience_type, expected kindergarten, primary_school, middle_school, university, young_adult, middle_aged, elderly or army"})
 		return
 	}