Update app.py

app.py

@@ -61,6 +61,12 @@ if 'previous_trust_tip' not in st.session_state:
     st.session_state.previous_trust_tip = None
 if 'previous_suggestion' not in st.session_state:
     st.session_state.previous_suggestion = None
+
+if 'used_trust_tips' not in st.session_state:
+    st.session_state.used_trust_tips = set()
+if 'used_suggestions' not in st.session_state:
+    st.session_state.used_suggestions = set()
+    
 def copy_to_clipboard(text):
     """Creates a button to copy text to clipboard."""
     escaped_text = text.replace('\n', '\\n').replace('"', '\\"')
@@ -197,11 +203,19 @@ def clean_text(text):
     return cleaned_text
 
 
-def get_trust_tip_and_suggestion():
-    trust_tip = random.choice(trust_tips)
-    suggestion = random.choice(suggestions)
+def get_trust_tip_and_suggestion(used_tips, used_suggestions):
+    available_tips = [tip for tip in trust_tips if tip not in used_tips]
+    available_suggestions = [suggestion for suggestion in suggestions if suggestion not in used_suggestions]
+    
+    if not available_tips:  # Reset if no available tips
+        available_tips = trust_tips
+    if not available_suggestions:  # Reset if no available suggestions
+        available_suggestions = suggestions
+
+    trust_tip = random.choice(available_tips)
+    suggestion = random.choice(available_suggestions)
+    
     return trust_tip, suggestion
-   
 
 
 
@@ -628,14 +642,12 @@ if prompt :
                 #cleaned_text = re.sub(r'</span>', '', cleaned_text) 
                 #cleaned_text = re.sub(r'<span[^>]*>', '', cleaned_text)
                 # Display the response
-                trust_tip, suggestion = get_trust_tip_and_suggestion()
-                if trust_tip == st.session_state.previous_trust_tip:
-                    trust_tip = random.choice([t for t in trust_tips if t != st.session_state.previous_trust_tip])
-                if suggestion == st.session_state.previous_suggestion:
-                    suggestion = random.choice([s for s in suggestions if s != st.session_state.previous_suggestion])
-                st.session_state.previous_trust_tip = trust_tip
-                st.session_state.previous_suggestion = suggestion
-                    
+                
+                trust_tip, suggestion = get_trust_tip_and_suggestion(st.session_state.used_trust_tips, st.session_state.used_suggestions)
+    
+                # Update used trust tips and suggestions
+                st.session_state.used_trust_tips.add(trust_tip)
+                st.session_state.used_suggestions.add(suggestion)
                 
                 combined_text = f"{cleaned_text}\n\n---\n\n**Trust Tip**: {trust_tip}\n\n**Suggestion**: {suggestion}"
                 st.markdown(combined_text, unsafe_allow_html=True)