Update app.py

app.py

@@ -243,8 +243,75 @@ def push_to_github(local_path, repo, github_token):
 
     origin.push(refspec=f'{current_branch}:{current_branch}')
 
-# 💾 Save or clone to Cosmos DB - Because every document deserves a twin
+
+
 def save_or_clone_to_cosmos_db(container, query=None, response=None, clone_id=None):
+    max_retries = 5
+    retry_delay = 1  # seconds
+
+    def generate_unique_id():
+        return f"{datetime.utcnow().strftime('%Y%m%d%H%M%S%f')}-{str(uuid.uuid4())}"
+
+    for attempt in range(max_retries):
+        try:
+            if not container:
+                st.error("Cosmos DB container is not initialized.")
+                return
+
+            new_id = generate_unique_id()
+
+            if clone_id:
+                # If clone_id is provided, we're cloning an existing document
+                try:
+                    existing_doc = container.read_item(item=clone_id, partition_key=clone_id)
+                    new_doc = existing_doc.copy()
+                    new_doc['id'] = new_id
+                    new_doc['cloned_from'] = clone_id
+                    new_doc['cloned_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
+            container.create_item(body=new_doc)
+
+            st.success(f"{'Cloned' if clone_id else 'New'} document saved successfully with ID: {new_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 new_id
+
+        except exceptions.CosmosHttpResponseError as e:
+            if e.status_code == 409:  # Conflict error
+                st.warning(f"ID conflict occurred. Retrying... (Attempt {attempt + 1})")
+                time.sleep(retry_delay)
+            else:
+                st.error(f"Error saving to Cosmos DB: {e}")
+                return
+        except Exception as e:
+            st.error(f"An unexpected error occurred: {str(e)}")
+            return
+
+    st.error("Failed to save document after maximum retries.")
+
+
+
+
+    
+# 💾 Save or clone to Cosmos DB - Because every document deserves a twin
+def save_or_clone_to_cosmos_db2(container, query=None, response=None, clone_id=None):
     try:
         if not container:
             st.error("Cosmos DB container is not initialized.")