Update app.py

app.py

@@ -5,31 +5,25 @@ import threading
 import requests
 import subprocess
 import base64
-from flask import Flask, render_template_string, request, jsonify, send_from_directory, abort
+from flask import Flask, render_template, request, jsonify, send_from_directory, abort # 'render_template' जोड़ा गया और 'render_template_string' हटाया गया
 import queue
 from groq import Groq
-import itertools # /// ADDED IMPORT /// - कुंजियों को बारी-बारी से उपयोग करने के लिए
+import itertools
 
-# ---------- CONFIG (MODIFIED FOR YOUR KEY NAMES) ----------
-# Hugging Face Spaces में Settings -> Secrets में इन्हें सेट करें
-
-# /// MODIFIED SECTION ///
-# अब यह btz1, btz2, आदि कुंजियों को ढूंढेगा।
+# ---------- CONFIG ----------
 API_KEYS = [os.environ.get(f"btz{i}") for i in range(1, 100) if os.environ.get(f"btz{i}")]
 if not API_KEYS: raise KeyError("No 'btzN' environment variables found (e.g., btz1, btz2). Please set them in Hugging Face secrets.")
 
-GROQ_API_KEY = os.environ.get("GROQ_KEY") # यह माना जा रहा है कि आपके पास GROQ_KEY नाम की एक ही कुंजी है
+GROQ_API_KEY = os.environ.get("GROQ_KEY")
 if not GROQ_API_KEY: raise KeyError("GROQ_KEY environment variable not found.")
 
-# /// MODIFIED SECTION ///
-# अब यह gmni1, gmni2, आदि कुंजियों को ढूंढेगा और बारी-बारी से उपयोग करेगा।
 GEMINI_API_KEYS = [os.environ.get(f"gmni{i}") for i in range(1, 100) if os.environ.get(f"gmni{i}")]
 if not GEMINI_API_KEYS: raise KeyError("No 'gmniN' environment variables found (e.g., gmni1, gmni2). Please set them in Hugging Face secrets.")
-gemini_key_cycler = itertools.cycle(GEMINI_API_KEYS) # बारी-बारी से कुंजी चुनने के लिए
+gemini_key_cycler = itertools.cycle(GEMINI_API_KEYS)
 
 DEFAULT_BYTEZ_MODEL = "ali-vilab/text-to-video-ms-1.7b"
 
-# ---------- MODEL HUNTER FUNCTIONS (No changes here) ----------
+# ---------- MODEL HUNTER FUNCTIONS ----------
 def find_best_groq_model(api_key):
     try:
         print("🤖 Hunting for the best Groq model...")
@@ -59,9 +53,8 @@ def find_best_gemini_vision_model(api_key):
         raise ValueError("No usable Gemini Vision models found.")
     except Exception as e: print(f"🛑 Gemini hunt failed: {e}. Using hardcoded fallback."); return "gemini-1.5-flash-latest"
 
-# ---------- INITIALIZATION (Modified) ----------
+# ---------- INITIALIZATION ----------
 GROQ_MODEL = find_best_groq_model(GROQ_API_KEY)
-# /// MODIFIED SECTION /// - We now check with the first Gemini key
 GEMINI_VISION_MODEL = find_best_gemini_vision_model(GEMINI_API_KEYS[0])
 print(f"✅ Loaded {len(API_KEYS)} Bytez keys and {len(GEMINI_API_KEYS)} Gemini keys. Using Groq: {GROQ_MODEL} and Gemini: {GEMINI_VISION_MODEL}")
 
@@ -69,14 +62,14 @@ OUTPUT_FOLDER = "output"
 os.makedirs(OUTPUT_FOLDER, exist_ok=True)
 os.makedirs("static", exist_ok=True)
 
