Update app.py

app.py

@@ -8,6 +8,10 @@ import io
 from PIL import Image
 from typing import List, Optional
 
+# ==============================================================
+#  CLASE PRINCIPAL: GENERADOR DE PROMPTS HIPERREALISTAS
+# ==============================================================
+
 class HyperrealisticPromptGenerator:
     def __init__(self):
         self.ROLES = [
@@ -45,11 +49,11 @@ class HyperrealisticPromptGenerator:
             "messy chic bun"
         ]
         self.POSES = [
-            "standing with one leg slightly forward, skirt shifting gently to subtly reveal lace thong, view from knees to head",
-            "seated on a chair edge, legs crossed, skirt moving slightly, natural sensual expression, low angle from knees",
-            "leaning against a desk with hips cocked, skirt riding up, captured from knees to head",
-            "walking with natural sway, skirt flowing, viewed contrapicado from knees",
-            "adjusting stockings or shoes, skirt slightly lifted revealing thong, viewed low angle knees up"
+            "standing with one leg slightly forward, natural elegance",
+            "seated on a chair edge, legs crossed, professional expression",
+            "leaning against a desk, confident look",
+            "walking with subtle grace, light movement",
+            "adjusting hair gently, natural body language"
         ]
         self.SETTINGS = [
             "modern office with elegant decor and warm ambient light",
@@ -67,19 +71,14 @@ class HyperrealisticPromptGenerator:
         ]
         self.TECHNICAL_DETAILS = (
             "Captured in ultra HD 16K (15360×8640) vertical 9:16 full body format. "
-            "Canon EOS R5 Cine RAW camera and Canon RF 85mm f/1.2L USM lens at f/1.2 aperture for creamy bokeh and realistic depth of field. "
-            "ARRI SkyPanel S360-C with soft shadowless 3:1 lighting ratio. "
-            "Advanced Path Tracing, Physically Based Rendering (PBR), Subsurface Scattering (SSS) for lifelike skin translucency, "
-            "Ray Tracing for global illumination and reflections. "
-            "Photogrammetry-based texture mapping, displacement maps for skin pores, delicate fabric weave and lace micro-details. "
-            "Natural, physics-driven hair strand flow. "
-            "Composition uses contrapicado low-angle (knee to head) shots emphasizing natural, sensual lingerie reveal."
+            "Canon EOS R5 Cine RAW camera and Canon RF 85mm f/1.2L USM lens at f/1.2 aperture for creamy bokeh. "
+            "ARRI SkyPanel S360-C soft lighting, Path Tracing, PBR, SSS for lifelike skin, and Ray Tracing. "
+            "Photogrammetry-based textures, displacement maps for skin pores, delicate fabric weave. "
+            "Natural hair strand flow, low-angle (knee to head) composition."
         )
         self.CONDITION_FIXED = (
-            "Wearing elegant thigh-high stockings, no bra, and high stilettos. "
-            "Delicately revealing a lace thong in a natural, seductive manner, as if caught candidly. "
-            "Age between 20 and 25, radiating youthfulness and fresh allure. "
-            "Pose and framing strictly low-angle, knees to head vertical 9:16 aspect ratio, full body filling the frame."
+            "Wearing elegant professional attire matching the role, natural posture, confident expression. "
+            "Full body portrait, cinematic tone, vertical 9:16 framing."
         )
 
     def _choose_random(self, options: List[str]) -> str:
@@ -94,29 +93,26 @@ class HyperrealisticPromptGenerator:
         pose = self._choose_random(self.POSES)
         setting = self._choose_random(self.SETTINGS)
         atmosphere = self._choose_random(self.ATMOSPHERES)
-        return (
-            ```
-            f"Role: {selected_role}\n"
-            f"Age: {age}\n"
-            f"Hair: {hair_style} in {hair_color}\n"
-            f"Eyes: {eye_color}\n"
-            f"Pose: {pose}\n"
-            f"Environment: {setting}\n"
-            f"Atmosphere: {atmosphere}\n"
-            f"Outfit: {self.CONDITION_FIXED}\n"
-            f"Technical specs: {self.TECHNICAL_DETAILS}\n"
-            "```"
-        )
+
+        prompt = f"""
+        """
+        return prompt
 
     def generate_prompt_automatic(self):
         return self.generate_single_prompt()
 
+
+# ==============================================================
+#  CONFIGURACIÓN SAMBANOVA API
+# ==============================================================
+
 gen = HyperrealisticPromptGenerator()
 
 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"}
 
+
 def process_image(image):
     if image is None:
         return None
@@ -124,13 +120,15 @@ def process_image(image):
     image.save(buffered, format="PNG")
     return base64.b64encode(buffered.getvalue()).decode("utf-8")
 
