Update app.py

app.py

@@ -1,288 +1,245 @@
 import os
-import io
-import base64
-import json
-import requests
-import tempfile
-from PIL import Image
 import gradio as gr
-from huggingface_hub import InferenceClient
+import base64
+from openai import OpenAI
 
-# ===========================================
+# ============================================================
 #   CONFIG
-# ===========================================
-API_KEY = os.getenv("REVE_API_KEY")
-HF_TOKEN = os.getenv("HF_TOKEN")
-BASE_URL = "https://api.sambanova.ai/v1"
+# ============================================================
+
+API_KEY = os.getenv("OPENROUTER_API_KEY")
+if not API_KEY:
+    raise ValueError("Falta la variable OPENROUTER_API_KEY.")
 
-DEBUG = False  # Ponlo True si quieres ver el payload exacto enviado
+client = OpenAI(
+    base_url="https://openrouter.ai/api/v1",
+    api_key=API_KEY
+)
 
+# ============================================================
+#   TODOS TUS MODELOS COMPLETOS
+# ============================================================
 
-# ===========================================
-#   TUS MODELOS (RESPETADOS 1:1)
-# ===========================================
 MODELS = {
-    "general_fast": {
-        "name": "Meta-Llama-3.1-8B-Instruct",
-        "role": "🔄 Respuestas rápidas y generales",
-        "description": "Conversación ligera y eficiente.",
-        "supports_images": False
+    # -------- META LLAMA --------
+    "llama_3.1_70b": {
+        "name": "meta-llama/llama-3.1-70b-instruct",
+        "role": "Respondes con precisión técnica y claridad."
     },
-    "general_smart": {
-        "name": "Meta-Llama-3.3-70B-Instruct",
-        "role": "🧠 Razonamiento profundo",
-        "description": "Análisis detallado y avanzado.",
-        "supports_images": False
+    "llama_3.1_405b": {
+        "name": "meta-llama/llama-3.1-405b-instruct",
+        "role": "Eres experto en programación, ciencia y análisis avanzado."
     },
-    "coding_expert": {
-        "name": "DeepSeek-V3.1",
-        "role": "💻 Programación y debugging",
-        "description": "Ideal para desarrollo.",
-        "supports_images": False
+
+    # -------- LLAMA 3.2 VISION --------
+    "llama_3.2_11b_vision": {
+        "name": "meta-llama/llama-3.2-11b-vision-instruct",
+        "role": "Eres un modelo experto en análisis visual detallado."
     },
-    "coding_alt": {
-        "name": "DeepSeek-V3-0324",
-        "role": "⚡ Código rápido",
-        "description": "Alternativa veloz.",
-        "supports_images": False
+    "llama_3.2_90b_vision": {
+        "name": "meta-llama/llama-3.2-90b-vision-instruct",
+        "role": "Eres un analista visual avanzado altamente preciso."
     },
-    "massive_brain": {
-        "name": "gpt-oss-120b",
-        "role": "🏛️ Sabiduría masiva",
-        "description": "Problemas pesados y complejos.",
-        "supports_images": False
+
+    # -------- QWEN --------
+    "qwen_72b": {
+        "name": "qwen/qwen2.5-72b-instruct",
+        "role": "Respondes de forma profesional, directa y clara."
     },
-    "specialized_1": {
-        "name": "DeepSeek-V3.1-Terminus",
-        "role": "🎯 Especialista técnico",
-        "description": "Tareas científicas y avanzadas.",
-        "supports_images": False
+    "qwen_110b": {
+        "name": "qwen/qwen2.5-110b-instruct",
+        "role": "Asistente experto en razonamiento estructurado."
     },
-    "specialized_2": {
-        "name": "Llama-3.3-Swallow-70B-Instruct-v0.4",
-        "role": "🔥 Sin censura",
-        "description": "Modelo sin restricciones.",
-        "supports_images": False
+
+    # -------- GPT / OPENAI --------
+    "gpt_4.1": {
+        "name": "openai/gpt-4.1",
+        "role": "Asistente avanzado para cualquier tarea general."
     },
-    "multilingual": {
-        "name": "Qwen3-32B",
-        "role": "🌍 Multilingüe",
-        "description": "Múltiples idiomas.",
-        "supports_images": False
+    "gpt_4.1_mini": {
+        "name": "openai/gpt-4.1-mini",
+        "role": "Modelo rápido y eficiente, ideal para respuestas concisas."
     },
-    "vision_expert": {
-        "name": "Llama-4-Maverick-17B-128E-Instruct",
-        "role": "👁️ Visión avanzada",
-        "description": "Análisis de imágenes.",
-        "supports_images": True
+    "gpt_4o_mini": {
+        "name": "openai/gpt-4o-mini",
+        "role": "Asistente veloz con buena comprensión general."
     },
-    "boudoir_specialist": {
-        "name": "ALLaM-7B-Instruct-preview",
-        "role": "🎭 Especialista en Fotografía Íntima Profesional",
-        "description": "Experto en prompts para fotografía boudoir.",
-        "supports_images": False,
-        "specialties": [
-            "Fotografía Boudoir",
-            "Desnudo Artístico",
-            "Moda Sensual",
-            "Lencería y moda íntima"
-        ],
-        "technical_expertise": [
-            "Iluminación suave",
-            "Composición elegante",
-            "Dirección de poses",
-            "Edición fina",
-            "Escenografía íntima"
-        ],
-        "ethical_principles": [
-            "Consentimiento explícito",
-            "Positividad corporal"
-        ]
-    }
-}
-
 
-# ===========================================
-#   HELPERS
-# ===========================================
-def encode_image(img):
-    if img is None:
-        return None
-    buffer = io.BytesIO()
-    img.save(buffer, format="PNG")
-    return base64.b64encode(buffer.getvalue()).decode("utf-8")
-
-
-def _safe_content(c):
-    """Convierte cualquier cosa a algo aceptable por SambaNova."""
-    if c is None:
-        return ""
-    if isinstance(c, str):
-        return c
-    if isinstance(c, dict):
-        return c
-    if isinstance(c, list):
-        return c
-    return str(c)
-
-
-def validate_messages(messages):
-    """
-    Corrige TODO lo que pueda romper SambaNova:
-    - Asegura diccionario {role, content}
-    - Limpia historial de Gradio
-    - Evita contenidos no serializables
-    """
-    out = []
-
-    for m in messages:
-
-        # Caso válido
-        if isinstance(m, dict) and "role" in m and "content" in m:
-            out.append({
-                "role": m["role"],
-                "content": _safe_content(m["content"])
-            })
-            continue
-
-        # Tuplas del historial (user, assistant)
-        if isinstance(m, (list, tuple)) and len(m) == 2:
-            u, a = m
-            out.append({"role": "user", "content": _safe_content(u)})
-            out.append({"role": "assistant", "content": _safe_content(a)})
-            continue
-
-        # Todo lo demás
-        out.append({"role": "user", "content": _safe_content(m)})
-
-    return out
-
-
-# ===========================================
-#   LLAMADA A SAMBANOVA (LA BUENA)
-# ===========================================
-def call_sambanova(model, messages):
-
-    messages = validate_messages(messages)
-
-    payload = {
-        "model": model,
-        "messages": messages,
-        "stream": False
-    }
-
-    if DEBUG:
-        return json.dumps(payload, indent=2)
+    # -------- CLAUDE --------
+    "claude_3.5_sonnet": {
+        "name": "anthropic/claude-3.5-sonnet",
+        "role": "Especialista en redacción, precisión y análisis profundo."
+    },
+    "claude_3.5_haiku": {
+        "name": "anthropic/claude-3.5-haiku",
+        "role": "Modelo rápido con buena comprensión general."
+    },
+    "claude_3_opus": {
+        "name": "anthropic/claude-3-opus",
+        "role": "Máxima capacidad de análisis y lenguaje."
+    },
 
-    headers = {
-        "Authorization": f"Bearer {API_KEY}",
-        "Content-Type": "application/json"
-    }
+    # -------- GOOGLE --------
+    "gemini_flash": {
+        "name": "google/gemini-flash-1.5",
+        "role": "Especialista en escenarios visuales y respuestas rápidas."
+    },
+    "gemini_pro": {
+        "name": "google/gemini-pro-1.5",
+        "role": "Razonador general robusto y flexible."
+    },
+    "gemini_thinking": {
+        "name": "google/gemini-1.5-thinking",
+        "role": "Modelo de razonamiento profundo y detallado."
+    },
 
-    r = requests.post(
-        f"{BASE_URL}/chat/completions",
-        json=payload,
-        headers=headers,
-        timeout=60
-    )
+    # -------- DEEPSEEK --------
+    "deepseek_r1": {
+        "name": "deepseek/deepseek-r1",
+        "role": "Razonamiento profundo y cadena de pensamiento estructurada."
+    },
 
-    try:
-        data = r.json()
-        return data["choices"][0]["message"]["content"]
-    except:
-        return "Respuesta inesperada:\n" + r.text
+    # -------- MISTRAL --------
+    "mistral_large": {
+        "name": "mistral/mistral-large-latest",
+        "role": "Asistente técnico avanzado y preciso."
+    },
+    "mixtral_8x7b": {
+        "name": "mistral/mixtral-8x7b-instruct",
+        "role": "Modelo eficiente para tareas complejas sin alto costo."
+    },
 
+    # -------- REKA --------
+    "reka_core": {
+        "name": "reka/core",
+        "role": "Asistente racional y estructurado."
+    },
 
-# ===========================================
-#   CHAT PRINCIPAL (CORREGIDO)
-# ===========================================
-def chat_logic(user_text, user_image, model_selection, history):
+    # -------- SAMBANOVA --------
+    "samba_allam_7b": {
+        "name": "sambanova/ALLAM-1-7B",
+        "role": "Asistente optimizado para rendimiento y claridad."
+    },
 
-    if history is None:
-        history = []
+    # -------- FLUX --------
+    "flux_pro": {
+        "name": "black-forest-labs/flux-1.1-pro",
+        "role": "Experto en generación y análisis de imágenes."
+    }
+}
 
-    model = MODELS[model_selection]
-    supports_images = model["supports_images"]
+# ============================================================
+#   BUILDER DE MENSAJES (CORRIGE EL ERROR)
+# ============================================================
 
+def build_messages(system_prompt, user_input, history):
     messages = []
 
     # SYSTEM
-    messages.append({"role": "system", "content": model["role"]})
-
-    # HISTORIAL
-    for u, a in history:
-        messages.append({"role": "user", "content": _safe_content(u)})
-        messages.append({"role": "assistant", "content": _safe_content(a)})
+    if system_prompt:
+        messages.append({"role": "system", "content": system_prompt})
 
-    # MENSAJE DEL USUARIO
-    if user_image and supports_images:
-        encoded = encode_image(user_image)
-        user_content = [
-            {"type": "text", "text": user_text or ""},
-            {"type": "image_url", "image_url": {"url": f"data:image/png;base64,{encoded}"}}
-        ]
-        messages.append({"role": "user", "content": user_content})
+    # HISTORIAL NORMALIZADO
+    for item in history:
+        if isinstance(item, tuple) and len(item) == 2:
 
-    else:
-        messages.append({"role": "user", "content": _safe_content(user_text)})
+            user_msg, assistant_msg = item
 
-    reply = call_sambanova(model["name"], messages)
+            if isinstance(user_msg, str) and user_msg.strip():
+                messages.append({"role": "user", "content": user_msg})
 
-    history.append((user_text or "[Imagen]", reply))
-    return history, history
+            if isinstance(assistant_msg, str) and assistant_msg.strip():
+                messages.append({"role": "assistant", "content": assistant_msg})
 
+    # MENSAJE ACTUAL
+    messages.append({"role": "user", "content": user_input})
 
-# ===========================================
-#   GENERACIÓN HF
-# ===========================================
-def generate_image(prompt):
-    if not HF_TOKEN:
-        return None, "❌ Falta HF_TOKEN", gr.update(visible=False)
+    return messages
 
-    try:
-        client = InferenceClient(token=HF_TOKEN)
-        img = client.text_to_image(prompt, model="stabilityai/stable-diffusion-xl-base-1.0")
+# ============================================================
+#   IMAGE → base64
+# ============================================================
 
-        with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as tmp:
-            img.save(tmp)
-            return img, "OK", gr.update(value=tmp.name, visible=True)
+def encode_image_to_base64(image):
+    if image is None:
+        return None
+    import io
+    import PIL.Image
+    buf = io.BytesIO()
+    image.save(buf, format="PNG")
+    return base64.b64encode(buf.getvalue()).decode("utf-8")
+
+# ============================================================
+#   GENERADOR
+# ============================================================
+
+def generate(model_key, prompt, image, history):
+    selected = MODELS.get(model_key)
+    if not selected:
+        return "Error: Modelo no encontrado."
+
+    system_prompt = selected["role"]
+    messages = build_messages(system_prompt, prompt, history)
+
+    # Imagen opcional
+    if image:
+        img_b64 = encode_image_to_base64(image)
+
+        messages.append({
+            "role": "user",
+            "content": [
+                {"type": "input_image", "image_url": f"data:image/png;base64,{img_b64}"},
+                {"type": "text", "text": prompt}
+            ]
+        })
+
+    # Request
+    response = client.chat.completions.create(
+        model=selected["name"],
+        messages=messages,
+        temperature=0.7
+    )
 
-    except Exception as e:
-        return None, f"❌ Error: {e}", gr.update(visible=False)
+    return response.choices[0].message.content
 
+# ============================================================
+#   UI GRADIO
+# ============================================================
 
-# ===========================================
-#   UI
-# ===========================================
 def create_ui():
+    with gr.Blocks(theme="soft") as demo:
+        gr.Markdown("## Multi-Modelo OpenRouter (Todos tus modelos activados)")
+
+        with gr.Row():
+            model_dropdown = gr.Dropdown(
+                choices=list(MODELS.keys()),
+                value="llama_3.1_70b",
+                label="Modelo"
+            )
 
-    with gr.Blocks() as demo:
-
-        gr.Markdown("# 🤖 Chat Multimodal SambaNova")
+        prompt = gr.Textbox(
+            label="Mensaje",
+            placeholder="Escribe tu mensaje…",
+            lines=4
+        )
 
-        with gr.Tab("Chat"):
-            model_sel = gr.Dropdown(choices=list(MODELS.keys()), value="general_fast", label="Modelo")
-            chat = gr.Chatbot(height=420)
-            img = gr.Image(type="pil")
-            txt = gr.Textbox(label="Mensaje")
-            send = gr.Button("Enviar")
+        image_input = gr.Image(type="pil", label="Imagen opcional")
+        chatbot = gr.Chatbot()
+        send = gr.Button("Enviar")
 
-            send.click(
-                chat_logic,
-                inputs=[txt, img, model_sel, chat],
-                outputs=[chat, chat]
-            )
+        def respond(m, t, img, hist):
+            output = generate(m, t, img, hist)
+            hist.append((t, output))
+            return hist, hist
 
-        with gr.Tab("Imágenes"):
-            p = gr.Textbox(label="Prompt")
-            b = gr.Button("Generar")
-            out = gr.Image()
-            d = gr.DownloadButton("Descargar", visible=False)
-            b.click(generate_image, inputs=[p], outputs=[out, txt, d])
+        send.click(
+            respond,
+            inputs=[model_dropdown, prompt, image_input, chatbot],
+            outputs=[chatbot, chatbot]
+        )
 
     return demo
 
-
 demo = create_ui()
 
 if __name__ == "__main__":