Update app.py

app.py

@@ -27,6 +27,21 @@ with st.sidebar:
             Enjoy chatting with Meta's Llama3 model!
         """)
 
+    # Example:
+    if st.button("Sample: What is Large Language Model?"):
+        sample_prompt = "What is Large Language Model?"
+    else:
+        sample_prompt = None
+    if st.button("Sample: Explain machine learning on high level terms."):
+        sample_prompt = "Explain machine learning on high level terms."
+    else:
+        sample_prompt = None
+    if st.button("Sample: What is supervised learning? And what data do I need?"):
+        sample_prompt = "What is supervised learning? And what data do I need?"
+    else:
+        sample_prompt = None
+
+
     # Add a button to clear the session state
     if st.button("Clear Session"):
         st.session_state.messages = []
@@ -57,4 +72,20 @@ if prompt := st.chat_input("What is up?"):
     with st.chat_message("assistant"):
         st.markdown(response)
     # Add assistant response to chat history
-    st.session_state.messages.append({"role": "assistant", "content": response})
+    st.session_state.messages.append({"role": "assistant", "content": response})
+elif sample_prompt is not None:
+    prompt = sample_prompt
+    # Display user message in chat message container
+    st.chat_message("user").markdown(prompt)
+    # Add user message to chat history
+    st.session_state.messages.append({"role": "user", "content": prompt})
+
+    response = call_llama(prompt)
+
+    # Display assistant response in chat message container
+    with st.chat_message("assistant"):
+        st.markdown(response)
+    # Add assistant response to chat history
+    st.session_state.messages.append({"role": "assistant", "content": response})
+else:
+    continue