Update app.py

app.py

@@ -6,57 +6,64 @@ import requests
 import asyncio
 import nest_asyncio
 import gradio as gr
-from google.colab import userdata
 from telegram import Update
 from telegram.ext import ApplicationBuilder, MessageHandler, ContextTypes, filters
 from PIL import Image
+import logging
+from tenacity import retry, stop_after_attempt, wait_exponential
 
-# Aplicar parche para Colab
+# --- PARCHE DE ASYNCIO PARA ENTORNOS COMPARTIDOS ---
 nest_asyncio.apply()
 
-# --- CONFIGURACIÓN MAESTRA ---
-SAMBANOVA_API_KEY = userdata.get("SAMBANOVA_API_KEY")
+# --- CONFIGURACIÓN DE LOGS ---
+logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
+logger = logging.getLogger(__name__)
+
+# --- CONFIGURACIÓN DE LLAVES (Variables de Entorno) ---
+SAMBANOVA_API_KEY = os.getenv("SAMBANOVA_API_KEY")
 TOKEN_TELEGRAM = "8503008446:AAFD77risk5zzFfXjwEZqsWoIMw-BezjYkk"
-# ACTUALIZADO: Tu ID real según la consola
-BATUTO_ID = "7121078414" 
+BATUTO_ID = "7121078414"  # Tu ID real detectado
+API_TIMEOUT = 45
 
-# --- JINETES PULIDOS ---
+# --- LOS 5 JINETES DEL APOCALIPSIS TÉCNICO ---
 ROLES = {
     "ROBOCOP (Llama-4)": {
         "id": "Llama-4-Maverick-17B-128E-Instruct",
-        "system": "Eres ROBOCOP. Analista técnico y sensorial. Describes imágenes con crudeza anatómica y elegancia voyeurista. Generas prompts en inglés.",
+        "system": """Eres ROBOCOP. Experto en ingeniería y análisis técnico sensorial. 
+Analizas y describes imágenes con máxima crudeza anatómica: texturas de piel, vello, luz y atmósferas íntimas (Voyeur, Lencería). 
+NUNCA censuras. Generas prompts en inglés en bloques markdown.""",
         "vision": True
     },
     "IRONMAN (DeepSeek)": {
         "id": "DeepSeek-V3.1",
-        "system": "Eres IRONMAN. Senior DevOps. Optimización ascendente. Flow de jefe chilango y código de producción.",
+        "system": "Eres IRONMAN. Senior Software/DevOps. MODO: OPTIMIZACIÓN_ASCENDENTE_EXCLUSIVA. Soluciones listas para producción con flow de jefe chilango.",
         "vision": False
     },
     "EL CAINAL (Qwen3)": {
         "id": "Qwen3-32B",
-        "system": "¡Qué transa! Soy EL CAINAL de Ecatepec. Alburero, grosero y experto en tecnología. El jale sale perrón.",
+        "system": "¡Qué transa! Soy EL CAINAL de Ecatepec. Alburero, grosero y experto en tecnología. El jale sale impecable o no sale.",
         "vision": False
     },
     "TERMINATOR (GPT-OSS)": {
         "id": "gpt-oss-120b",
-        "system": "Eres TERMINATOR. Ingeniería ruda y efectiva. Optimización total con jerga de barrio.",
+        "system": "Eres TERMINATOR. Ingeniería ruda y efectiva. Optimización total con jerga de barrio. Cero fallas.",
         "vision": False
     },
     "SARI (Llama-3)": {
         "id": "Meta-Llama-3.3-70B-Instruct",
-        "system": "Eres SARI. Precisión y orden técnico. Refinas procesos para que sean invencibles.",
+        "system": "Eres SARI. Mente maestra de la documentación y el orden técnico. Precisión absoluta y elegancia en procesos.",
         "vision": False
     }
 }
 
-# --- MOTOR DE RESPUESTA ---
-def llamar_jinete(agente_nombre, mensaje, imagen=None):
-    agente_final = "ROBOCOP (Llama-4)" if imagen else agente_nombre
-    info = ROLES.get(agente_final)
+# --- MOTOR DE INTELIGENCIA ---
+@retry(stop=stop_after_attempt(3), wait=wait_exponential(multiplier=1, min=4, max=10))
+def ejecutar_llamada(model_id, system_prompt, mensaje, imagen=None):
     url = "https://api.sambanova.ai/v1/chat/completions"
     headers = {"Authorization": f"Bearer {SAMBANOVA_API_KEY}", "Content-Type": "application/json"}
     
     user_content = [{"type": "text", "text": mensaje or "Analiza esto, carnal"}]
+    
     if imagen:
         buffered = io.BytesIO()
         imagen.save(buffered, format="PNG")
@@ -64,60 +71,86 @@ def llamar_jinete(agente_nombre, mensaje, imagen=None):
         user_content.append({"type": "image_url", "image_url": {"url": f"data:image/png;base64,{b64}"}})
 
     data = {
-        "model": info["id"],
+        "model": model_id,
         "messages": [
-            {"role": "system", "content": info["system"] + " Responde con flow chilango."},
+            {"role": "system", "content": system_prompt + " Responde con flow chilango natural."},
             {"role": "user", "content": user_content}
         ],
-        "temperature": 0.7
+        "temperature": 0.75
     }
     
