Upload app.py

app.py

@@ -0,0 +1,186 @@
+# -*- coding: utf-8 -*-
+import os
+import random
+from pathlib import Path
+
+import gradio as gr
+import spaces
+import torch
+from diffusers import CogVideoXPipeline
+from diffusers.utils import export_to_video
+
+MODEL_ID = "zai-org/CogVideoX-2b"
+OUTPUT_DIR = Path("outputs")
+OUTPUT_DIR.mkdir(exist_ok=True)
+CHANNEL_URL = "https://t.me/Profik_Studio"
+
+T = {
+    "ru": {
+        "title": "AI Video Studio", "sub": "Генератор видео из текста • CogVideoX-2B",
+        "lang": "Язык", "gen": "Генератор", "hist": "История", "settings": "Настройки",
+        "subscribe_title": "Доступ к генератору", "subscribe_text": "Подпишись на канал **@Profik_Studio**, затем нажми кнопку подтверждения.",
+        "open": "📢 Открыть канал", "check": "✅ Я подписался", "locked": "🔒 Сначала подпишись на канал и нажми «Я подписался». ",
+        "unlocked": "🟢 Доступ открыт — можно создавать видео.",
+        "prompt": "Описание видео", "prompt_ph": "Например: футуристический город ночью, неон, дождь, кинематографичный пролёт камеры...",
+        "negative": "Что исключить", "negative_ph": "Размытие, артефакты, плохое качество...",
+        "frames": "Кадры", "steps": "Шаги", "fps": "FPS", "guidance": "Guidance", "seed": "Seed (-1 = случайный)",
+        "create": "🚀 СОЗДАТЬ ВИДЕО", "result": "Результат", "ready": "🟢 Готово к генерации.",
+        "examples": "Примеры", "history_empty": "История пока пуста.", "refresh": "🔄 Обновить",
+        "settings_text": "**Модель:** CogVideoX-2B  \n**Режим:** Hugging Face ZeroGPU  \n**API-ключ:** не требуется",
+        "error_prompt": "Введите описание видео.", "error_access": "Сначала подпишитесь на @Profik_Studio.",
+        "error_model": "Модель ещё загружается. Попробуйте снова через несколько секунд.",
+        "footer": "AI Video Studio • Profik • CogVideoX-2B"
+    },
+    "en": {
+        "title": "AI Video Studio", "sub": "Text-to-video generator • CogVideoX-2B",
+        "lang": "Language", "gen": "Generator", "hist": "History", "settings": "Settings",
+        "subscribe_title": "Generator access", "subscribe_text": "Subscribe to **@Profik_Studio**, then press the confirmation button.",
+        "open": "📢 Open channel", "check": "✅ I subscribed", "locked": "🔒 Subscribe to the channel and press “I subscribed”.",
+        "unlocked": "🟢 Access unlocked — you can create videos.",
+        "prompt": "Video description", "prompt_ph": "Example: futuristic city at night, neon, rain, cinematic camera fly-through...",
+        "negative": "Negative prompt", "negative_ph": "Blur, artifacts, low quality...",
+        "frames": "Frames", "steps": "Steps", "fps": "FPS", "guidance": "Guidance", "seed": "Seed (-1 = random)",
+        "create": "🚀 CREATE VIDEO", "result": "Result", "ready": "🟢 Ready to generate.",
+        "examples": "Examples", "history_empty": "History is empty.", "refresh": "🔄 Refresh",
+        "settings_text": "**Model:** CogVideoX-2B  \n**Runtime:** Hugging Face ZeroGPU  \n**API key:** not required",
+        "error_prompt": "Enter a video description.", "error_access": "Subscribe to @Profik_Studio first.",
+        "error_model": "The model is still loading. Try again in a few seconds.",
+        "footer": "AI Video Studio • Profik • CogVideoX-2B"
+    }
+}
+
+CSS = """
+body { background:#06080b !important; }
+.gradio-container { max-width:1180px !important; background:transparent !important; }
+.hero { text-align:center; padding:34px 12px 22px; }
+.logo { width:72px;height:72px;margin:auto;border-radius:22px;display:flex;align-items:center;justify-content:center;font-size:34px;background:linear-gradient(145deg,#10251a,#0b1210);border:1px solid #245f3c;box-shadow:0 0 45px #1ce66b18; }
+.hero h1 { font-size:38px !important; margin:12px 0 4px !important; letter-spacing:-1px; }
+.hero p { color:#84909d !important; }
+.panel { border:1px solid #1d2833 !important; border-radius:20px !important; background:linear-gradient(145deg,#0d1319,#090d12) !important; padding:18px !important; }
+.badge { display:inline-block;padding:6px 10px;border:1px solid #254735;border-radius:999px;color:#63e99a;background:#0b1a12;font-size:12px; }
+.gr-button-primary { background:linear-gradient(90deg,#20c96b,#42f58a) !important;color:#031008 !important;border:0 !important;font-weight:900 !important; }
+.bigbtn button { min-height:56px !important;font-size:16px !important; }
+.nav button { border-radius:13px !important; }
+.footer { text-align:center;color:#667381;padding:25px 0 8px;font-size:12px; }
+"""
+
+pipe = None
+try:
+    pipe = CogVideoXPipeline.from_pretrained(MODEL_ID, torch_dtype=torch.float16)
+    pipe.enable_model_cpu_offload()
+    try: pipe.vae.enable_tiling()
+    except Exception: pass
+except Exception as e:
+    print("MODEL LOAD ERROR:", repr(e))
+
+
+def history(lang):
+    t=T[lang]; files=sorted(OUTPUT_DIR.glob("*.mp4"), key=lambda p:p.stat().st_mtime, reverse=True)
+    if not files: return f"### {t['history_empty']}"
+    return "### " + t["hist"] + "\n\n" + "\n".join(f"- `{p.name}` — {p.stat().st_size/1048576:.1f} MB" for p in files[:30])
+
+
+def unlock(lang):
+    return True, T[lang]["unlocked"]
+
+
+@spaces.GPU
+def generate(prompt, negative, frames, steps, fps, guidance, seed, lang, access, progress=gr.Progress()):
+    t=T[lang]
+    if not access: raise gr.Error(t["error_access"])
+    if not prompt or not prompt.strip(): raise gr.Error(t["error_prompt"])
+    if pipe is None: raise gr.Error(t["error_model"])
+    seed=int(seed)
+    if seed < 0: seed=random.randint(0,2**31-1)
+    try:
+        progress(0.08, desc="Generating...")
+        g=torch.Generator(device="cuda").manual_seed(seed)
+        out=pipe(prompt=prompt.strip(), negative_prompt=(negative or "").strip() or None,
+                 num_frames=int(frames), num_inference_steps=int(steps),
+                 guidance_scale=float(guidance), generator=g)
+        progress(0.9, desc="Encoding...")
+        file=OUTPUT_DIR/f"video_{seed}_{random.randint(1000,9999)}.mp4"
+        export_to_video(out.frames[0], str(file), fps=int(fps))
+        progress(1.0)
+        return str(file), f"### 🟢 {t['result']}\nSeed: `{seed}`"
+    except Exception as e:
+        raise gr.Error(str(e))
+
+
+def language_ui(lang):
+    t=T[lang]
+    hero=f'<div class="hero"><div class="logo">🎬</div><h1>{t["title"]}</h1><p>{t["sub"]}</p><span class="badge">● ZERO API KEY</span></div>'
+    return (hero,t["lang"],t["gen"],t["hist"],t["settings"],t["subscribe_title"],t["subscribe_text"],t["open"],t["check"],t["locked"],t["prompt"],t["prompt_ph"],t["negative"],t["negative_ph"],t["frames"],t["steps"],t["fps"],t["guidance"],t["seed"],t["create"],t["result"],t["ready"],t["examples"],t["settings_text"],t["refresh"],t["footer"],history(lang))
+
+with gr.Blocks(css=CSS, title="AI Video Studio") as demo:
+    access=gr.State(False)
+    hero=gr.HTML()
+    language=gr.Radio(["ru","en"], value="ru", label="Язык", show_label=True)
+
+    with gr.Row(elem_classes="nav"):
+        nav_gen=gr.Button("🎬 Генератор", variant="primary")
+        nav_hist=gr.Button("📚 История")
+        nav_set=gr.Button("⚙️ Настройки")
+
+    with gr.Group(elem_classes="panel"):
+        sub_title=gr.Markdown()
+        sub_text=gr.Markdown()
+        with gr.Row():
+            open_btn=gr.Button("📢 Открыть канал")
+            check_btn=gr.Button("✅ Я подписался", variant="primary")
+        sub_status=gr.Markdown()
+
+    with gr.Tab("🎬 Генератор"):
+        with gr.Row():
+            with gr.Column(scale=7, elem_classes="panel"):
+                prompt=gr.Textbox(lines=5)
+                negative=gr.Textbox(lines=3)
+                with gr.Row():
+                    frames=gr.Slider(8,49,49,step=1)
+                    fps=gr.Slider(4,16,8,step=1)
+                with gr.Row():
+                    steps=gr.Slider(10,50,30,step=1)
+                    guidance=gr.Slider(1,12,6,step=.5)
+                seed=gr.Number(value=-1, precision=0)
+                create=gr.Button("🚀 СОЗДАТЬ ВИДЕО", variant="primary", elem_classes="bigbtn")
+            with gr.Column(scale=5, elem_classes="panel"):
+                result_title=gr.Markdown()
+                result=gr.Markdown()
+                video=gr.Video(label="Video", format="mp4")
+        examples_title=gr.Markdown()
+        gr.Examples([
+            ["Кинематографичный ночной город будущего, неон, дождь, плавный пролёт камеры"],
+            ["Заснеженные горы на рассвете, золотой свет, облака движутся над вершинами"],
+            ["Маленький робот идёт по космической станции, кинематографичный свет"],
+        ], inputs=prompt)
+
+    with gr.Tab("📚 История"):
+        history_box=gr.Markdown()
+        refresh=gr.Button("🔄 Обновить")
+
+    with gr.Tab("⚙️ Настройки"):
+        settings_box=gr.Markdown()
+
+    footer=gr.Markdown(elem_classes="footer")
+
+    vals=language_ui("ru")
+    for comp,val in zip([hero,language,nav_gen,nav_hist,nav_set,sub_title,sub_text,open_btn,check_btn,sub_status,prompt,negative,frames,steps,fps,guidance,seed,create,result_title,result,examples_title,settings_box,refresh,footer,history_box], vals):
+        if comp is language: continue
+    hero.value=vals[0]; language.label=vals[1]; nav_gen.value=vals[2]; nav_hist.value=vals[3]; nav_set.value=vals[4]
+    sub_title.value=vals[5]; sub_text.value=vals[6]; open_btn.value=vals[7]; check_btn.value=vals[8]; sub_status.value=vals[9]
+    prompt.label=vals[10]; prompt.placeholder=vals[11]; negative.label=vals[12]; negative.placeholder=vals[13]
+    frames.label=vals[14]; steps.label=vals[15]; fps.label=vals[16]; guidance.label=vals[17]; seed.label=vals[18]
+    create.value=vals[19]; result_title.value=vals[20]; result.value=vals[21]; examples_title.value=vals[22]; settings_box.value=vals[23]; refresh.value=vals[24]; footer.value=vals[25]; history_box.value=vals[26]
+
+    open_btn.click(None, None, None, js=f"() => window.open('{CHANNEL_URL}','_blank')")
+    check_btn.click(unlock, language, [access,sub_status])
+    create.click(generate, [prompt,negative,frames,steps,fps,guidance,seed,language,access], [video,result])
+    refresh.click(history, language, history_box)
+
+    language.change(
+        language_ui, language,
+        [hero,language,nav_gen,nav_hist,nav_set,sub_title,sub_text,open_btn,check_btn,sub_status,prompt,negative,frames,steps,fps,guidance,seed,create,result_title,result,examples_title,settings_box,refresh,footer,history_box],
+    )
+
+if __name__ == "__main__":
+    demo.launch()