Update app.py

app.py

@@ -346,14 +346,22 @@ def get_zip_download_link(zip_file):
     return href
 
 def main():
+
+
     st.title(" DrLlama7B")
     prompt = f"Write ten funny jokes that are tweet length stories that make you laugh.  Show as markdown outline with emojis for each."
+    
+    # Add Wit and Humor buttons
+    add_witty_humor_buttons()
+    
     example_input = st.text_input("Enter your example text:", value=prompt)
     if st.button("Run Prompt With Dr Llama"):
         try:
             StreamLLMChatResponse(example_input)
         except:
             st.write('Dr. Llama is asleep.  Starting up now on A10 - please give 5 minutes then retry as KEDA scales up from zero to activate running container(s).')
+
+    
     openai.api_key = os.getenv('OPENAI_KEY')
     menu = ["txt", "htm", "xlsx", "csv", "md", "py"]
     choice = st.sidebar.selectbox("Output File Type:", menu)