Update app.py

app.py

@@ -40,11 +40,8 @@ def get_ai_response(prompt, chat_history):
     st.session_state.is_streaming = True
     st.session_state.response = ""
 
-    chat_history.insert(0, {"role": "system", "content": system_prompt})
-    chat_history.append({"role": "user", "content": prompt})
-
-    system_tokens = num_tokens_from_string(system_message["content"])
-    user_tokens = num_tokens_from_string(user_message["content"])
+    system_tokens = num_tokens_from_string(system_prompt)
+    user_tokens = num_tokens_from_string(prompt)
 
     # Maximum allowed tokens
     context_length = 4096 - max_tokens
@@ -63,7 +60,7 @@ def get_ai_response(prompt, chat_history):
             break
 
     # Construct final message list
-    final_messages = [system_message] + trimmed_history + [user_message]
+    final_messages = [{"role": "system", "content": system_prompt}] + trimmed_history + [{"role": "user", "content": prompt}]
 
     try:
         with st.chat_message("assistant", avatar=st.session_state.assistant_avatar):