Upload 29 files

ui_logic.py

@@ -101,32 +101,36 @@ def main_chat(
         yield app_state, history, user_text, files
         return
 
-    # Add user message to history so the user sees it immediately
+    # Add user message + placeholder assistant so the UI updates at once
     history_messages.append({"role": "user", "content": user_text})
+    assistant_message = {"role": "assistant", "content": "⏳ Preparando respuesta..."}
+    history_messages.append(assistant_message)
+    yield app_state, _messages_to_pairs(history_messages), "", files
 
     model_ready = model in downloaded_models or is_model_available(model)
     if not model_ready:
         gr.Info(f"El modelo '{model}' no está disponible localmente. Intentando descargarlo...")
-        download_msg = {"role": "assistant", "content": f"📥 Descargando modelo '{model}'..."}
-        history_messages.append(download_msg)
+        assistant_message["content"] = f"📥 Descargando modelo '{model}'..."
         yield app_state, _messages_to_pairs(history_messages), "", files
 
-        pull_status = ""
         pull_success = False
         for status in pull_model_with_progress(model):
-            pull_status = status
-            download_msg["content"] = status
+            assistant_message["content"] = status
             yield app_state, _messages_to_pairs(history_messages), "", files
             if status.startswith("✅"):
                 pull_success = True
 
         if not pull_success:
             gr.Error(f"No se pudo descargar el modelo '{model}'. Por favor, verifica el nombre o hazlo manualmente.")
+            assistant_message["content"] = f"⚠️ No se pudo descargar '{model}'."
+            yield app_state, _messages_to_pairs(history_messages), user_text, files
             return
 
         if model not in downloaded_models:
             downloaded_models.append(model)
         gr.Info(f"Modelo '{model}' descargado con éxito.")
+        assistant_message["content"] = "⏳ Preparando respuesta..."
+        yield app_state, _messages_to_pairs(history_messages), "", files
 
     # Prepare inputs
     files_blob, preview, _ = read_uploaded_files(files, "")
@@ -139,8 +143,7 @@ def main_chat(
     )
     
     # Stream response
-    assistant_message = {"role": "assistant", "content": ""}
-    history_messages.append(assistant_message)
+    assistant_message["content"] = ""
     
     # The history sent to the model should not include the latest empty assistant message
     model_history_pairs = _messages_to_pairs(history_messages[:-1])