Upload app.py with huggingface_hub

app.py

@@ -14,21 +14,21 @@ def get_token(request: gr.Request):
 def chat_stream(message, history, request: gr.Request):
     token = get_token(request)
     if not token:
-        yield history + [{"role": "assistant", "content": "⚠️ Por favor, inicia sesión con tu cuenta de Hugging Face para usar el chat."}]
+        yield history + [(message, "⚠️ Por favor, inicia sesión con tu cuenta de Hugging Face para usar el chat.")]
         return
     
     messages = []
-    for msg in history:
-        if msg["role"] == "user":
-            messages.append({"role": "user", "content": msg["content"]})
-        elif msg["role"] == "assistant":
-            messages.append({"role": "assistant", "content": msg["content"]})
+    for user_msg, bot_msg in history:
+        if user_msg:
+            messages.append({"role": "user", "content": user_msg})
+        if bot_msg:
+            messages.append({"role": "assistant", "content": bot_msg})
     messages.append({"role": "user", "content": message})
     
     headers = {"Authorization": f"Bearer {token}", "Content-Type": "application/json"}
     payload = {"model": MODEL, "messages": messages, "stream": True, "max_tokens": 2048}
     
-    chat_history = history + [{"role": "user", "content": message}]
+    chat_history = history + [(message, None)]
     yield chat_history
     
     full_response = ""
@@ -36,7 +36,7 @@ def chat_stream(message, history, request: gr.Request):
         with httpx.Client(timeout=60.0) as client:
             with client.stream("POST", API_URL, json=payload, headers=headers) as response:
                 if response.status_code != 200:
-                    yield chat_history + [{"role": "assistant", "content": f"❌ Error {response.status_code}"}]
+                    yield chat_history[:-1] + [(message, f"❌ Error {response.status_code}")]
                     return
                 for line in response.iter_lines():
                     if not line.startswith("data: "):
@@ -50,11 +50,11 @@ def chat_stream(message, history, request: gr.Request):
                         content = delta.get("content", "")
                         if content:
                             full_response += content
-                            yield chat_history + [{"role": "assistant", "content": full_response}]
+                            yield chat_history[:-1] + [(message, full_response)]
                     except (json.JSONDecodeError, IndexError, KeyError):
                         continue
     except Exception as e:
-        yield chat_history + [{"role": "assistant", "content": f"❌ Error: {str(e)}"}]
+        yield chat_history[:-1] + [(message, f"❌ Error: {str(e)}")]
 
 custom_css = """
 @import url('https://fonts.googleapis.com/css2?family=Inter:wght@300;400;500;600;700;800;900&family=JetBrains+Mono:wght@400;500;600&display=swap');
@@ -479,29 +479,7 @@ document.addEventListener('DOMContentLoaded', () => { initTheme(); setupNavbar()
 # ─── App ──────────────────────────────────────────────────────────────────────
 with gr.Blocks(
     fill_height=False,
-    css=custom_css,
-    js=custom_js,
     title="Yuuki-RxG",
-    theme=gr.themes.Base(
-        primary_hue="zinc", neutral_hue="zinc",
-        font=gr.themes.GoogleFont("Inter"), font_mono=gr.themes.GoogleFont("JetBrains Mono"),
-        radius_size=gr.themes.sizes.radius_sm, spacing_size=gr.themes.sizes.spacing_sm,
-    ).set(
-        body_background_fill="#09090b", body_text_color="#fafafa",
-        background_fill_primary="#09090b", background_fill_secondary="#111113",
-        border_color_primary="rgba(255,255,255,0.08)", color_accent="#22c55e",
-        color_accent_soft="rgba(34,197,94,0.12)",
-        button_primary_background_fill="#fafafa", button_primary_text_color="#09090b",
-        button_primary_background_fill_hover="#e4e4e7",
-        button_secondary_background_fill="transparent",
-        button_secondary_border_color="rgba(255,255,255,0.08)",
-        button_secondary_text_color="#a1a1aa",
-        block_border_color="rgba(255,255,255,0.08)", block_background_fill="transparent",
-        block_shadow="none", input_background_fill="rgba(255,255,255,0.04)",
-        input_border_color="rgba(255,255,255,0.08)",
-        input_border_color_focus="rgba(255,255,255,0.25)", input_shadow_focus="none",
-        shadow_drop="none", shadow_drop_lg="none",
-    )
 ) as demo:
     # ─── Hero ─────────────────────────────────────────────────────────────
     gr.HTML("""
@@ -577,7 +555,7 @@ with gr.Blocks(
 
         login_btn = gr.LoginButton("Iniciar sesión con Hugging Face", elem_classes="auth-login-btn")
 
-        chatbot = gr.Chatbot(type="messages", height=480)
+        chatbot = gr.Chatbot(height=480)
         msg = gr.Textbox(placeholder="Escribe tu mensaje...", lines=1)
         clear = gr.Button("Limpiar chat", variant="secondary")
 
@@ -597,7 +575,30 @@ with gr.Blocks(
 
 if __name__ == "__main__":
     try:
-        demo.launch()
+        demo.launch(
+            css=custom_css,
+            js=custom_js,
+            theme=gr.themes.Base(
+                primary_hue="zinc", neutral_hue="zinc",
+                font=gr.themes.GoogleFont("Inter"), font_mono=gr.themes.GoogleFont("JetBrains Mono"),
+                radius_size=gr.themes.sizes.radius_sm, spacing_size=gr.themes.sizes.spacing_sm,
+            ).set(
+                body_background_fill="#09090b", body_text_color="#fafafa",
+                background_fill_primary="#09090b", background_fill_secondary="#111113",
+                border_color_primary="rgba(255,255,255,0.08)", color_accent="#22c55e",
+                color_accent_soft="rgba(34,197,94,0.12)",
+                button_primary_background_fill="#fafafa", button_primary_text_color="#09090b",
+                button_primary_background_fill_hover="#e4e4e7",
+                button_secondary_background_fill="transparent",
+                button_secondary_border_color="rgba(255,255,255,0.08)",
+                button_secondary_text_color="#a1a1aa",
+                block_border_color="rgba(255,255,255,0.08)", block_background_fill="transparent",
+                block_shadow="none", input_background_fill="rgba(255,255,255,0.04)",
+                input_border_color="rgba(255,255,255,0.08)",
+                input_border_color_focus="rgba(255,255,255,0.25)", input_shadow_focus="none",
+                shadow_drop="none", shadow_drop_lg="none",
+            )
+        )
     finally:
         import asyncio
         try: