File size: 2,077 Bytes
ed64d4e 15d0d7e ed64d4e 15d0d7e ed64d4e 15d0d7e ed64d4e 15d0d7e ed64d4e 15d0d7e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 |
import streamlit as st
def handle_prompt_click(prompt_text, key):
st.session_state[f'selected_prompt_{key}'] = prompt_text
st.write(f"Generated prompt for: {prompt_text}")
def main():
st.title("🎨 Art Prompt Generator")
st.markdown("### Select a prompt style to generate artwork:")
# Dictionary mapping prompts to emojis
prompt_emojis = {
"AIart/AIArtistCommunity": "🤖",
"Black & White": "⚫⚪",
"Black & Yellow": "⚫💛",
"Blindfold": "🙈",
"Break": "💔",
"Broken": "🔨",
"Christmas Celebrations art": "🎄",
"Colorful Art": "🎨",
"Crimson art": "🔴",
"Eyes Art": "👁️",
"Going out with Style": "💃",
"Hooded Girl": "🧥",
"Lips": "👄",
"MAEKHLONG": "🏮",
"Mermaid": "🧜♀️",
"Morning Sunshine": "🌅",
"Music Art": "🎵",
"Owl": "🦉",
"Pink": "💗",
"Purple": "💜",
"Rain": "🌧️",
"Red Moon": "🌑",
"Rose": "🌹",
"Snow": "❄️",
"Spacesuit Girl": "👩🚀",
"Steampunk": "⚙️",
"Succubus": "😈",
"Sunlight": "☀️",
"Weird art": "🎭",
"White Hair": "👱♀️",
"Wings art": "👼",
"Woman with Sword": "⚔️"
}
# Create columns for better button layout
col1, col2, col3 = st.columns(3)
# Distribute buttons across columns
for idx, (prompt, emoji) in enumerate(prompt_emojis.items()):
full_prompt = f"QT {prompt}"
col = [col1, col2, col3][idx % 3]
with col:
if st.button(f"{emoji} {prompt}", key=f"btn_{idx}"):
handle_prompt_click(full_prompt, idx)
# Display selected prompt if any
st.markdown("---")
st.markdown("### Generated Prompts:")
for key in st.session_state:
if key.startswith('selected_prompt_'):
st.write(st.session_state[key])
if __name__ == "__main__":
main() |