+
 def analizar_imagen_y_generar_prompt(image_base64):
     if not API_KEY:
         return "``````"
     messages = [
         {"role": "system", "content": "Describe images in detailed English."},
         {
-            "role": "user", "content": [
+            "role": "user",
+            "content": [
                 {"type": "image_url", "image_url": {"url": f"data:image/png;base64,{image_base64}"}},
                 {"type": "text", "text": "Provide a detailed English description for a prompt."}
             ]
@@ -142,9 +140,14 @@ def analizar_imagen_y_generar_prompt(image_base64):
         response.raise_for_status()
         text_resp = response.json()["choices"][0]["message"]["content"]
         return f"``````"
-    except Exception as e:
+    except Exception:
         return f"``````"
 
+
+# ==============================================================
+#  FUNCIÓN PRINCIPAL DE CHAT Y PROMPT
+# ==============================================================
+
 def chat_sambanova(user_message, image_input, auto_mode, chat_history, loading_state):
     updated_history = chat_history[:] if chat_history else []
     image_base64 = process_image(image_input) if image_input else None
@@ -197,49 +200,49 @@ def chat_sambanova(user_message, image_input, auto_mode, chat_history, loading_s
                 except json.JSONDecodeError:
                     continue
         yield "", updated_history, "", ""
-    except requests.exceptions.HTTPError as http_err:
-        error_msg = f"Error HTTP {http_err.response.status_code}: {http_err.response.text}"
-        if updated_history:
-            updated_history[-1] = (user_message, error_msg)
-        yield "", updated_history, error_msg, ""
-    except requests.exceptions.ConnectionError:
-        error_msg = "Error: No se pudo conectar con la API de SambaNova. Verifica tu conexión."
-        if updated_history:
-            updated_history[-1] = (user_message, error_msg)
-        yield "", updated_history, error_msg, ""
-    except requests.exceptions.Timeout:
-        error_msg = "Error: La solicitud a la API timed out."
-        if updated_history:
-            updated_history[-1] = (user_message, error_msg)
-        yield "", updated_history, error_msg, ""
     except Exception as e:
         error_msg = f"Error inesperado: {str(e)}"
-        if updated_history:
-            updated_history[-1] = (user_message, error_msg)
+        updated_history[-1] = (user_message, error_msg)
         yield "", updated_history, error_msg, ""
 
+
 def generar_prompt_interno():
     return gen.generate_prompt_automatic(), ""
 
-with gr.Blocks() as demo:
-    gr.Markdown("# Hyperrealistic Prompt Generator & Chatbot")
+
+# ==============================================================
+#  INTERFAZ GRADIO CON MODO OSCURO Y ESTILO BATUTO
+# ==============================================================
+
+css_batuto = """
+body {background-color: #05070A; color: #B0C8FF; font-family: 'Poppins', sans-serif;}
+h1, h2, h3, h4 {color: #5CA8FF; text-align: center;}
+.gradio-container {background-color: #05070A !important;}
+button {background-color: #0B1A33 !important; color: #B0C8FF !important; border-radius: 12px;}
+button:hover {background-color: #1B335F !important;}
+.prompt-output {background-color: #0A0F1A; color: #A8CFFF; border-radius: 10px; padding: 10px;}
+input, textarea {background-color: #0B101A !important; color: #DDE8FF !important;}
+"""
+
+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="Mensajes de error", interactive=False, visible=True)
+    error_display = gr.Textbox(label="System messages", interactive=False, visible=True)
     loading_state = gr.State("")
 
-    chatbot = gr.Chatbot(label="Chatbot IA (SambaNova - Llama-4 Maverick)", type='messages')
-    prompt_output = gr.Markdown(label="Prompt Generado", elem_classes=["prompt-output"])
+    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="Escribe tu mensaje", scale=4)
-        img_input = gr.Image(label="Subir imagen (opcional)", type="pil", scale=2)
+        msg = gr.Textbox(label="Tu mensaje", scale=4)
+        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)
+        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")
+        btn_gen_prompt = gr.Button("🎲 Generar prompt automático", variant="secondary")
+        copy_button = gr.Button("📋 Copiar Prompt")
 
     with gr.Row():
         loading = gr.Markdown(value=lambda x: f"**{x}**" if x else "", label="Estado")
@@ -259,10 +262,21 @@ with gr.Blocks() as demo:
         inputs=[],
         outputs=[msg, prompt_output]
     )
+
+    # ✅ Botón copiar funcional con alerta
     copy_button.click(
-        fn=None,
-        _js="() => { navigator.clipboard.writeText(document.querySelector('.prompt-output').innerText); }",
-        outputs=None
+        None,
+        [],
+        [],
+        _js="""() => { 
+            const el = document.querySelector('.prompt-output'); 
+            if (el) { 
+                navigator.clipboard.writeText(el.innerText); 
+                alert('✅ Prompt copiado al portapapeles');
+            } else { 
+                alert('❌ No se encontró el prompt para copiar');
+            } 
+        }"""
     )
 
 if __name__ == "__main__":
@@ -270,5 +284,4 @@ if __name__ == "__main__":
         demo.launch()
     except Exception as e:
         print(f"Error al iniciar Gradio: {str(e)}")
-