-# ---------- APP & STATE (No changes here) ----------
+# ---------- APP & STATE ----------
 app = Flask(__name__)
 progress = { "active": False, "step": 0, "total": 0, "status": "idle", "message": "", "error": None, "video_relpath": None, "live_log": [] }
 job_queue = queue.Queue()
 generated_clips_dict = {}
 clips_lock = threading.Lock()
 
-# ---------- HELPER FUNCTIONS (No changes in these functions themselves) ----------
+# ---------- HELPER FUNCTIONS ----------
 def set_progress(log_message=None, **kwargs):
     global progress
     with threading.Lock():
@@ -120,27 +113,46 @@ def generate_visual_blueprint_with_groq(user_prompt, api_key, model_name):
         return visual_blueprint, None
     except Exception as e: return None, f"Groq API Error: {e}"
 
+def distill_prompt_for_short_video(enhanced_prompt, api_key, model_name):
+    print("🧪 Contacting Groq (Prompt Distiller)...")
+    try:
+        client = Groq(api_key=api_key)
+        system_prompt = """You are an expert video editor who specializes in ultra-short clips (3-4 seconds). 
+        Your task is to take a long, hyper-detailed prompt and distill it into a concise, powerful version perfect for a brief video.
+        RULES:
+        1. Identify the single most important subject and a very brief, clear action.
+        2. Focus the entire prompt on ONE dynamic moment or a powerful, static scene. Do not describe multiple complex actions.
+        3. Condense descriptive keywords. Instead of 'photorealistic, 4k, ultra realistic, hyper-detailed', choose the best two, like 'cinematic, hyper-detailed'.
+        4. The final prompt must be significantly shorter but retain the core artistic vision.
+        Output ONLY the final, short, distilled prompt.
+        """
+        response = client.chat.completions.create(
+            model=model_name,
+            messages=[
+                {"role": "system", "content": system_prompt},
+                {"role": "user", "content": enhanced_prompt}
+            ]
+        )
+        distilled_prompt = response.choices[0].message.content.strip()
+        print(f"✨ Distiller's final prompt: {distilled_prompt}")
+        return distilled_prompt, None
+    except Exception as e:
+        print(f"⚠️ Groq Distiller failed: {e}. Falling back to original enhanced prompt.")
+        return enhanced_prompt, None
+
 def generate_clip(prompt, idx, api_key, bytez_model):
     from bytez import Bytez
     sdk = Bytez(api_key)
     model = sdk.model(bytez_model)
-    
-    # --- MODIFIED SECTION TO FIX THE CRASH ---
     try:
-        # हम उम्मीद करते हैं कि सफलता पर model.run() केवल एक आउटपुट देगा।
         out = model.run(prompt)
-        err = None # यदि यह लाइन चलती है, तो कोई एरर नहीं है।
+        err = None
     except Exception as e:
-        # यदि model.run() के दौरान कोई भी एरर आता है, तो हम उसे यहां पकड़ लेंगे।
         print(f"🛑 Error during model.run() with key ...{api_key[-4:]}: {e}")
         out = None
-        err = str(e) # एरर संदेश को सहेजें।
-    # --- END OF MODIFIED SECTION ---
-
+        err = str(e)
     if err:
-        # यदि कोई एरर था, तो यहां से बाहर निकल जाएं।
         return None, f"Model Error (Key ...{api_key[-4:]}): {err}"
-        
     filename = f"clip_{idx}_{uuid.uuid4().hex}.mp4"
     filepath = os.path.join(OUTPUT_FOLDER, filename)
     try:
@@ -154,7 +166,6 @@ def generate_clip(prompt, idx, api_key, bytez_model):
             return None, f"Unexpected output type from model: {type(out)}"
     except Exception as e:
         return None, f"Failed to save or download the generated clip: {e}"
-        
     return filepath, None
 
 def process_and_merge_clips(clip_files):
@@ -217,18 +228,11 @@ def generate_video_job(prompt, num_clips, bytez_model):
         for clip in temp_clip_paths:
             if os.path.exists(clip): os.remove(clip)
 
