Update app.py

app.py

@@ -108,7 +108,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"}
 
@@ -123,18 +124,20 @@ def process_image(image):
 
 def analizar_imagen_y_generar_prompt(image_base64):
     if not API_KEY:
-        return "⚠️ SAMBANOVA_API_KEY not configured."
+        return "```\n⚠️ SAMBANOVA_API_KEY no configurada.\n```"
+    
+    # Formato de contenido para el análisis de imagen (puede variar)
     messages = [
         {"role": "system", "content": "Describe images in detailed English."},
         {
             "role": "user",
-            "content": {
-                "type": "image",
-                "image_data": f"data:image/png;base64,{image_base64}",
-                "text": "Provide a detailed English description suitable for a prompt."
-            }
+            "content": [
+                {"type": "image_url", "image_url": {"url": f"data:image/png;base64,{image_base64}"}},
+                {"type": "text", "text": "Provide a detailed English description suitable for a prompt."}
+            ]
         }
     ]
+    
     json_data = {"model": "Llama-4-Maverick-17B-128E-Instruct", "messages": messages, "stream": False}
     try:
         response = requests.post(API_URL, headers=headers, json=json_data)
@@ -142,40 +145,46 @@ def analizar_imagen_y_generar_prompt(image_base64):
         text_resp = response.json()["choices"][0]["message"]["content"]
         return f"```\n{text_resp}\n```"
     except Exception as e:
-        return f"Error analyzing image: {str(e)}"
+        return f"```\nError analizando imagen: {str(e)}\n```"
 
 
 # ==============================================================
-#  CHAT / PROMPT FUNCTION - CORREGIDA
+#  FUNCIÓN PRINCIPAL DE CHAT Y PROMPT (CORREGIDA)
 # ==============================================================
 
 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
+    
+    # 1. Yield inicial para mostrar "Procesando..." y borrar la entrada del usuario
+    yield "", updated_history, "", "Procesando..."
 
     if not API_KEY:
         error_msg = "Error: SAMBANOVA_API_KEY no configurada."
         updated_history.append((user_message, error_msg))
-        yield "", updated_history, error_msg, ""
+        # 2. Yield de error
+        yield user_message, updated_history, error_msg, ""
         return
 
     if auto_mode and image_base64:
         prompt = analizar_imagen_y_generar_prompt(image_base64)
         updated_history.append((user_message or "Análisis automático (Imagen)", f"IA - Prompt generado:\n{prompt}"))
+        # 3. Yield de finalización de análisis
         yield "", updated_history, "", ""
         return
 
+    # Preparar el historial de chat para la API
     messages = [{"role": "system", "content": "Eres un asistente útil"}]
     for user_msg, ai_msg in updated_history:
-        messages.append({"role": "user", "content": user_msg})
+        # Los mensajes del historial deben ser solo de texto para el rol 'user'
+        # Esto previene errores de formato en mensajes pasados.
+        messages.append({"role": "user", "content": [{"type": "text", "text": user_msg}]}) 
         messages.append({"role": "assistant", "content": ai_msg})
 
-    user_content = {"type": "text", "text": user_message}
+    # Preparar el mensaje actual (puede ser texto o multimodal)
+    user_content = [{"type": "text", "text": user_message}]
     if image_base64:
-        user_content = [
-            {"type": "text", "text": user_message},
-            {"type": "image", "image_data": f"data:image/png;base64,{image_base64}"}
-        ]
+        user_content.append({"type": "image_url", "image_url": {"url": f"data:image/png;base64,{image_base64}"}})
     messages.append({"role": "user", "content": user_content})
 
     json_data = {"model": "Llama-4-Maverick-17B-128E-Instruct", "messages": messages, "stream": True}
@@ -184,7 +193,8 @@ def chat_sambanova(user_message, image_input, auto_mode, chat_history):
         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, "")) 
+
         for line in response.iter_lines(decode_unicode=True):
             if line.startswith("data: "):
                 json_str = line[len("data: "):]
@@ -196,9 +206,11 @@ def chat_sambanova(user_message, image_input, auto_mode, chat_history):
                     text_fragment = delta.get("content", "")
                     collected_text += text_fragment
                     updated_history[-1] = (user_message, collected_text)
+                    # 4. Yield de streaming
                     yield "", updated_history, "", "Procesando..."
                 except json.JSONDecodeError:
                     continue
+        # 5. Yield final
         yield "", updated_history, "", ""
     except Exception as e:
         error_msg = f"Error inesperado de la API: {str(e)}"
@@ -206,15 +218,16 @@ def chat_sambanova(user_message, image_input, auto_mode, chat_history):
             updated_history[-1] = (user_message, error_msg)
         else:
             updated_history.append((user_message, error_msg))
-        yield "", updated_history, error_msg, ""
-
+        # 6. Yield de error final
+        yield user_message, updated_history, error_msg, ""
 
 def generar_prompt_interno():
