Update app.py

app.py

@@ -112,7 +112,8 @@ She is {pose}, in a {setting}. {atmosphere}.
 
 gen = HyperrealisticPromptGenerator()
 
-API_KEY = os.getenv("SAMBANOVA_API_KEY")
+# Asegúrate de configurar SAMBANOVA_API_KEY en tu entorno
+API_KEY = os.getenv("SAMBANOVA_API_KEY") 
 API_URL = "https://api.sambanova.ai/v1/chat/completions"
 headers = {"Authorization": f"Bearer {API_KEY}", "Content-Type": "application/json"}
 
@@ -127,7 +128,7 @@ def process_image(image):
 
 def analizar_imagen_y_generar_prompt(image_base64):
     if not API_KEY:
-        return "``````"
+        return "```\n[Error: SAMBANOVA_API_KEY no configurada]\n```"
     messages = [
         {"role": "system", "content": "Describe images in detailed English."},
         {
@@ -144,36 +145,40 @@ def analizar_imagen_y_generar_prompt(image_base64):
         response.raise_for_status()
         text_resp = response.json()["choices"][0]["message"]["content"]
         return f"```\n{text_resp}\n```"
-    except Exception:
-        return f"```\n[Error generating prompt]\n```"
+    except Exception as e:
+        return f"```\n[Error generating prompt: {str(e)}]\n```"
 
 
 # ==============================================================
 #  FUNCIÓN PRINCIPAL DE CHAT Y PROMPT
 # ==============================================================
 
-def chat_sambanova(user_message, image_input, auto_mode, chat_history, loading_state):
+def chat_sambanova(user_message, image_input, auto_mode, chat_history):
     updated_history = chat_history[:] if chat_history else []
     image_base64 = process_image(image_input) if image_input else None
 
-    loading_state = "Procesando..."
-    yield "", updated_history, "", loading_state
+    # Inicializa el estado de carga y borra el mensaje de entrada
+    yield "", updated_history, "", "Procesando..."
 
     if not API_KEY:
         error_msg = "Error: SAMBANOVA_API_KEY no configurada."
         updated_history.append((user_message, error_msg))
+        # Devuelve el mensaje de error y limpia el estado de carga
         yield "", updated_history, error_msg, ""
         return
 
+    # Si está en modo automático y hay imagen, analiza la imagen
     if auto_mode and image_base64:
         prompt = analizar_imagen_y_generar_prompt(image_base64)
-        updated_history.append((user_message or "Análisis automático", f"IA - Prompt generado:\n{prompt}"))
-        yield "", updated_history, "", ""
+        updated_history.append((user_message or "Análisis automático (Imagen)", f"IA - Prompt generado:\n{prompt}"))
+        yield "", updated_history, "", "" # Limpia el estado de carga al terminar
         return
 
+    # Si es un chat normal, prepara los mensajes para la API
     messages = [{"role": "system", "content": "Eres un asistente útil"}]
     for user_msg, ai_msg in updated_history:
-        messages.append({"role": "user", "content": [{"type": "text", "text": user_msg}]})
+        # Asegúrate de que el contenido del usuario en el historial anterior sea solo texto si no se manejan imágenes correctamente
+        messages.append({"role": "user", "content": user_msg}) 
         messages.append({"role": "assistant", "content": ai_msg})
 
     user_content = [{"type": "text", "text": user_message}]
@@ -187,7 +192,7 @@ def chat_sambanova(user_message, image_input, auto_mode, chat_history, loading_s
         response = requests.post(API_URL, headers=headers, json=json_data, stream=True)
         response.raise_for_status()
         collected_text = ""
-        updated_history.append((user_message, ""))
+        updated_history.append((user_message, "")) # Añade una entrada vacía para ir actualizando
 
         for line in response.iter_lines(decode_unicode=True):
             if line.startswith("data: "):
@@ -200,14 +205,15 @@ def chat_sambanova(user_message, image_input, auto_mode, chat_history, loading_s
                     text_fragment = delta.get("content", "")
                     collected_text += text_fragment
                     updated_history[-1] = (user_message, collected_text)
-                    yield "", updated_history, "", "Procesando..."
+                    # No actualices el estado de carga aquí, déjalo en "Procesando..." hasta el final
+                    yield "", updated_history, "", "Procesando..." 
                 except json.JSONDecodeError:
                     continue
-        yield "", updated_history, "", ""
+        yield "", updated_history, "", "" # Limpia el estado de carga al terminar
     except Exception as e:
-        error_msg = f"Error inesperado: {str(e)}"
+        error_msg = f"Error inesperado de la API: {str(e)}"
         updated_history[-1] = (user_message, error_msg)
-        yield "", updated_history, error_msg, ""
+        yield "", updated_history, error_msg, "" # Muestra el error y limpia el estado de carga
 
 
 def generar_prompt_interno():
@@ -232,49 +238,54 @@ with gr.Blocks(css=css_batuto, theme="gradio/soft") as demo:
     gr.Markdown("# ⚡ BATUTO / Prompt Studio — Hyperrealistic Generator")
 
     chat_history = gr.State([])
-    error_display = gr.Textbox(label="System messages", interactive=False, visible=True)
-    loading_state = gr.State("")
-
+    # Cambié error_display a gr.Markdown para mejor visualización del estado
+    error_display = gr.Markdown(label="System messages", value="", visible=True) 
+    
     chatbot = gr.Chatbot(label="💬 BATUTO Assistant (SambaNova - Llama-4 Maverick)", type='messages')
     prompt_output = gr.Markdown(label="🎨 Prompt generado", elem_classes=["prompt-output"])
 
     with gr.Row():
-        msg = gr.Textbox(label="Tu mensaje", scale=4)
+        msg = gr.Textbox(label="Tu mensaje", scale=4, placeholder="Escribe tu mensaje o usa el modo automático...")
         img_input = gr.Image(label="Sube una imagen (opcional)", type="pil", scale=2)
 
     with gr.Row():
-        auto_mode = gr.Checkbox(label="Modo automático (Generar prompt desde imagen)", value=False)
+        # Ahora auto_mode solo devuelve el estado
+        auto_mode = gr.Checkbox(label="Modo automático (Generar prompt desde imagen)", value=False) 
         btn_send = gr.Button("Enviar mensaje", variant="primary")
         btn_gen_prompt = gr.Button("🎲 Generar prompt automático", variant="secondary")
         copy_button = gr.Button("📋 Copiar Prompt")
 
-    with gr.Row():
-        loading = gr.Markdown(value="", label="Estado")
+    # Mantiene el estado de carga visible
+    loading_state = gr.Markdown(value="", label="Estado")
 
+    # Modificado para usar loading_state y error_display
     btn_send.click(
         fn=chat_sambanova,
-        inputs=[msg, img_input, auto_mode, chat_history, loading_state],
-        outputs=[msg, chatbot, chat_history, error_display, loading]
+        inputs=[msg, img_input, auto_mode, chat_history], # Removido loading_state de inputs
+        outputs=[msg, chatbot, chat_history, error_display, loading_state] # Usando loading_state como salida
     )
     msg.submit(
         fn=chat_sambanova,
-        inputs=[msg, img_input, auto_mode, chat_history, loading_state],
-        outputs=[msg, chatbot, chat_history, error_display, loading]
+        inputs=[msg, img_input, auto_mode, chat_history], # Removido loading_state de inputs
+        outputs=[msg, chatbot, chat_history, error_display, loading_state] # Usando loading_state como salida
     )
     btn_gen_prompt.click(
         fn=generar_prompt_interno,
         inputs=[],
-        outputs=[msg, prompt_output]
+        outputs=[prompt_output, error_display] # prompt_output es ahora la primera salida
     )
 
+    # CORRECCIÓN: Se cambió el argumento _js a js para compatibilidad con Gradio.
     copy_button.click(
         None,
         [],
         [],
-        _js="""() => { 
+        js="""() => { 
             const el = document.querySelector('.prompt-output'); 
             if (el) { 
-                navigator.clipboard.writeText(el.innerText); 
+                // Intenta obtener el contenido de la etiqueta <p> dentro del Markdown para evitar copiar la etiqueta
+                const promptText = el.innerText.replace('🎨 Prompt generado', '').trim();
+                navigator.clipboard.writeText(promptText); 
                 alert('✅ Prompt copiado al portapapeles');
             } else { 
                 alert('❌ No se encontró el prompt para copiar');