Update app.py

app.py

@@ -4,6 +4,16 @@ import google.generativeai as genai
 # Streamlit app layout
 st.title('PromptLab')
 
+# Create two columns for the Shinobi and Raikage buttons
+col1, col2 = st.columns(2)
+
+# Style the buttons to look like rounded slots
+with col1:
+    shinobi_selected = st.button("🌀 Shinobi")
+
+with col2:
+    raikage_selected = st.button("⚡ Raikage")
+
 # Retrieve the API key from Streamlit secrets
 GOOGLE_API_KEY = st.secrets["GEMINI_API_KEY"]
 
@@ -11,4 +21,4 @@ GOOGLE_API_KEY = st.secrets["GEMINI_API_KEY"]
 genai.configure(api_key=GOOGLE_API_KEY)
 
 # Input field for the blog topic
-topic = st.text_area('Enter your prompt: ') #placeholder='E.g., The Future of Artificial Intelligence'
+topic = st.text_area('Enter your prompt:')