upload imagenes/videos del cliente al crear canal

cliente_template.py

@@ -39,6 +39,7 @@ except Exception as e:
 GROQ_KEY   = "GROQ_PLACEHOLDER"
 TIPO_VIDEO = "TIPO_PLACEHOLDER"
 PEXELS_KEY = "PEXELS_PLACEHOLDER"
+os.makedirs("media_cliente", exist_ok=True)
 LOGS       = []
 STATUS     = "Iniciando..."
 
@@ -88,19 +89,31 @@ async def pipeline():
                 import subprocess
                 subprocess.run(["ffmpeg","-y","-f","lavfi","-i","anullsrc=r=44100:cl=mono","-t","60","voz.mp3"], capture_output=True)
             log("Voz lista")
-            STATUS = "Descargando fondo..."
-            h = {"Authorization": PEXELS_KEY}
-            r = requests.get(f"https://api.pexels.com/videos/search?query={NICHO}&per_page=15", headers=h)
-            vids = r.json().get("videos", [])
+            STATUS = "Preparando fondos..."
             fondos = []
-            if vids:
-                random.shuffle(vids)
-                for i, v in enumerate(vids[:6]):
-                    try:
-                        u = v["video_files"][0]["link"]
-                        with open(f"fondo{i}.mp4","wb") as f: f.write(requests.get(u, timeout=30).content)
-                        fondos.append(f"fondo{i}.mp4")
-                    except: pass
+            # Revisar si hay media propia del cliente
+            archivos_propios = []
+            if os.path.exists("media_cliente"):
+                for f in sorted(os.listdir("media_cliente")):
+                    if f.lower().endswith((".mp4", ".mov", ".jpg", ".jpeg", ".png", ".webp")):
+                        archivos_propios.append(f"media_cliente/{f}")
+            if archivos_propios:
+                random.shuffle(archivos_propios)
+                fondos = archivos_propios[:6]
+                log(f"Usando {len(fondos)} archivos del cliente")
+            else:
+                STATUS = "Descargando fondo..."
+                h = {"Authorization": PEXELS_KEY}
+                r = requests.get(f"https://api.pexels.com/videos/search?query={NICHO}&per_page=15", headers=h)
+                vids = r.json().get("videos", [])
+                if vids:
+                    random.shuffle(vids)
+                    for i, v in enumerate(vids[:6]):
+                        try:
+                            u = v["video_files"][0]["link"]
+                            with open(f"fondo{i}.mp4","wb") as f: f.write(requests.get(u, timeout=30).content)
+                            fondos.append(f"fondo{i}.mp4")
+                        except: pass
             log(f"Fondos listos: {len(fondos)}")
             STATUS = "Renderizando video..."
             from moviepy.editor import AudioFileClip as AC, VideoFileClip as VC, concatenate_videoclips

main.py

@@ -114,7 +114,7 @@ cargar_clientes()
 # ══════════════════════════════════════════
 #  CREAR SPACE EN HUGGING FACE
 # ══════════════════════════════════════════