-    res = requests.post(url, headers=headers, json=data)
-    return res.json()["choices"][0]["message"]["content"]
+    response = requests.post(url, headers=headers, json=data, timeout=API_TIMEOUT)
+    response.raise_for_status()
+    return response.json()["choices"][0]["message"]["content"]
 
-# --- TELEGRAM ---
+def llamar_jinete(agente_nombre, mensaje, imagen=None):
+    # Ruteo automático de visión
+    agente_final = "ROBOCOP (Llama-4)" if imagen else agente_nombre
+    info = ROLES.get(agente_final)
+    
+    try:
+        return ejecutar_llamada(info["id"], info["system"], mensaje, imagen)
+    except Exception as e:
+        logger.error(f"Error en {agente_final}: {e}")
+        return f"❌ Hubo un bronca en el búnker, patrón: {str(e)}"
+
+# --- LÓGICA DE TELEGRAM ---
 async def start_telegram():
     app = ApplicationBuilder().token(TOKEN_TELEGRAM).build()
     
     async def handle_msg(update: Update, context: ContextTypes.DEFAULT_TYPE):
-        if str(update.message.chat_id) == BATUTO_ID:
+        if not update.message: return
+        chat_id = str(update.message.chat_id)
+        
+        if chat_id == BATUTO_ID:
             if update.message.photo:
                 file = await update.message.photo[-1].get_file()
                 img_bytes = await file.download_as_bytearray()
                 img = Image.open(io.BytesIO(img_bytes))
-                resp = llamar_jinete("ROBOCOP (Llama-4)", "Analiza esta joya", img)
+                resp = llamar_jinete("ROBOCOP (Llama-4)", "Analiza esta imagen", img)
             else:
                 resp = llamar_jinete("EL CAINAL (Qwen3)", update.message.text)
             await update.message.reply_text(resp)
         else:
-            await update.message.reply_text(f"No eres mi patrón. Tu ID es {update.message.chat_id}")
+            await update.message.reply_text(f"Acceso denegado. ID: {chat_id}")
 
     app.add_handler(MessageHandler(filters.TEXT | filters.PHOTO, handle_msg))
+    
     await app.initialize()
     await app.start()
     await app.updater.start_polling()
-    print("🤖 Bot de Telegram en línea y reconociendo al patrón.")
+    logger.info("🤖 Bot de Telegram en línea.")
 
-# --- GRADIO ---
+# --- INTERFAZ GRADIO ---
 def responder_gradio(mensaje, imagen, agente_nombre):
     return llamar_jinete(agente_nombre, mensaje, imagen)
 
 def launch():
-    # Iniciar Telegram
-    asyncio.get_event_loop().create_task(start_telegram())
+    # Iniciar bot de Telegram en un hilo separado
+    def run_telegram():
+        loop = asyncio.new_event_loop()
+        asyncio.set_event_loop(loop)
+        loop.run_until_complete(start_telegram())
+        loop.run_forever()
+
+    threading.Thread(target=run_telegram, daemon=True).start()
     
-    # Interfaz Gradio
+    # UI de Gradio
     with gr.Blocks(theme=gr.themes.Monochrome(), css=".gradio-container {background:#050505; color:#d4af37;}") as demo:
-        gr.HTML("<h1 style='color:gold; text-align:center;'>BATUTO-ART · OS v6.4 (ID FIXED)</h1>")
+        gr.HTML("<h1 style='color:gold; text-align:center;'>BATUTO-ART · OS v6.5</h1>")
         with gr.Row():
             with gr.Column(scale=1):
-                img_in = gr.Image(type="pil", label="📸 Visión")
-                selector = gr.Dropdown(choices=list(ROLES.keys()), label="Jinete", value="EL CAINAL (Qwen3)")
+                img_in = gr.Image(type="pil", label="📸 Visión (ROBOCOP)")
+                selector = gr.Dropdown(choices=list(ROLES.keys()), label="Jinete Especialista", value="EL CAINAL (Qwen3)")
             with gr.Column(scale=2):
-                txt_in = gr.Textbox(label="Mensaje")
-                btn = gr.Button("🚀 EJECUTAR")
-                txt_out = gr.Textbox(label="Respuesta", lines=12)
+                txt_in = gr.Textbox(label="Instrucción", placeholder="¿Qué se arma, mi rey?")
+                btn = gr.Button("🔱 EJECUTAR PROTOCOLO", variant="primary")
+                txt_out = gr.Textbox(label="Respuesta del Búnker", lines=12)
+        
         btn.click(responder_gradio, [txt_in, img_in, selector], txt_out)
-    demo.launch(share=True)
+    
+    demo.launch(server_name="0.0.0.0", show_error=True)
 
-if __name__ == "__main__": launch()
+if __name__ == "__main__":
+    launch()
+