-    prompt, _ = gen.generate_prompt_automatic()
+    prompt, _ = gen.generate_single_prompt(), ""
+    # Retorna el prompt como texto de salida y una cadena vacía para el display de error
     return prompt, ""
 
 # ==============================================================
-#  INTERFAZ GRADIO - TEMA OSCURO BATUTO - CORREGIDA
+#  INTERFAZ GRADIO (CORREGIDA)
 # ==============================================================
 
 css_batuto = """
@@ -227,25 +240,18 @@ button:hover {background-color: #1B335F !important;}
 input, textarea {background-color: #0B101A !important; color: #DDE8FF !important;}
 """
 
-# Verificar que Gradio esté disponible
-try:
-    import gradio as gr
-    print("✅ Gradio importado correctamente")
-except ImportError as e:
-    print(f"❌ Error importando Gradio: {e}")
-    print("Instala Gradio con: pip install gradio")
-    exit(1)
-
 with gr.Blocks(css=css_batuto, theme=gr.themes.Soft()) as demo:
     gr.Markdown("# ⚡ BATUTO / Prompt Studio — Hyperrealistic Generator")
 
     chat_history = gr.State([])
     error_display = gr.Textbox(label="System messages", value="", visible=True, interactive=False)
-
+    
     chatbot = gr.Chatbot(label="💬 BATUTO Assistant (SambaNova - Llama-4 Maverick)", type='messages')
-    prompt_output = gr.Textbox(label="🎨 Prompt generado", elem_classes=["prompt-output"], lines=5, max_lines=10)
+    # Cambiado a Textbox para mejor funcionalidad de copia.
+    prompt_output = gr.Textbox(label="🎨 Prompt generado", elem_classes=["prompt-output"], lines=5, max_lines=10) 
 
     with gr.Row():
+        # La salida msg se usa para limpiar la caja de texto después del envío
         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)
 
@@ -255,9 +261,10 @@ with gr.Blocks(css=css_batuto, theme=gr.themes.Soft()) as demo:
         btn_gen_prompt = gr.Button("🎲 Generar prompt automático", variant="secondary")
         copy_button = gr.Button("📋 Copiar Prompt")
 
-    loading_state = gr.Textbox(value="", label="Estado", visible=False)
+    # Componente para mostrar el estado de carga (Procesando...)
+    loading_state = gr.Textbox(value="", label="Estado", interactive=False) 
 
-    # CORRECCIÓN PRINCIPAL: Ajustar las salidas para que coincidan
+    # Asignación de outputs corregida para coincidir con la función chat_sambanova
     btn_send.click(
         fn=chat_sambanova,
         inputs=[msg, img_input, auto_mode, chat_history],
@@ -270,35 +277,22 @@ with gr.Blocks(css=css_batuto, theme=gr.themes.Soft()) as demo:
         outputs=[msg, chatbot, error_display, loading_state]
     )
     
+    # La salida del prompt va al Textbox y la segunda salida es para limpiar errores
     btn_gen_prompt.click(
         fn=generar_prompt_interno,
         inputs=[],
         outputs=[prompt_output, error_display]
     )
 
+    # CORRECCIÓN DE ERROR: Cambiado _js a js
     copy_button.click(
         None,
         [],
         [],
         js="""() => { 
-            const textareas = document.querySelectorAll('textarea');
-            let promptText = '';
-            
-            // Buscar el textarea del prompt
-            for (let textarea of textareas) {
-                if (textarea.value && textarea.value.includes('Highly detailed hyperrealistic')) {
-                    promptText = textarea.value;
-                    break;
-                }
-            }
-            
-            if (!promptText) {
-                // Si no se encuentra, buscar en los elementos de texto
-                const promptElements = document.querySelectorAll('.prompt-output');
-                if (promptElements.length > 0) {
-                    promptText = promptElements[0].value || promptElements[0].textContent || promptElements[0].innerText;
-                }
-            }
+            // Usa el ID o selector de clase del Textbox
+            const promptBox = document.querySelector('.prompt-output textarea'); 
+            const promptText = promptBox ? promptBox.value : '';
             
             if (promptText) {
                 navigator.clipboard.writeText(promptText).then(() => {
@@ -307,14 +301,14 @@ with gr.Blocks(css=css_batuto, theme=gr.themes.Soft()) as demo:
                     alert('❌ Error al copiar: ' + err);
                 });
             } else {
-                alert('❌ No se encontró el prompt para copiar');
+                alert('❌ No se encontró el prompt para copiar. Genera uno primero.');
             }
         }"""
     )
 
 if __name__ == "__main__":
     try:
-        print("🚀 Iniciando BATUTO Prompt Studio...")
-        demo.launch(server_name="0.0.0.0", server_port=7860, share=False)
+        demo.launch()
     except Exception as e:
-        print(f"❌ Error al iniciar Gradio: {str(e)}")
+        print(f"Error al iniciar Gradio: {str(e)}")
+