Spaces:
Running
Running
fix: add analysisMode to AnalysisResult and update decisionSource handling in analyze functions
Browse files- 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=
|
|
|
|
| 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",
|