Update app.py

app.py

@@ -1,49 +1,98 @@
 import gradio as gr
 from huggingface_hub import hf_hub_download
 from llama_cpp import Llama
+import traceback
 
-# ⚡ Configuración ultra ligera para CPU gratuita
+# ⚡ Descarga y carga del modelo (solo una vez, se cachea)
 REPO_ID = "unsloth/Qwen2.5-Coder-0.5B-Instruct-GGUF"
 FILENAME = "Qwen2.5-Coder-0.5B-Instruct-Q4_K_M.gguf"
-
-# Descarga el modelo (cacheado después del primer uso)
 model_path = hf_hub_download(repo_id=REPO_ID, filename=FILENAME)
 
-# Carga el modelo con parámetros optimizados
 llm = Llama(
     model_path=model_path,
-    n_ctx=1024,
-    n_threads=1,
-    n_batch=64,
+    n_ctx=1024,        # Contexto suficiente para respuestas
+    n_threads=2,       # Usa ambos núcleos virtuales
+    n_batch=128,
     use_mmap=True,
     verbose=False
 )
 
-def roblox_master(message, history):
-    prompt = f"""<|im_start|>system
-Eres RobloxMaster AI, experto en Luau, Roblox, construcción y marketing. Responde en español conciso y útil.
-<|im_end|>
-<|im_start|>user
-{message}<|im_end|>
-<|im_start|>assistant
+SYSTEM_PROMPT = (
+    "Eres TixAI, un asistente de IA superinteligente especializado en Roblox, scripting Luau, "
+    "construcción de juegos y marketing. Responde SIEMPRE en español, de forma concisa y útil. "
+    "Incluye ejemplos de código cuando sea relevante."
+)
+
+def format_prompt(message, history):
+    """Construye el prompt con el formato de chat de Qwen2.5."""
+    prompt = f"<|im_start|>system\n{SYSTEM_PROMPT}<|im_end|>\n"
+    for user_msg, bot_msg in history:
+        prompt += f"<|im_start|>user\n{user_msg}<|im_end|>\n"
+        prompt += f"<|im_start|>assistant\n{bot_msg}<|im_end|>\n"
+    prompt += f"<|im_start|>user\n{message}<|im_end|>\n"
+    prompt += "<|im_start|>assistant\n"
+    return prompt
+
+def generate_response(message, history):
+    """Genera la respuesta token a token (streaming)."""
+    prompt = format_prompt(message, history)
+    try:
+        # Streaming: devolvemos cada token según se genera
+        stream = llm(
+            prompt,
+            max_tokens=256,
+            temperature=0.6,
+            top_p=0.9,
+            stop=["<|im_end|>", "<|im_start|>"],
+            stream=True,
+        )
+        partial = ""
+        for chunk in stream:
+            token = chunk["choices"][0]["text"]
+            partial += token
+            yield partial
+    except Exception as e:
+        # En caso de error, lo mostramos en el chat
+        yield f"❌ Error interno: {traceback.format_exc()}"
+
+# 🎨 Interfaz estilo ChatGPT / DeepSeek
+custom_css = """
+.gradio-container {
+    max-width: 800px;
+    margin: auto;
+    font-family: 'Inter', system-ui, sans-serif;
+}
+.bubble {
+    border-radius: 18px;
+    padding: 12px 18px;
+    margin: 8px 0;
+    max-width: 80%;
+    line-height: 1.5;
+}
+.user {
+    background: #f0f0f0;
+    color: #111;
+}
+.bot {
+    background: #e3f2fd;
+    color: #111;
+}
+footer {visibility: hidden}  /* Oculta el "Flag" de Hugging Face */
 """
-    output = llm(
-        prompt,
-        max_tokens=128,
-        temperature=0.6,
-        top_p=0.9,
-        stop=["<|im_end|>", "<|im_start|>"],
-        echo=False
-    )
-    return output["choices"][0]["text"].strip()
-
-# Interfaz de chat
+
 iface = gr.ChatInterface(
-    fn=roblox_master,
-    title="🎮 TixAI – Tu asistente Roblox",
-    description="Experto en scripting, construcción y marketing. Pregúntame lo que quieras.",
-    examples=["¿Cómo hacer un botón que te dé dinero?", "Explica qué es un LocalScript"]
+    fn=generate_response,
+    title="🤖 TixAI — Roblox Expert",
+    description="Tu asistente definitivo para scripting, construcción y marketing en Roblox.",
+    theme=gr.themes.Soft(),
+    css=custom_css,
+    examples=[
+        "¿Cómo crear un sistema de mascotas que sigan al jugador?",
+        "Dame un script para un obby con checkpoints.",
+        "¿Qué gamepasses me recomiendas para monetizar un simulador?"
+    ],
+    cache_examples=False,
 )
 
-# ⚠️ Launch SIN show_api (compatible con Gradio 6)
-iface.launch(server_name="0.0.0.0")
+if __name__ == "__main__":
+    iface.launch(server_name="0.0.0.0")