Update main.py

main.py

@@ -389,6 +389,7 @@ Se o contexto enviado pelo usuário não for verdadeiro ou estiver impreciso, ig
 
 class VideoFilterRequest(BaseModel):
     video_url: str
+    context: Optional[str] = None
 
 @app.post("/video-filter")
 async def video_filter_endpoint(request: VideoFilterRequest):
@@ -410,16 +411,23 @@ async def video_filter_endpoint(request: VideoFilterRequest):
 
         video_path_to_analyze = temp_file.name
 
-        prompt = """Analise o vídeo anexado e retorne um JSON, apenas o JSON, sem texto adicional, de um filtro sobre o vídeo.
+        contexto_add = f"\n\nContexto Adicional / Legenda Original:\n{request.context}" if request.context else ""
+
+        prompt = f"""Analise o vídeo anexado e retorne um JSON, apenas o JSON, sem texto adicional, de um filtro sobre o vídeo.
 
-Base do JSON: {
+Base do JSON: {{
+  "description": "Explicação detalhada sobre o que acontece no vídeo, contexto visual e assunto discutido.",
   "viralization_probability": 88,
   "sensitive_content": false,
   "political_content": false,
   "should_post": true
-}
+}}
+
+Estou criando uma página do Instagram e quero saber o que postar ou não postar.{contexto_add}
+
+description
 
-Estou criando uma página do Instagram e quero saber o que postar ou não postar.
+Descrição e explicação detalhada sobre o que o vídeo se trata. Descreva o visual, as ações, pessoas envolvidas e, de forma resumida, o assunto principal.
 
 viralization_probability
 
@@ -437,6 +445,9 @@ should_post
 
 Decisão final da IA indicando se o vídeo deve ou não ser postado. Essa decisão deve considerar todos os fatores anteriores, como potencial de viralização, presença de conteúdo sensível, política, marca d’água e outros sinais relevantes."""
 
+        # get_gemini_model("flash") chamará "Model.G_3_0_FLASH", que é o modelo Flash rápido. 
+        # A demora de alguns segundos é comum porque o vídeo precisa ser enviado e processado 
+        # nos servidores do Gemini, o que leva alguns segundos pelo próprio tamanho do vídeo.
         model_obj = get_gemini_model("flash")
         print(f"🧠 Enviando para Gemini (flash) para filtro de vídeo...")