Update app.py

app.py

@@ -50,40 +50,46 @@ should_save = st.sidebar.checkbox("💾 Save", value=True)
 def add_witty_humor_buttons():
     with st.expander("Wit and Humor 🤣", expanded=True):
         
-        # Button for Generating Limericks
-        limerick_description = "Write ten random adult limericks based on quotes that are tweet length and make you laugh 🎭"
-        if st.button("Generate Limericks 😂"):
-            StreamLLMChatResponse(limerick_description)
+        # Define button descriptions
+        descriptions = {
+            "Generate Limericks 😂": "Write ten random adult limericks based on quotes that are tweet length and make you laugh 🎭",
+            "Wise Quotes 🧙": "Generate ten wise quotes that are tweet length 🦉",
+            "Funny Rhymes 🎤": "Create ten funny rhymes that are tweet length 🎶",
+            "Medical Jokes 💉": "Create ten medical jokes that are tweet length 🏥",
+            "Minnesota Humor ❄️": "Create ten jokes about Minnesota that are tweet length 🌨️",
+            "Top Funny Stories 📖": "Create ten funny stories that are tweet length 📚",
+            "More Funny Rhymes 🎙️": "Create ten more funny rhymes that are tweet length 🎵"
+        }
         
-        # Button for Wise Quotes
-        wise_description = "Generate ten wise quotes that are tweet length 🦉"
-        if st.button("Wise Quotes 🧙"):
-            StreamLLMChatResponse(wise_description)
+        # Create columns
+        col1, col2, col3 = st.columns([1, 1, 1], gap="small")
         
-        # Button for Funny Rhymes
-        rhyme_description = "Create ten funny rhymes that are tweet length 🎶"
-        if st.button("Funny Rhymes 🎤"):
-            StreamLLMChatResponse(rhyme_description)
+        # Add buttons to columns
+        if col1.button("Generate Limericks 😂"):
+            StreamLLMChatResponse(descriptions["Generate Limericks 😂"])
         
-        # Button for Medical Jokes
-        medical_description = "Create ten medical jokes that are tweet length 🏥"
-        if st.button("Medical Jokes 💉"):
-            StreamLLMChatResponse(medical_description)
+        if col2.button("Wise Quotes 🧙"):
+            StreamLLMChatResponse(descriptions["Wise Quotes 🧙"])
         
-        # Button for Minnesota Humor
-        minnesota_description = "Create ten jokes about Minnesota that are tweet length 🌨️"
-        if st.button("Minnesota Humor ❄️"):
-            StreamLLMChatResponse(minnesota_description)
+        if col3.button("Funny Rhymes 🎤"):
+            StreamLLMChatResponse(descriptions["Funny Rhymes 🎤"])
         
-        # Button for Top Funny Stories
-        funny_stories_description = "Create ten funny stories that are tweet length 📚"
-        if st.button("Top Funny Stories 📖"):
-            StreamLLMChatResponse(funny_stories_description)
+        col4, col5, col6 = st.columns([1, 1, 1], gap="small")
         
-        # Button for Funny Rhymes (another one)
-        funny_rhymes_description = "Create ten more funny rhymes that are tweet length 🎵"
-        if st.button("More Funny Rhymes 🎙️"):
-            StreamLLMChatResponse(funny_rhymes_description)
+        if col4.button("Medical Jokes 💉"):
+            StreamLLMChatResponse(descriptions["Medical Jokes 💉"])
+        
+        if col5.button("Minnesota Humor ❄️"):
+            StreamLLMChatResponse(descriptions["Minnesota Humor ❄️"])
+        
+        if col6.button("Top Funny Stories 📖"):
+            StreamLLMChatResponse(descriptions["Top Funny Stories 📖"])
+        
+        col7 = st.columns(1, gap="small")
+        
+        if col7[0].button("More Funny Rhymes 🎙️"):
+            StreamLLMChatResponse(descriptions["More Funny Rhymes 🎙️"])
+
 
 # Function to Stream Inference Client for Inference Endpoint Responses
 def StreamLLMChatResponse(prompt):