Update app.py

app.py

@@ -56,16 +56,29 @@ st.write("Fields Usually Needed:")
 for field in letter_types[selected_letter_type]["Fields"]:
     st.button(field)
 
+
 def generate_referral_letters_ui():
     st.title("🤝 Referral Letters")
 
+    # Define the keys for each UI element
+    keys = {
+        "search_query": "search_query",
+        "first_button": "first_button",
+        "middle_button": "middle_button",
+        "last_button": "last_button",
+        "first_text": "first_text",
+        "middle_text": "middle_text",
+        "last_text": "last_text",
+        "username": "username"
+    }
+
     # Search-style text box
-    search_query = st.text_input("Search Referral Letters")
+    search_query = st.text_input("Search Referral Letters", key=keys["search_query"])
 
     # Buttons with emojis and boldface text
-    st.button("📝 **First**")
-    st.button("📝 **Middle**")
-    st.button("📝 **Last**")
+    st.button("📝 **First**", key=keys["first_button"])
+    st.button("📝 **Middle**", key=keys["middle_button"])
+    st.button("📝 **Last**", key=keys["last_button"])
 
     # Text areas with default values based on letter examples
     first_default = "Urgent need for further diagnostic testing for Mrs. Smith, who has persistent stomach issues\n" \
@@ -75,12 +88,12 @@ def generate_referral_letters_ui():
     last_default = "Patient demographics, Referral details\n" \
                    "Diagnostic test reports, Medication details"
 
-    first_text = st.text_area("First: State the request for consultation/treatment", value=first_default, max_chars=None, height=None)
-    middle_text = st.text_area("Middle: Explain the reason for referral", value=middle_default, max_chars=None, height=None)
-    last_text = st.text_area("Last: Provide patient info needed for the referred service", value=last_default, max_chars=None, height=None)
+    first_text = st.text_area("First: State the request for consultation/treatment", value=first_default, max_chars=None, height=None, key=keys["first_text"])
+    middle_text = st.text_area("Middle: Explain the reason for referral", value=middle_default, max_chars=None, height=None, key=keys["middle_text"])
+    last_text = st.text_area("Last: Provide patient info needed for the referred service", value=last_default, max_chars=None, height=None, key=keys["last_text"])
 
     # Username field with an emoji
-    username = st.text_input("👤 Username")
+    username = st.text_input("👤 Username", key=keys["username"])
 
 
 def generate_medical_certificate_letters_ui():