Update app.py

app.py

@@ -244,60 +244,79 @@ def push_to_github(local_path, repo, github_token):
     origin.push(refspec=f'{current_branch}:{current_branch}')
 
 
-def save_or_clone_to_cosmos_db(container, query=None, response=None, clone_id=None):
-    def generate_unique_id():
-        return f"{datetime.utcnow().strftime('%Y%m%d%H%M%S%f')}-{str(uuid.uuid4())}"
 
-    try:
-        if not container:
-            st.error("Cosmos DB container is not initialized.")
-            return
+def save_or_clone_to_cosmos_db(container, query=None, response=None, clone_id=None):
+    def generate_complex_unique_id():
+        timestamp = datetime.utcnow().strftime('%Y%m%d%H%M%S%f')
+        random_component = ''.join(random.choices('abcdefghijklmnopqrstuvwxyz0123456789', k=8))
+        return f"{timestamp}-{random_component}-{str(uuid.uuid4())}"
 
-        new_id = generate_unique_id()
+    max_retries = 10
+    base_delay = 0.1  # 100 ms
 
-        if clone_id:
-            # If clone_id is provided, we're creating a new document based on an existing one
-            try:
-                existing_doc = container.read_item(item=clone_id, partition_key=clone_id)
+    for attempt in range(max_retries):
+        try:
+            new_id = generate_complex_unique_id()
+
+            if clone_id:
+                try:
+                    existing_doc = container.read_item(item=clone_id, partition_key=clone_id)
+                    new_doc = {
+                        'id': new_id,
+                        'originalText': existing_doc.get('originalText', ''),
+                        'qtPrompts': existing_doc.get('qtPrompts', []),
+                        'cloned_from': clone_id,
+                        'cloned_at': datetime.utcnow().isoformat()
+                    }
+                except exceptions.CosmosResourceNotFoundError:
+                    st.error(f"Document with ID {clone_id} not found for cloning.")
+                    return None
+            else:
                 new_doc = {
                     'id': new_id,
-                    'originalText': existing_doc.get('originalText', ''),
-                    'qtPrompts': existing_doc.get('qtPrompts', []),
-                    'cloned_from': clone_id,
-                    'cloned_at': datetime.utcnow().isoformat()
+                    'query': query,
+                    'response': response,
+                    'created_at': datetime.utcnow().isoformat()
                 }
-            except exceptions.CosmosResourceNotFoundError:
-                st.error(f"Document with ID {clone_id} not found for cloning.")
-                return
-        else:
-            # If no clone_id, we're creating a new document
-            new_doc = {
-                'id': new_id,
-                'query': query,
-                'response': response,
-                'created_at': datetime.utcnow().isoformat()
-            }
-
-        # Insert the new document
-        response = container.create_item(body=new_doc)
 
-        st.success(f"{'Cloned' if clone_id else 'New'} document saved successfully with ID: {response['id']}")
-
-        # Refresh the documents in the session state
-        st.session_state.documents = list(container.query_items(
-            query="SELECT * FROM c ORDER BY c._ts DESC",
-            enable_cross_partition_query=True
-        ))
+            # Attempt to create the item
+            response = container.create_item(body=new_doc)
+            
+            st.success(f"{'Cloned' if clone_id else 'New'} document saved successfully with ID: {response['id']}")
+            
+            # Refresh the documents in the session state
+            st.session_state.documents = list(container.query_items(
+                query="SELECT * FROM c ORDER BY c._ts DESC",
+                enable_cross_partition_query=True
+            ))
+            
+            return response['id']
 
-        return response['id']
+        except exceptions.CosmosHttpResponseError as e:
+            if e.status_code == 409:  # Conflict error
+                delay = base_delay * (2 ** attempt) + random.uniform(0, 0.1)
+                st.warning(f"ID conflict occurred. Retrying in {delay:.2f} seconds... (Attempt {attempt + 1})")
+                time.sleep(delay)
+            else:
+                st.error(f"Error saving to Cosmos DB: {e}")
+                return None
 
-    except exceptions.CosmosHttpResponseError as e:
-        st.error(f"Error saving to Cosmos DB: {e}")
-    except Exception as e:
-        st.error(f"An unexpected error occurred: {str(e)}")
+        except Exception as e:
+            st.error(f"An unexpected error occurred: {str(e)}")
+            return None
 
+    st.error("Failed to save document after maximum retries.")
     return None
 
+# Usage in your Streamlit app
+if st.button("📄 Clone Document", key=f'clone_button_{idx}'):
+    with st.spinner("Cloning document..."):
+        cloned_id = save_or_clone_to_cosmos_db(container, clone_id=doc['id'])
+        if cloned_id:
+            st.success(f"Document cloned successfully with new ID: {cloned_id}")
+            st.rerun()
+        else:
+            st.error("Failed to clone document. Please try again.")
 
     
 # 💾 Save or clone to Cosmos DB - Because every document deserves a twin