Rthur2003 commited on
Commit
7f1abd6
·
1 Parent(s): cbff5e0

fix: add analysisMode to AnalysisResult and update decisionSource handling in analyze functions

Browse files
Files changed (1) hide show
  1. app/routes/analyze.py +5 -1
app/routes/analyze.py CHANGED
@@ -52,6 +52,7 @@ class AnalysisResult(BaseModel):
52
  processingTime: float
53
  modelVersion: str
54
  decisionSource: Literal["music_ai", "ses_analizi", "preview"]
 
55
  source: Dict[str, Any]
56
  features: AnalysisFeatures
57
  audioInfo: AudioInfo
@@ -110,12 +111,14 @@ async def _analyze_youtube(
110
  preview = preview_model.analyze(yt_result.source.video_id)
111
  features_data = preview.get("features", {})
112
 
 
113
  result = AnalysisResult(
114
  isAIGenerated=yt_result.summary.is_ai_generated,
115
  confidence=yt_result.summary.confidence,
116
  processingTime=processing_time,
117
  modelVersion=yt_result.summary.model_version,
118
- decisionSource=yt_result.summary.decision_source,
 
119
  source={
120
  "kind": "youtube",
121
  "url": url,
@@ -180,6 +183,7 @@ async def _analyze_file(request_id: str, file: UploadFile, start_time: float) ->
180
  processingTime=processing_time,
181
  modelVersion=preview["model_version"],
182
  decisionSource="preview",
 
183
  source={
184
  "kind": "file",
185
  "fileName": file.filename or "unknown",
 
52
  processingTime: float
53
  modelVersion: str
54
  decisionSource: Literal["music_ai", "ses_analizi", "preview"]
55
+ analysisMode: Literal["production", "preview"]
56
  source: Dict[str, Any]
57
  features: AnalysisFeatures
58
  audioInfo: AudioInfo
 
111
  preview = preview_model.analyze(yt_result.source.video_id)
112
  features_data = preview.get("features", {})
113
 
114
+ decision = yt_result.summary.decision_source
115
  result = AnalysisResult(
116
  isAIGenerated=yt_result.summary.is_ai_generated,
117
  confidence=yt_result.summary.confidence,
118
  processingTime=processing_time,
119
  modelVersion=yt_result.summary.model_version,
120
+ decisionSource=decision,
121
+ analysisMode="preview" if decision == "preview" else "production",
122
  source={
123
  "kind": "youtube",
124
  "url": url,
 
183
  processingTime=processing_time,
184
  modelVersion=preview["model_version"],
185
  decisionSource="preview",
186
+ analysisMode="preview",
187
  source={
188
  "kind": "file",
189
  "fileName": file.filename or "unknown",