Update app.py

app.py

@@ -124,10 +124,11 @@ def get_response(system_message, chat_history, user_text, max_new_tokens=512):
     filtered_history = "\n".join(f"{msg['role']}: {msg['content']}" for msg in chat_history)
 
     prompt = PromptTemplate.from_template(
-        "[INST] {system_message}\n\nCurrent Conversation:\n{chat_history}\n\n"
+        "[INST] You are a helpful AI assistant.\n\nCurrent Conversation:\n{chat_history}\n\n"
         "User: {user_text}.\n [/INST]\n"
-        "AI: Provide a detailed explanation. Use a conversational tone. "
+        "AI: Provide a detailed explanation with depth. Use a conversational tone. "
         "🚨 Answer **only in English**."
+        "Ensure a friendly, engaging tone."
         "\nHAL:"
     )
 
@@ -142,13 +143,14 @@ def get_response(system_message, chat_history, user_text, max_new_tokens=512):
 
     follow_up = generate_follow_up(user_text)
 
-    chat_history.extend([
-        {'role': 'user', 'content': user_text},
-        {'role': 'assistant', 'content': response},
-        {'role': 'assistant', 'content': follow_up}
-    ])
+     # ✅ Preserve conversation history
+    st.session_state.chat_history.append({'role': 'user', 'content': user_text})
+    st.session_state.chat_history.append({'role': 'assistant', 'content': response})
+    st.session_state.chat_history.append({'role': 'assistant', 'content': follow_up})
 
-    return response, follow_up, chat_history, None
+    return response, follow_up
+
+    #return response, follow_up, chat_history, None
 
 # ✅ Streamlit UI
 st.title("🚀 HAL - NASA AI Assistant")