-async def crear_space(nicho: str, email: str, tipo: str = "largo", config_parsed: dict = None) -> str:
+async def crear_space(nicho: str, email: str, tipo: str = "largo", config_parsed: dict = None, archivos_subidos: list = None) -> str:
     space_id = f"tb-{re.sub(r'[^a-z0-9]','', nicho.lower()[:12])}-{int(time.time())%9999}"
     cuentas = [
         (os.environ.get("HF_TOKEN_5",""), os.environ.get("HF_USER_5","Alejandro")),
@@ -167,14 +167,22 @@ RUN pip install -r requirements.txt
 COPY main.py .
 EXPOSE 7860
 CMD [\"uvicorn\", \"main:app\", \"--host\", \"0.0.0.0\", \"--port\", \"7860\"]"""
+        files_lista = [
+            {"path": "main.py", "content": b64(main_code), "encoding": "base64"},
+            {"path": "requirements.txt", "content": b64(req), "encoding": "base64"},
+            {"path": "README.md", "content": b64(readme), "encoding": "base64"},
+            {"path": "Dockerfile", "content": b64(dockerfile), "encoding": "base64"}
+        ]
+        # Añadir archivos del cliente si los subió
+        if archivos_subidos:
+            for i, arc in enumerate(archivos_subidos[:20]):
+                nombre = arc.get("nombre", f"media_{i}.jpg").replace("/","_").replace("\\","_")
+                contenido = arc.get("contenido", "")
+                if contenido:
+                    files_lista.append({"path": f"media_cliente/{nombre}", "content": contenido, "encoding": "base64"})
         commit_payload = {
             "summary": "TubeBot init",
-            "files": [
-                {"path": "main.py", "content": b64(main_code), "encoding": "base64"},
-                {"path": "requirements.txt", "content": b64(req), "encoding": "base64"},
-                {"path": "README.md", "content": b64(readme), "encoding": "base64"},
-                {"path": "Dockerfile", "content": b64(dockerfile), "encoding": "base64"}
-            ]
+            "files": files_lista
         }
         r2 = await client.post(
             f"https://huggingface.co/api/spaces/{repo_name}/commit/main",
@@ -503,6 +511,9 @@ input:focus,select:focus{border-color:#ff2d2d;}
       </select>
       <label style="font-size:.85rem;color:#aaa;margin-top:12px;display:block">✨ ¿Tienes alguna preferencia? <span style="color:#666">(opcional)</span></label>
       <textarea id="config${i}" rows="4" placeholder="Ej: voz femenina en ingles, subtitulos grandes arriba, tono motivacional, sin musica..." style="width:100%;padding:12px;background:#080808;color:#fff;border:1px solid #333;border-radius:8px;resize:vertical;font-family:system-ui;margin-top:6px"></textarea>
+      <label style="font-size:.85rem;color:#aaa;margin-top:12px;display:block">📸 Sube tus imagenes o videos <span style="color:#666">(opcional, max 20)</span></label>
+      <input type="file" id="media${i}" multiple accept="image/*,video/*" style="margin-top:6px;width:100%;padding:10px;background:#080808;color:#fff;border:1px solid #333;border-radius:8px"/>
+      <div id="mediainfo${i}" style="color:#888;font-size:.8rem;margin-top:4px"></div>
     `;
     container.appendChild(div);
   }
@@ -547,6 +558,26 @@ async function activar(){
     const conf = document.getElementById('config'+i);
     if(n && t) canales.push({nicho: n.value, tipo: t.value, config: conf ? conf.value : ""});
   }
+  
+  // Recoger archivos y convertir a base64
+  const archivos = [];
+  const fileInput = document.getElementById('media1');
+  if (fileInput && fileInput.files && fileInput.files.length > 0) {
+    document.getElementById('mediainfo1').textContent = "Procesando archivos...";
+    const files = Array.from(fileInput.files).slice(0, 20);
+    for (const f of files) {
+      try {
+        const b64 = await new Promise((resolve, reject) => {
+          const reader = new FileReader();
+          reader.onload = () => resolve(reader.result.split(',')[1]);
+          reader.onerror = reject;
+          reader.readAsDataURL(f);
+        });
+        archivos.push({nombre: f.name, contenido: b64});
+      } catch(e) { console.log("Error archivo", f.name, e); }
+    }
+    document.getElementById('mediainfo1').textContent = archivos.length + " archivos listos";
+  }
   document.getElementById('btn').disabled = true;
   document.getElementById('loading').style.display = 'block';
   try{
@@ -556,7 +587,7 @@ async function activar(){
       method:'POST',
       signal: ctrl.signal,
       headers:{'Content-Type':'application/json'},
-      body: JSON.stringify({email, canales, token: "TOKEN_PLACEHOLDER"})
+      body: JSON.stringify({email, canales, archivos, token: "TOKEN_PLACEHOLDER"})
     });
     const d = await r.json();
     if(d.url){
@@ -644,6 +675,7 @@ async def crear(request: Request):
     data = await request.json()
     email = data.get("email","").strip().lower()
     canales_data = data.get("canales", [])
+    archivos_subidos = data.get("archivos", [])
     if canales_data:
         nicho = canales_data[0].get("nicho","tecnologia")
         tipo = canales_data[0].get("tipo","largo")
@@ -669,7 +701,7 @@ async def crear(request: Request):
     # Parsear preferencias del cliente con Groq ANTES de crear
     config_parsed = await parsear_config_groq(config_libre)
     # Crear el canal pasando la config
-    resultado = await crear_space(nicho, email, tipo, config_parsed)
+    resultado = await crear_space(nicho, email, tipo, config_parsed, archivos_subidos)
     if not resultado:
         return JSONResponse({"error": "No se pudo crear el canal"}, status_code=500)
     u = resultado["url"]