import gradio as gr
import os, requests, json, subprocess
import boto3
from groq import Groq
import google.generativeai as genai

# SECRETS HF - Settings > Variables and Secrets
GROQ_KEY = os.getenv("GROQ_API_KEY")
GEMINI_KEY = os.getenv("GEMINI_API_KEY")
R2_ACCESS = os.getenv("R2_ACCESS")
R2_SECRET = os.getenv("R2_SECRET")
R2_BUCKET = os.getenv("R2_BUCKET")
R2_BUCKET_URL = os.getenv("R2_BUCKET_URL")  # https://xxx.r2.cloudflarestorage.com
R2_PUBLIC = os.getenv("R2_PUBLIC_URL")  # https://pub-xxx.r2.dev
LWS_WEBHOOK = "https://opus.dostodgroup.com/webhook.php?token=change_moi_12345_opus"

genai.configure(api_key=GEMINI_KEY)
groq_client = Groq(api_key=GROQ_KEY)

def upload_r2(local_path, remote_name):
    s3 = boto3.client('s3',
        endpoint_url=R2_BUCKET_URL,
        aws_access_key_id=R2_ACCESS,
        aws_secret_access_key=R2_SECRET
    )
    s3.upload_file(local_path, R2_BUCKET, remote_name)
    return f"{R2_PUBLIC}/{remote_name}"

def factory(job_id, video_url):
    try:
        tmp_in = f"/tmp/{job_id}.mp4"
        gr.Info(f"Download {video_url}...")
        with requests.get(video_url, stream=True, timeout=120) as r:
            r.raise_for_status()
            with open(tmp_in, 'wb') as f:
                for chunk in r.iter_content(chunk_size=1024*1024):
                    if chunk: f.write(chunk)

        gr.Info("Transcription Whisper...")
        from faster_whisper import WhisperModel
        model = WhisperModel("small", device="cpu", compute_type="int8")
        segments, _ = model.transcribe(tmp_in, language="fr")
        full_text = " ".join([s.text for s in segments])[:8000]

        if not full_text.strip():
            return "❌ Transcription vide"

        gr.Info("Analyse virale Groq...")
        prompt = f"""Tu es Opus.pro. Trouve 10 clips viraux 30-60s max score viral.
        Format JSON STRICT sans texte autour, juste le tableau: [{{"start":12.5,"end":45.2,"title":"Hook choc","score":92}}]
        Transcription: {full_text}"""

        comp = groq_client.chat.completions.create(
            model="llama-3.3-70b-versatile",
            messages=[{"role":"user","content":prompt}],
            temperature=0.7
        )
        txt = comp.choices[0].message.content.replace("```json","").replace("```","").strip()
        clips = json.loads(txt)

        final_for_lws = []
        gr.Info(f"{len(clips)} clips detectes, rendu 1080p...")
        for i, c in enumerate(clips[:10]):
            out = f"/tmp/{job_id}_clip{i+1}_1080p.mp4"
            cmd = [
                "ffmpeg","-ss",str(c['start']),"-to",str(c['end']),"-i",tmp_in,
                "-vf","crop=1080:1920:(in_w-1080)/2:0,scale=1080:1920:flags=lanczos",
                "-c:v","libx264","-crf","18","-preset","ultrafast","-c:a","aac","-b:a","128k","-y",out
            ]
            subprocess.run(cmd, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)

            if os.path.exists(out) and os.path.getsize(out) > 10000:
                remote = f"finales/{job_id}_clip{i+1}_1080p.mp4"
                r2_url = upload_r2(out, remote)
                final_for_lws.append({
                    "url": r2_url,
                    "title": c.get('title',''),
                    "score": c.get('score',0)
                })

        # Envoie à LWS
        gr.Info(f"Envoi {len(final_for_lws)} clips vers LWS...")
        requests.post(LWS_WEBHOOK, json={"job_id": job_id, "clips": final_for_lws}, timeout=60)

        return f"✅ {len(final_for_lws)} clips 1080p envoyés sur opus.dostodgroup.com/videos/finales/\n\n" + "\n".join([f"{x['score']} - {x['title']} -> {x['url']}" for x in final_for_lws])

    except Exception as e:
        import traceback
        return f"❌ Erreur: {str(e)}\n{traceback.format_exc()}"

with gr.Blocks(title="OPUS 1080p Factory") as demo:
    gr.Markdown("# 🏭 OPUS 1080p - Usine Gratuite ZeroGPU\nR2 -> 10 clips viraux 1080p -> LWS opus.dostodgroup.com")
    job_id = gr.Textbox(label="job_id (ex: job_123)", value="job_test_1")
    video_url = gr.Textbox(label="URL video longue R2 (https://pub-.../video.mp4)", placeholder="https://pub-xxx.r2.dev/uploads/...")
    btn = gr.Button("🚀 RENDRE 10 CLIPS 1080p", variant="primary")
    output = gr.Textbox(label="Logs", lines=15)
    btn.click(factory, [job_id, video_url], output)

demo.launch()