-# ---------- UI HTML (No changes here) ----------
-HTML_TEMPLATE = """
-<!doctype html><html lang="en" data-bs-theme="dark"><head><meta charset="utf-8"><title>AI Video Generator</title><meta name="viewport" content="width=device-width, initial-scale=1"><meta name="theme-color" content="#111111"><link rel="manifest" href="/manifest.json"><link rel="apple-touch-icon" href="/static/icons/icon-512x512.png"><link href="https://cdn.jsdelivr.net/npm/[email protected]/dist/css/bootstrap.min.css" rel="stylesheet"><style>:root { --bs-body-bg: #111; --bs-body-color: #eee; } .card { background: #1a1a1a; border: 1px solid #333; border-radius: 1rem; } .btn-primary { background-color: #0d6efd; border-color: #0d6efd; } .form-control, .form-select { background-color: #222; border-color: #444; color: #eee; } .form-control:focus, .form-select:focus { background-color: #222; border-color: #0d6efd; box-shadow: 0 0 0 .25rem rgba(13, 110, 253, .25); color: #eee; } video { border-radius: 0.75rem; background-color: #000; } .container { max-width: 720px; } .page { display: none; } .page.active { display: block; } .form-label h5 { margin-bottom: 0.5rem; } #liveLog { background-color: #000; border: 1px solid #333; border-radius: 0.5rem; height: 150px; overflow-y: auto; font-family: monospace; font-size: 0.9em; padding: 10px; white-space: pre-wrap; margin-top: 1rem; text-align: left; } #liveLog p { margin: 0; padding: 2px 0; } </style></head><body><div class="container py-4"><div id="inputPage" class="page active"><h2 class="text-center mb-4">🎬 AI Video Generator</h2><div class="card p-4"><div class="mb-3"><label for="prompt" class="form-label"><h5>1. Describe your video (or use an image)</h5></label><textarea id="prompt" class="form-control" rows="3" placeholder="e.g., A robot walking through a neon-lit city in the rain"></textarea></div><div class="mb-3"><label for="imageUpload" class="form-label"><h5>2. Upload Image (Optional)</h5></label><input type="file" id="imageUpload" class="form-control" accept="image/png, image/jpeg"></div><div class="row g-3"><div class="col-md-6 mb-3"><label for="num_clips" class="form-label"><h5>3. Number of clips</h5></label><input id="num_clips" type="number" class="form-control" min="1" max="20" value="3"></div><div class="col-md-6 mb-3"><label for="style" class="form-label"><h5>4. Style</h5></label><select id="style" class="form-select"><option value="none" selected>Default</option><option value="cinematic">Cinematic</option><option value="cartoon">Cartoon</option><option value="realistic">Realistic</option><option value="minimalist">Minimalist</option></select></div></div><div class="mb-3"><label for="bytezModelCustom" class="form-label"><h5>5. Custom AI Model (Optional)</h5></label><input type="text" id="bytezModelCustom" class="form-control" placeholder="Default: ali-vilab/text-to-video-ms-1.7b"></div><div class="d-grid mt-2"><button id="startBtn" class="btn btn-primary btn-lg">🚀 Generate Video</button></div></div><div id="inputError" class="text-danger mt-3 text-center d-none"></div></div><div id="progressPage" class="page"><div class="card p-4 text-center"><div id="progressBox"><h3 class="mb-3">Generating your video...</h3><div class="d-flex justify-content-center align-items-center mb-3"><div class="spinner-border me-3" role="status"></div><div class="mono" id="statusText" style="font-size: 1.1rem;">Initializing...</div></div><div class="progress" role="progressbar" style="height: 10px"><div id="bar" class="progress-bar progress-bar-striped progress-bar-animated" style="width: 2%"></div></div><div id="liveLogContainer"><h6 class="text-secondary mt-3">Mission Control Log:</h6><div id="liveLog"></div></div></div><div id="errorBox" class="d-none"><h3 class="text-danger">An Error Occurred</h3><p id="errorMessage" class="mono bg-dark p-3 rounded"></p><button class="btn btn-secondary" onclick="goHomeAndReset()">Try Again</button></div><div id="resultBox" class="d-none"><h3 class="mb-3">✅ Your video is ready!</h3><video id="player" width="100%" controls playsinline></video><div class="d-grid gap-2 mt-3"><a id="dl" class="btn btn-success btn-lg" download>⬇️ Download Video</a><button class="btn btn-outline-secondary" onclick="goHomeAndReset()">Create Another Video</button></div></div></div></div></div>
-<script>
-    const bytezModelCustomEl = document.getElementById("bytezModelCustom"),startBtn=document.getElementById("startBtn"),promptEl=document.getElementById("prompt"),clipsEl=document.getElementById("num_clips"),styleEl=document.getElementById("style"),imageUploadEl=document.getElementById("imageUpload"),liveLogEl=document.getElementById("liveLog"),inputPage=document.getElementById("inputPage"),progressPage=document.getElementById("progressPage"),progressBox=document.getElementById("progressBox"),statusText=document.getElementById("statusText"),bar=document.getElementById("bar"),errorBox=document.getElementById("errorBox"),errorMessage=document.getElementById("errorMessage"),inputError=document.getElementById("inputError"),resultBox=document.getElementById("resultBox"),player=document.getElementById("player"),dl=document.getElementById("dl");let pollTimer=null;function showPage(e){document.querySelectorAll(".page").forEach(e=>e.classList.remove("active")),document.getElementById(e).classList.add("active")}function resetUI(){inputError.classList.add("d-none"),errorBox.classList.add("d-none"),resultBox.classList.add("d-none"),progressBox.style.display="block",bar.style.width="2%",startBtn.disabled=!1,imageUploadEl.value=null,promptEl.value="",liveLogEl.innerHTML="",bytezModelCustomEl.value=""}function goHomeAndReset(){resetUI(),showPage("inputPage")}async function startJob(){inputError.classList.add("d-none"),liveLogEl.innerHTML="";const e=promptEl.value.trim(),t=imageUploadEl.files[0];if(!e&&!t)return inputError.textContent="Please provide a text prompt or upload an image.",void inputError.classList.remove("d-none");startBtn.disabled=!0,showPage("progressPage");try{const a=new FormData;a.append("prompt",e),a.append("num_clips",parseInt(clipsEl.value,10)),a.append("style",styleEl.value),t&&a.append("image",t);const o=bytezModelCustomEl.value.trim();a.append("bytez_model",o);const n=await fetch("/start",{method:"POST",body:a}),s=await n.json();if(!n.ok)throw new Error(s.error||"Failed to start job.");pollTimer=setInterval(pollProgress,1500)}catch(l){errorMessage.textContent=l.message,progressBox.style.display="none",errorBox.classList.remove("d-none"),startBtn.disabled=!1}}async function pollProgress(){try{const e=await fetch("/progress"),t=await e.json(),a=Math.max(1,t.total),o=Math.min(100,Math.floor(t.step/a*100));if(bar.style.width=o+"%",statusText.textContent=t.message,t.live_log&&(liveLogEl.innerHTML=t.live_log.map(e=>`<p>${e}</p>`).join(""),liveLogEl.scrollTop=liveLogEl.scrollHeight),t.status==="done"&&t.video_relpath){clearInterval(pollTimer);const n=`/${t.video_relpath.replace(/\\\\/g,"/")}`;player.src=n,dl.href=n,dl.download=`ai_video_${Date.now()}.mp4`,progressBox.style.display="none",resultBox.classList.remove("d-none")}else if(t.status==="error"){clearInterval(pollTimer),errorMessage.textContent=t.error||"An unknown error occurred.",progressBox.style.display="none",errorBox.classList.remove("d-none")}}catch(r){clearInterval(pollTimer),errorMessage.textContent="Connection to server lost. Please try again.",progressBox.style.display="none",errorBox.classList.remove("d-none")}}startBtn.addEventListener("click",startJob);if('serviceWorker' in navigator){window.addEventListener('load',()=>{navigator.serviceWorker.register('/service-worker.js').then(reg=>console.log('Service worker registered.')).catch(err=>console.log('Service worker registration failed: ',err));});}
-</script>
-</body></html>"""
-
-# ---------- FLASK ROUTES (Modified start route) ----------
+# ---------- FLASK ROUTES ----------
 @app.route("/", methods=["GET"])
 def home():
