| import os |
| import random |
|
|
| import gradio as gr |
|
|
|
|
| def describe_product(image) -> str: |
| if image is None: |
| return "a premium product" |
| img_path = image.get("path") if isinstance(image, dict) else str(image) |
| img_url = image.get("url", "") if isinstance(image, dict) else str(image) |
| try: |
| from gradio_client import Client, handle_file |
| source = img_path if (img_path and os.path.exists(img_path)) else img_url |
| if not source: |
| return "a premium product" |
| client = Client("vikhyatk/moondream2", verbose=False) |
| result = client.predict( |
| handle_file(source), |
| "Describe this product briefly and concisely. What is it?", |
| api_name="/answer_question", |
| ) |
| return str(result).strip() if result else "a premium product" |
| except Exception: |
| return "a premium product" |
|
|
|
|
| def craft_marketing_prompt(caption: str) -> str: |
| if not caption: |
| caption = "a premium product" |
| caption = caption.strip().rstrip(".") |
| styles = [ |
| ( |
| f"Professional product advertisement photograph of {caption}, " |
| "studio lighting, clean white background, commercial photography, " |
| "ultra-sharp, 8K quality" |
| ), |
| ( |
| f"Cinematic product shot of {caption}, dramatic lighting, " |
| "aspirational lifestyle context, premium brand aesthetic, " |
| "shot on Hasselblad, magazine cover quality" |
| ), |
| ( |
| f"Bold marketing campaign visual of {caption}, vibrant colors, " |
| "dynamic composition, modern editorial style, " |
| "award-winning commercial photography" |
| ), |
| ] |
| return random.choice(styles) |
|
|
|
|
| demo = gr.Workflow(bind=[describe_product, craft_marketing_prompt]) |
|
|
| if __name__ == "__main__": |
| demo.launch() |
