Update app.py

app.py

@@ -4,10 +4,6 @@ import os
 import json
 import time
 
-# Set up the Streamlit app
-st.title("Llama Food Critic Chatbot")
-st.markdown("Ask me anything about food")
-
 # Get the API key from the environment variable
 HF_TOKEN = os.getenv("HF_TOKEN")
 
@@ -27,16 +23,6 @@ headers = {
 }
 
 
-st.sidebar.markdown("Model Parameters")
-temperature = st.sidebar.slider("Temperature", 0.0, 1.0, 0.7, 0.1)
-max_new_tokens = st.sidebar.number_input("Max New Tokens", 50, 500, 256, step=50)
-top_p = st.sidebar.slider("Top P", 0.1, 1.0, 0.9, 0.1)
-system_prompt = st.sidebar.text_area(
-    "System Prompt",
-    "You are a food critic Chatbot. Reply in the style of a New York Times food critic",
-    )
-
-
 def format_llama_messages(messages):
     """Format messages for Llama-3 chat models.
     """
@@ -52,9 +38,6 @@ def format_llama_messages(messages):
 
     return "".join(prompt)
 
-if "messages" not in st.session_state:
-    st.session_state.messages = [{"role": "system", "content": system_prompt}]
-
 def handle_query(query):
     st.session_state.messages.append({'role': 'user', "content": query})
     formatted_messages=format_llama_messages(st.session_state.messages)
@@ -79,11 +62,31 @@ def handle_query(query):
     except:
         return response
 
-
+def prompt_changed(new_prompt):
+    st.session_state.messages[0]["content"]=new_prompt
+    
 def streamer(text):
     for i in text:
         yield i
         time.sleep(0.005)
+        
+# Set up the Streamlit app
+st.title("Llama Food Critic Chatbot")
+st.markdown("Ask me anything about food")
+
+st.sidebar.markdown("Model Parameters")
+temperature = st.sidebar.slider("Temperature", 0.0, 1.0, 0.7, 0.1)
+max_new_tokens = st.sidebar.number_input("Max New Tokens", 50, 500, 256, step=50)
+top_p = st.sidebar.slider("Top P", 0.1, 1.0, 0.9, 0.1)
+
+if "messages" not in st.session_state:
+    st.session_state.messages = [{"role": "system", "content": system_prompt}]
+
+system_prompt = st.sidebar.text_area(
+    "System Prompt",
+    "You are a food critic Chatbot. Reply in the style of a New York Times food critic",
+    key="system_prompt", on_change=prompt_changed, args=("system_prompt")
+    )
 
 for message in st.session_state.messages[1:]:
     with st.chat_message(message['role'], avatar=icons[message['role']]):