Update app.py

app.py

@@ -433,7 +433,44 @@ def whisper_main():
         transcription = transcribe_audio(filename)
         st.write(transcription)
 
+        # 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 🎵"
+        }
+        
+        # Create columns
+        col1, col2, col3 = st.columns([1, 1, 1], gap="small")
+        
+        # Add buttons to columns
+        if col1.button("Generate Limericks: 😂"):
+            StreamLLMChatResponse(descriptions["Generate Limericks: 😂"] + ' using the context information from this transcript: ' + transcription)
+        # Create columns
+        col1, col2, col3 = st.columns([1, 1, 1], gap="small")
+        # Add buttons to columns
+        if col1.button("Generate Limericks: 😂"):
+            StreamLLMChatResponse(descriptions["Generate Limericks: 😂"] + ' using the context information from this transcript: ' + transcription)
+        if col2.button("Wise Quotes 🧙"):
+            StreamLLMChatResponse(descriptions["Wise Quotes: 🧙"] + ' using the context information from this transcript: ' + transcription)
+        if col3.button("Funny Rhymes: 🎤"):
+            StreamLLMChatResponse(descriptions["Funny Rhymes: 🎤"] + ' using the context information from this transcript: ' + transcription)
+        col4, col5, col6 = st.columns([1, 1, 1], gap="small")
+        if col4.button("Medical Jokes: 💉"):
+            StreamLLMChatResponse(descriptions["Medical Jokes: 💉"] + ' using the context information from this transcript: ' + transcription)
+        if col5.button("Minnesota Humor ❄️"):
+            StreamLLMChatResponse(descriptions["Minnesota Humor: ❄️"] + ' using the context information from this transcript: ' + transcription)
+        if col6.button("Top Funny Stories 📖"):
+            StreamLLMChatResponse(descriptions["Top Funny Stories: 📖"] + ' using the context information from this transcript: ' + transcription)
+        col7 = st.columns(1, gap="small")
+        if col7[0].button("More Funny Rhymes: 🎙️"):
+            StreamLLMChatResponse(descriptions["More Funny Rhymes: 🎙️"] + ' using the context information from this transcript: ' + transcription)
 
+            
 def main():
 
     st.title("AI Drome Llama")