-    return render_template_string(HTML_TEMPLATE)
+    # अब यह 'templates' ��ोल्डर से 'index.html' फाइल को प्रस्तुत करेगा।
+    return render_template("index.html")
 
 @app.route("/start", methods=["POST"])
 def start():
@@ -250,26 +254,29 @@ def start():
             image_data = image_file.read()
             mime_type = image_file.mimetype
             set_progress(status="running", message="Initializing...", error=None, active=True, step=0, log_message="🧠 Director (Gemini) analyzing image...")
-            
-            # /// MODIFIED SECTION ///
-            # अगली उपलब्ध जेमिनी कुंजी का चयन करें
             selected_gemini_key = next(gemini_key_cycler)
             gemini_prompt, err = get_prompt_from_gemini(image_data, user_prompt, mime_type, selected_gemini_key)
-            
             if err:
                 set_progress(status="error", error=err, active=False, log_message=f"🛑 Gemini Failure: {err}"); return jsonify({"error": err}), 500
             initial_prompt = gemini_prompt
             set_progress(log_message=f"🎬 Gemini's Idea: \"{initial_prompt[:80]}...\"")
         except Exception as e:
             err_msg = f"Failed to process image: {e}"; set_progress(status="error", error=err_msg, active=False, log_message=f"🛑 Image Error: {e}"); return jsonify({"error": err_msg}), 500
