Update app.py

app.py

@@ -167,35 +167,62 @@ import uuid
 def generate_unique_id():
     return str(uuid.uuid4())
     
+
 def save_to_cosmos_db(container, query, response1, response2):
-    try:
-        if container:
-            max_retries = 3
-            for attempt in range(max_retries):
+    max_retries = 5
+    base_wait_time = 0.1  # 100 ms
+
+    def generate_unique_id():
+        timestamp = datetime.utcnow().strftime('%Y%m%d%H%M%S%f')
+        unique_uuid = str(uuid.uuid4())
+        return f"{timestamp}-{unique_uuid}"
+
+    for attempt in range(max_retries):
+        try:
+            if container:
+                base_id = generate_unique_id()
+                record = {
+                    "id": base_id,
+                    "query": query,
+                    "response1": response1,
+                    "response2": response2,
+                    "timestamp": datetime.utcnow().isoformat()
+                }
+                
+                # Check if document with this ID exists
                 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)}")
+                    existing_doc = container.read_item(item=base_id, partition_key=base_id)
+                    # If it exists, append a random string to the ID
+                    record["id"] = f"{base_id}-{str(uuid.uuid4())[:8]}"
+                except exceptions.CosmosResourceNotFoundError:
+                    # If it doesn't exist, we can use the original ID
+                    pass
+                
+                container.upsert_item(body=record)
+                st.success(f"Record saved successfully with ID: {record['id']}")
+                st.session_state.documents = container.query_items(
+                    query="SELECT * FROM c ORDER BY c._ts DESC",
+                    enable_cross_partition_query=True
+                )
+                return
+            else:
+                st.error("Cosmos DB container is not initialized.")
+                return
+        except exceptions.CosmosHttpResponseError as e:
+            if e.status_code == 409:  # Conflict error
+                wait_time = (2 ** attempt) * base_wait_time
+                st.warning(f"ID conflict occurred. Retrying in {wait_time:.2f} seconds... (Attempt {attempt + 1})")
+                time.sleep(wait_time)
+            else:
+                st.error(f"Error saving record to Cosmos DB: {e}")
+                return
+        except Exception as e:
+            st.error(f"An unexpected error occurred: {str(e)}")
+            return
+
+    st.error("Failed to save record after maximum retries.")
+
+    
 
 # Add dropdowns for model and database choices
 def search_glossary(query):