Update app.py

app.py

@@ -54,19 +54,17 @@ def format_llama_messages(messages):
 
 if "messages" not in st.session_state:
     st.session_state.messages = [{"role": "system", "content": system_prompt}]
-    print(1, st.session_state.messages)
 
 def handle_query(query):
     st.session_state.messages.append({'role': 'user', "content": query})
     formatted_messages=format_llama_messages(st.session_state.messages)
-    print(2, st.session_state.messages)
-    print(23, formatted_messages)
+
     json_body = {
         "inputs": formatted_messages,
                 "parameters": {"max_new_tokens":max_new_tokens, "top_p":top_p, "temperature":temperature}
         }
     data = json.dumps(json_body)
-    print(3, data)
+    print("JSON Sent:", data)
     response = requests.request("POST", API_URL, headers=headers, data=data)
     try:
         data=json.loads(response.content.decode("utf-8"))
@@ -96,11 +94,11 @@ user_prompt = st.chat_input("Ask me anything about foods..")
 if user_prompt:
     with st.chat_message("user", avatar=icons["user"]):
         st.write(user_prompt)
-    print(4, user_prompt)
+
     # Trigger assistant's response retrieval and update UI
     with st.spinner("Thinking..."):
         response = handle_query(user_prompt)
     with st.chat_message("assistant", avatar=icons["assistant"]):
         st.write(streamer(response))
-    print(5, response)
+
     st.session_state.messages.append({'role': 'assistant', "content": response})