Spaces:
Sleeping
Sleeping
| import random | |
| import gradio as gr | |
| from data import fashion_elements, PhotoReal | |
| negative_prompt_fusion = "blurry, deformed, ugly, mutilated, disfigured, extra limbs, extra fingers, poorly drawn hands, poorly drawn feet, mutated hands, mutation, deformed, bad anatomy, bad proportions, extra arms, extra legs, fused fingers, too many fingers, long neck, username, watermark, signature, overexposed, underexposed, bad lighting, distorted face, low quality, artifacts" | |
| class FashionPromptGenerator: | |
| def __init__(self): | |
| self.previous_prompts = set() | |
| self.photo_real = PhotoReal() | |
| # Asegurarse de que todos los atributos necesarios existan | |
| self._ensure_attributes() | |
| def _ensure_attributes(self): | |
| """Asegura que todos los atributos necesarios existan en photo_real""" | |
| if not hasattr(self.photo_real, 'STOCKINGS'): | |
| self.photo_real.STOCKINGS = "realistic sheer stockings with authentic fabric weave" | |
| if not hasattr(self.photo_real, 'HEELS'): | |
| self.photo_real.HEELS = "classic black leather heels" | |
| if not hasattr(self.photo_real, 'CAMERA'): | |
| self.photo_real.CAMERA = "photorealistic 8K vertical capture 9:16 aspect" | |
| if not hasattr(self.photo_real, 'SENSUAL_POSES'): | |
| self.photo_real.SENSUAL_POSES = [ | |
| {"pose": "standing confidently", "view": "frontal"}, | |
| {"pose": "leaning forward", "view": "frontal"} | |
| ] | |
| def generate_unique_prompts(self, subject_name, num_prompts=5): | |
| """Genera una lista de prompts únicos.""" | |
| if not subject_name or subject_name.isspace(): | |
| subject_name = "a beautiful woman" | |
| prompts = [] | |
| max_attempts = num_prompts * 10 | |
| attempts = 0 | |
| while len(prompts) < num_prompts and attempts < max_attempts: | |
| prompt = self._build_fused_prompt(subject_name) | |
| prompt_hash = hash(prompt) | |
| if prompt_hash not in self.previous_prompts: | |
| self.previous_prompts.add(prompt_hash) | |
| prompts.append(prompt) | |
| attempts += 1 | |
| # Si no se logran suficientes prompts únicos, rellena con algunos aleatorios | |
| while len(prompts) < num_prompts: | |
| prompt = self._build_fused_prompt(subject_name) | |
| prompts.append(prompt) | |
| return prompts | |
| def _build_fused_prompt(self, subject_name): | |
| """Construye un prompt fusionando los estilos photorealistic y voyeuristic.""" | |
| try: | |
| # --- Elementos de Photorealistic --- | |
| role = random.choice(self.photo_real.ROLES) | |
| body = random.choice(self.photo_real.BODY) | |
| skin = random.choice(self.photo_real.SKIN) | |
| hair = random.choice(self.photo_real.HAIR) | |
| expression = random.choice(self.photo_real.EXPRESSION) | |
| sensual_pose = random.choice(self.photo_real.SENSUAL_POSES) | |
| # --- Elementos de Voyeuristic --- | |
| primary_color = random.choice(fashion_elements["colors"]) | |
| discovery_moment = random.choice(fashion_elements["discovery_moments"]) | |
| reveal_action = random.choice(fashion_elements["lingerie_reveal_actions"]) | |
| background = random.choice(fashion_elements["backgrounds"]) | |
| lighting = random.choice(fashion_elements["lighting"]) | |
| # --- Construcción del Prompt --- | |
| prompt = ( | |
| f"Photorealistic portrait of {subject_name} as a {role['role']}, " | |
| f"captured from an extreme low angle, floor-level perspective, shooting upward. " | |
| f"The composition focuses on a moment of discovery, {discovery_moment}. " | |
| f"She is {reveal_action}, creating a natural and unposed exposure.\n\n" | |
| f"**Body & Appearance:**\n" | |
| f"- **Physique:** {body} with {skin}.\n" | |
| f"- **Hair:** {hair}.\n" | |
| f"- **Expression:** {expression}.\n" | |
| f"- **Outfit:** She wears {role['outfit']} over {primary_color} lingerie. " | |
| f"Complemented by {self.photo_real.STOCKINGS} and {self.photo_real.HEELS}.\n\n" | |
| f"**Scene & Atmosphere:**\n" | |
| f"- **Setting:** {background} with {lighting}.\n" | |
| f"- **Pose:** {sensual_pose['pose']} ({sensual_pose['view']} view), emphasizing a sensual and captivating stance.\n" | |
| f"- **Mood:** Intimate, sensual, and elegantly composed.\n\n" | |
| f"**Technical Details:**\n" | |
| f"- **Camera:** {self.photo_real.CAMERA}.\n" | |
| f"- **Quality:** Ultra HD, professional retouching, sharp focus on intimate details and facial expression.\n\n" | |
| f"The image captures a fleeting, intimate moment that feels both spontaneous and beautifully composed, emphasizing the sensual discovery of hidden lingerie through natural movement and expert framing. " | |
| f"--ar 9:16 --v 6 --q 2 --style raw --stylize 100" | |
| ) | |
| return prompt.strip() | |
| except Exception as e: | |
| # Fallback en caso de error | |
| return f"Photorealistic portrait of {subject_name} in an elegant setting, captured from a low angle with intimate lighting. --ar 9:16 --v 6" | |
| def clear_history(self): | |
| """Limpia el historial de prompts generados.""" | |
| self.previous_prompts.clear() | |
| def create_gradio_interface(): | |
| generator = FashionPromptGenerator() | |
| with gr.Blocks() as demo: | |
| gr.Markdown(""" | |
| # 🎭 Advanced Fashion Prompt Generator | |
| *Create photorealistic, discovery-focused prompts with a single, powerful style.* | |
| """) | |
| with gr.Row(): | |
| with gr.Column(scale=2): | |
| celebrity_input = gr.Textbox( | |
| label="Celebrity Name (Optional)", | |
| placeholder="Enter a name or leave blank for 'a beautiful woman'", | |
| value="" | |
| ) | |
| num_prompts_slider = gr.Slider( | |
| minimum=1, maximum=10, value=5, step=1, | |
| label="Number of Prompts to Generate" | |
| ) | |
| with gr.Column(scale=1): | |
| generate_btn = gr.Button("✨ Generate Prompts", variant="primary") | |
| clear_btn = gr.Button("🗑️ Clear History", variant="secondary") | |
| gr.Markdown("---") | |
| with gr.Row(): | |
| with gr.Column(): | |
| gr.Markdown("### Negative Prompt") | |
| negative_prompt_output = gr.Textbox( | |
| value=negative_prompt_fusion, | |
| label="", | |
| lines=4, | |
| interactive=False | |
| ) | |
| gr.Markdown("## Generated Prompts") | |
| output_columns = [] | |
| output_prompts = [] | |
| copy_buttons = [] | |
| # Crear 10 columnas (máximo) | |
| for i in range(10): | |
| with gr.Column(visible=(i < 5)) as col: # Mostrar solo 5 inicialmente | |
| gr.Markdown(f"### 🎯 Prompt {i+1}") | |
| prompt_output = gr.Textbox( | |
| label="", | |
| lines=12, | |
| interactive=False, | |
| show_copy_button=True # Esto agrega el botón de copia integrado | |
| ) | |
| output_columns.append(col) | |
| output_prompts.append(prompt_output) | |
| def generate_and_display(celebrity_name, num_prompts): | |
| num_prompts = int(num_prompts) | |
| prompts = generator.generate_unique_prompts(celebrity_name, num_prompts) | |
| updates = [] | |
| for i in range(10): | |
| if i < num_prompts: | |
| # Mostrar columna y establecer valor | |
| updates.append(gr.update(visible=True)) | |
| updates.append(gr.update(value=prompts[i])) | |
| else: | |
| # Ocultar columna y limpiar valor | |
| updates.append(gr.update(visible=False)) | |
| updates.append(gr.update(value="")) | |
| return updates | |
| # Aplanar la lista de outputs para Gradio | |
| flat_outputs = [] | |
| for i in range(10): | |
| flat_outputs.append(output_columns[i]) | |
| flat_outputs.append(output_prompts[i]) | |
| generate_btn.click( | |
| fn=generate_and_display, | |
| inputs=[celebrity_input, num_prompts_slider], | |
| outputs=flat_outputs | |
| ) | |
| def clear_history_and_outputs(): | |
| generator.clear_history() | |
| updates = [] | |
| for i in range(10): | |
| updates.append(gr.update(visible=(i < 5))) # Mostrar solo 5 por defecto | |
| updates.append(gr.update(value="")) | |
| return updates | |
| clear_btn.click( | |
| fn=clear_history_and_outputs, | |
| inputs=None, | |
| outputs=flat_outputs | |
| ) | |
| return demo | |
| if __name__ == "__main__": | |
| demo = create_gradio_interface() | |
| demo.launch( | |
| share=True, | |
| server_name="0.0.0.0", | |
| server_port=7860 | |
| ) |