Update app.py

app.py

@@ -1,9 +1,7 @@
 import os
 import gradio as gr
-from sambanova import SambaNova  # Asegúrate de tener instalado el SDK de SambaNova
-import requests
+
 from typing import List, Dict, Union, Tuple
-import json
 from data import (
     PROFESSIONS,
     LACE_THONG_STYLES,
@@ -42,11 +40,19 @@ VOYEUR_SPECIALIST_CONFIG = {
     ],
 }
 
-# Cliente SambaNova inicializado con variables de entorno
-client = SambaNova(
-    api_key=os.getenv("SAMBANOVA_API_KEY"),
-    base_url=os.getenv("SAMBANOVA_BASE_URL")
-)
+# ============================================
+# CLIENTE DEL MODELO (OPCIONAL SI FALLA)
+# ============================================
+
+try:
+    from sambanova import SambaNova
+
+    client = SambaNova(
+        api_key=os.getenv("SAMBANOVA_API_KEY"),
+        base_url=os.getenv("SAMBANOVA_BASE_URL"),
+    )
+except Exception:
+    client = None
 
 # ============================================
 # PROMPT DEL SISTEMA
@@ -91,14 +97,25 @@ def build_prompt(user_text: str) -> str:
     return SYSTEM_PROMPT_BASE + "\n\nUSER:\n" + user_text
 
 def call_model(prompt: str) -> str:
-    response = client.chat.completions.create(
-        model=MODEL_NAME,
-        messages=[
-            {"role": "system", "content": SYSTEM_PROMPT_BASE},
-            {"role": "user", "content": prompt}
-        ],
-    )
-    return response.choices[0].message["content"]
+    if client is None:
+        return "⚠️ Error de configuración del modelo (SambaNova). Revisa tu SAMBANOVA_API_KEY y SAMBANOVA_BASE_URL."
+    try:
+        response = client.chat.completions.create(
+            model=MODEL_NAME,
+            messages=[
+                {"role": "system", "content": SYSTEM_PROMPT_BASE},
+                {"role": "user", "content": prompt},
+            ],
+        )
+        # Ajusta según el formato real del SDK
+        choice = response.choices[0]
+        # algunas SDKs exponen .message.content, otras solo .text
+        content = getattr(getattr(choice, "message", None), "content", None)
+        if not content:
+            content = getattr(choice, "text", "")
+        return content or "Sin contenido en la respuesta del modelo."
+    except Exception as e:
+        return f"⚠️ Error llamando al modelo: {e}"
 
 def infer(user_text: str) -> str:
     prompt = build_prompt(user_text)
@@ -134,20 +151,20 @@ def get_system_config_ui() -> str:
     """
 
 # ============================================
-# CSS MODERNO (USAR VIA gr.HTML)
+# CSS MODERNO
 # ============================================
 
 CUSTOM_CSS = """
 <style>
-body { background-color: #1a1a1a; }
-.gradio-container { background: #282c34; border-radius: 12px; padding: 20px; }
+body { background-color: #111111; }
+.gradio-container { background: #181a1f; border-radius: 12px; padding: 20px; }
 h1 { color: #ff7043; }
 button { background-color: #ff7043 !important; color:white !important; }
 </style>
 """
 
 # ============================================
-# INTERFAZ GRADIO (SIN add_component)
+# INTERFAZ GRADIO
 # ============================================
 
 with gr.Blocks(title="Voyeur Prompt Generator") as demo: