Update app.py

app.py

@@ -53,6 +53,9 @@ with st.sidebar:
             Enjoy chatting with Meta's Llama3 model!
         """)
 
+    # Add a button to submit
+    submit_button = st.button("Submit")
+
     # Add a button to clear the session state
     if st.button("Clear Session"):
         st.session_state.messages = []
@@ -74,16 +77,23 @@ for message in st.session_state.messages:
 
 
 # React to user input
-if prompt := st.chat_input("What is up?"):
+if submit_button:
+
+    # Initiatization
+    prompt = "Ask a data science question"
+
     # 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})
 
+    # API call to Llama
     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})
+    st.session_state.messages.append({"role": "assistant", "content": response})