Update app.py

app.py

@@ -162,30 +162,41 @@ def delete_record(container, name, id):
     except Exception as e:
         return False, f"An unexpected error occurred: {traceback.format_exc()} 😱"
 
+import uuid
+
+def generate_unique_id():
+    return str(uuid.uuid4())
+    
 def save_to_cosmos_db(container, query, response1, response2):
     try:
         if container:
-            record = {
-                "id": generate_unique_id(),
-                "query": query,
-                "response1": response1,
-                "response2": response2,
-                "timestamp": datetime.utcnow().isoformat()
-            }
-            try:
-                container.create_item(body=record)
-                st.success(f"Record saved successfully with ID: {record['id']}")
-                # Refresh the documents display
-                st.session_state.documents = get_documents(container)
-            except exceptions.CosmosHttpResponseError as e:
-                st.error(f"Error saving record to Cosmos DB: {e}")
+            max_retries = 3
+            for attempt in range(max_retries):
+                try:
+                    record = {
+                        "id": generate_unique_id(),
+                        "query": query,
+                        "response1": response1,
+                        "response2": response2,
+                        "timestamp": datetime.utcnow().isoformat()
+                    }
+                    container.upsert_item(body=record)
+                    st.success(f"Record saved successfully with ID: {record['id']}")
+                    st.session_state.documents = get_documents(container)
+                    break
+                except exceptions.CosmosHttpResponseError as e:
+                    if e.status_code == 409 and attempt < max_retries - 1:  # Conflict error
+                        st.warning(f"ID conflict occurred. Retrying... (Attempt {attempt + 1})")
+                        continue
+                    else:
+                        raise
         else:
             st.error("Cosmos DB container is not initialized.")
+    except exceptions.CosmosHttpResponseError as e:
+        st.error(f"Error saving record to Cosmos DB: {e}")
     except Exception as e:
         st.error(f"An unexpected error occurred: {str(e)}")
 
-
-
 # Add dropdowns for model and database choices
 def search_glossary(query):
     st.markdown(f"### 🔍 Search Glossary for: `{query}`")