+    
     current_step = 1 if is_image_job else 0
     set_progress(status="running", message="Creating blueprint...", active=True, step=current_step, log_message="🎨 Quality Enhancer (Groq) creating blueprint...")
     visual_blueprint, err = generate_visual_blueprint_with_groq(initial_prompt, GROQ_API_KEY, GROQ_MODEL)
     if err:
-        set_progress(status="error", error=err, active=False, log_message=f"🛑 Groq Failure: {err}"); return jsonify({"error": err}), 500
-    set_progress(log_message=f"✨ Final Blueprint: \"{visual_blueprint[:80]}...\"")
+        set_progress(status="error", error=err, active=False, log_message=f"🛑 Groq Enhancer Failure: {err}"); return jsonify({"error": err}), 500
+    set_progress(log_message=f"✨ Enhanced Blueprint: \"{visual_blueprint[:80]}...\"")
+
+    set_progress(log_message="🧪 Distilling prompt for short video clip...")
+    distilled_prompt, err = distill_prompt_for_short_video(visual_blueprint, GROQ_API_KEY, GROQ_MODEL)
+    set_progress(log_message=f"🎯 Final Distilled Prompt: \"{distilled_prompt[:80]}...\"")
+    
     negative_keywords = "blurry, deformed, ugly, bad anatomy, watermark, noise, grain, low quality, distortion, glitch, pixelated, artifacts"
-    final_prompt = f"{visual_blueprint}, {negative_keywords}"
+    final_prompt = f"{distilled_prompt}, {negative_keywords}"
+    
     print(f"🚀 Final Prompt for Bytez (Model: {bytez_model}): {final_prompt}")
     job_thread = threading.Thread(target=generate_video_job, args=(final_prompt, num_clips, bytez_model), daemon=True)
     job_thread.start()