Wajahat698 commited on
Commit
6e3eb7c
·
verified ·
1 Parent(s): 53a8010

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -13
app.py CHANGED
@@ -1567,7 +1567,7 @@ def retrieve_user_data(user_id):
1567
 
1568
  def handle_memory_queries(prompt):
1569
  prompt = prompt.lower().strip()
1570
-
1571
  # Save as TrustBuilder
1572
  if re.search(r"\b(keep|add|save|store)\s*(this)?\s*(as)?\s*(trust\s*builder|trustbuilder)\b", prompt, re.IGNORECASE):
1573
  content_to_save = re.sub(r"\b(keep|add|save|store)\s*(this)?\s*(as)?\s*(trust\s*builder|trustbuilder)\b", "", prompt, flags=re.IGNORECASE).strip()
@@ -1591,23 +1591,13 @@ def handle_memory_queries(prompt):
1591
  elif "show my saved trustbuilders" in prompt or "find my saved trustbuilders" in prompt:
1592
  trustbuilders = st.session_state.get("TrustBuilder", {})
1593
  if trustbuilders:
1594
- # Return only the content messages without any IDs or extra details
1595
  response = "\n".join([f"- {entry['message']}" for entry in trustbuilders.values()])
1596
  return f"Here are your saved TrustBuilders:\n{response}"
1597
  else:
1598
  return "You haven't saved any TrustBuilders yet."
1599
 
1600
- # Retrieve saved Brand Tonality
1601
- elif "show my saved brand tonality" in prompt or "find my saved brand tonality" in prompt:
1602
- brand_tonality = st.session_state.get("BrandTonality", {})
1603
- if brand_tonality:
1604
- # Return only the content messages without any IDs or extra details
1605
- response = "\n".join([f"- {entry['message']}" for entry in brand_tonality.values()])
1606
- return f"Here are your saved Brand Tonality entries:\n{response}"
1607
- else:
1608
- return "You haven't saved any Brand Tonality entries yet."
1609
 
1610
- return None #
1611
 
1612
  def delete_entry(category, entry_id):
1613
  try:
@@ -1725,7 +1715,11 @@ global combined_text
1725
  if st.session_state["used_messages"] < st.session_state["message_limit"]:
1726
  if prompt:
1727
  st.session_state.chat_started = True
1728
- st.session_state.chat_history.append({"role": "user", "content": prompt})
 
 
 
 
1729
 
1730
 
1731
  # Handle save commands based on the user prompt
 
1567
 
1568
  def handle_memory_queries(prompt):
1569
  prompt = prompt.lower().strip()
1570
+
1571
  # Save as TrustBuilder
1572
  if re.search(r"\b(keep|add|save|store)\s*(this)?\s*(as)?\s*(trust\s*builder|trustbuilder)\b", prompt, re.IGNORECASE):
1573
  content_to_save = re.sub(r"\b(keep|add|save|store)\s*(this)?\s*(as)?\s*(trust\s*builder|trustbuilder)\b", "", prompt, flags=re.IGNORECASE).strip()
 
1591
  elif "show my saved trustbuilders" in prompt or "find my saved trustbuilders" in prompt:
1592
  trustbuilders = st.session_state.get("TrustBuilder", {})
1593
  if trustbuilders:
 
1594
  response = "\n".join([f"- {entry['message']}" for entry in trustbuilders.values()])
1595
  return f"Here are your saved TrustBuilders:\n{response}"
1596
  else:
1597
  return "You haven't saved any TrustBuilders yet."
1598
 
1599
+ return None
 
 
 
 
 
 
 
 
1600
 
 
1601
 
1602
  def delete_entry(category, entry_id):
1603
  try:
 
1715
  if st.session_state["used_messages"] < st.session_state["message_limit"]:
1716
  if prompt:
1717
  st.session_state.chat_started = True
1718
+ # Prevent duplicate messages in chat history
1719
+ if not any(msg["content"] == prompt for msg in st.session_state["chat_history"]):
1720
+ st.session_state.chat_history.append({"role": "user", "content": prompt})
1721
+
1722
+
1723
 
1724
 
1725
  # Handle save commands based on the user prompt