Update app.py

app.py

@@ -116,9 +116,9 @@ def preprocess_text(text):
     return text.strip()
 
 def sanitize_json_text(text):
+    # Remove control characters but keep newlines intact
     text = re.sub(r'[\x00-\x08\x0B\x0C\x0E-\x1F]', '', text)
-    text = text.replace("\n", "\\n").replace("\r", "\\r").replace("\t", "\\t")
-    return text
+    return text.strip()
 
 # =============================================================================
 # ───────────── COSMOS DB FUNCTIONS ─────────────
@@ -155,15 +155,16 @@ def update_record(container, updated_record):
 def delete_record(container, record):
     try:
         doc_id = record["id"]
-        partition_key_value = record.get("pk", doc_id)
+        partition_key_value = record.get("pk", doc_id)  # Fallback to id if pk missing
+        st.write(f"Deleting {doc_id} with partition key {partition_key_value}")  # Debug log
         container.delete_item(item=doc_id, partition_key=partition_key_value)
         return True, f"Record {doc_id} deleted. 🗑️"
     except exceptions.CosmosResourceNotFoundError:
-        return True, f"Record {doc_id} not found. 🗑️"
+        return True, f"Record {doc_id} not found (already deleted). 🗑️"
     except exceptions.CosmosHttpResponseError as e:
-        return False, f"HTTP error: {str(e)} 🚨"
+        return False, f"HTTP error deleting {doc_id}: {str(e)} 🚨"
     except Exception as e:
-        return False, f"Error: {str(e)} 😱"
+        return False, f"Unexpected error deleting {doc_id}: {str(e)} 😱"
 
 def save_to_cosmos_db(container, query, response1, response2):
     try:
@@ -366,9 +367,13 @@ def edit_all_documents(container):
             with col_save:
                 if st.button("💾 Save", key=f"save_{doc['id']}"):
                     try:
-                        updated_doc = json.loads(sanitize_json_text(edited_content))
+                        # Parse edited content
+                        cleaned_content = sanitize_json_text(edited_content)
+                        updated_doc = json.loads(cleaned_content)
+                        # Preserve identity fields
                         updated_doc['id'] = doc['id']
                         updated_doc['pk'] = doc.get('pk', doc['id'])
+                        # Remove system fields
                         for field in ['_ts', '_rid', '_self', '_etag', '_attachments']:
                             updated_doc.pop(field, None)
                         success, message = update_record(container, updated_doc)
@@ -377,8 +382,8 @@ def edit_all_documents(container):
                             st.session_state[doc_key] = json.dumps(updated_doc, indent=2)
                         else:
                             st.error(message)
-                    except json.JSONDecodeError:
-                        st.error("Invalid JSON format.")
+                    except json.JSONDecodeError as e:
+                        st.error(f"Invalid JSON format: {str(e)}\nProblematic input:\n{cleaned_content}")
                     except Exception as e:
                         st.error(f"Save error: {str(e)}")
             with col_delete:
@@ -386,6 +391,8 @@ def edit_all_documents(container):
                     success, message = delete_record(container, doc)
                     if success:
                         st.success(message)
+                        if doc_key in st.session_state:
+                            del st.session_state[doc_key]  # Clean up session state
                         st.rerun()
                     else:
                         st.error(message)