app.py

import gradio as gr
from gradio_client import Client

client = Client("stabilityai/stable-diffusion-3-medium")

def generate_logo(prompt):
  """Generates a logo using the provided prompt."""

  result = client.predict(
      prompt=prompt,
      negative_prompt="",
      seed=0,
      randomize_seed=True,
      width=1024,
      height=1024,
      guidance_scale=7, 
      num_inference_steps=50,
      api_name="/infer"
  )
  return result[0] 

# Define example prompts
examples = [
    "Logo for a restaurant named 'Haandi'",
    "Modern tech company logo named 'InnoTech'",
    "Vintage coffee shop logo for 'Brewed Bliss'",
    "Luxury hotel logo named 'Regal Stay'",
    "Minimalist yoga studio logo called 'Zen Space'",
    "Colorful bakery logo for 'Sweet Treats'",
    "Elegant fashion brand logo for 'Chic Couture'",
    "Rustic farm-to-table restaurant logo 'Harvest Moon'",
    "Bold sports team logo for 'Thunderbolts'",
    "Playful children's toy store logo 'Toy Land'",
]

# Define the Gradio interface
iface = gr.Interface(
    fn=generate_logo,
    inputs=gr.Textbox(label="Describe your logo idea"), 
    outputs=gr.Image(label="Generated Logo"),
    examples=examples
)

iface.launch()