Update main.py

main.py

@@ -387,6 +387,72 @@ Se o contexto enviado pelo usuário não for verdadeiro ou estiver impreciso, ig
         if cut_file and os.path.exists(cut_file.name): os.unlink(cut_file.name)
 
 
+class VideoFilterRequest(BaseModel):
+    video_url: str
+
+@app.post("/video-filter")
+async def video_filter_endpoint(request: VideoFilterRequest):
+    if not client:
+        raise HTTPException(status_code=500, detail="Gemini client is not initialized")
+    temp_file = None
+    try:
+        if not request.video_url:
+            raise HTTPException(status_code=400, detail="URL do vídeo é obrigatória")
+
+        print(f"📥 Baixando vídeo para filtro: {request.video_url}")
+        response = download_file_with_retry(request.video_url, timeout=600)
+        ext = '.webm' if 'webm' in response.headers.get('content-type', '').lower() else '.mp4'
+        
+        temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=ext)
+        for chunk in response.iter_content(chunk_size=1024*1024):
+            if chunk: temp_file.write(chunk)
+        temp_file.close()
+
+        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.
+
+Base do JSON: {
+  "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.
+
+viralization_probability
+
+Probabilidade estimada (0–100) de o vídeo viralizar. A IA deve estimar isso analisando fatores modernos de performance no Instagram, como força do “hook” nos primeiros segundos, potencial de retenção, probabilidade de compartilhamento, identificação com o público da geração Z (14–30 anos), impacto emocional, alinhamento com tendências e potencial de replay.
+
+sensitive_content
+
+Valor booleano que indica se o vídeo contém conteúdo perturbador, trágico ou pesado. Isso inclui desastres, acidentes, violência ou tragédias. Como a página tem o lema de que o mundo não precisa ser focado em desastre ou tragédia, esse tipo de conteúdo deve ser marcado como true.
+
+political_content
+
+Valor booleano que indica se o vídeo menciona, mostra ou faz referência a qualquer político, evento político, discussão política ou símbolo político, independentemente de ser positivo, negativo ou neutro. Caso haja qualquer referência política, o valor deve ser true.
+
+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."""
+
+        model_obj = get_gemini_model("flash")
+        print(f"🧠 Enviando para Gemini (flash) para filtro de vídeo...")
+        
+        response_gemini = await client.generate_content(prompt, files=[video_path_to_analyze], model=model_obj)
+        
+        filter_data = extract_json_from_text(response_gemini.text)
+        if filter_data is None:
+            return JSONResponse(content={"raw_content": response_gemini.text, "error": "Failed to parse JSON"}, status_code=200)
+
+        return filter_data
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"Erro interno: {str(e)}")
+    finally:
+        if temp_file and os.path.exists(temp_file.name): os.unlink(temp_file.name)
+
+
 # ==========================================
 # GROQ ENDPOINTS / SUBTITLES
 # ==========================================