Update app.py

app.py

@@ -244,69 +244,59 @@ 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):
-    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
+    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_id = generate_unique_id()
+
+        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)
                 new_doc = {
                     'id': new_id,
-                    'query': query,
-                    'response': response,
-                    'created_at': datetime.utcnow().isoformat()
+                    '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
+        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
-            ))
+        # Insert the new document
+        response = container.create_item(body=new_doc)
 
-            return new_id
+        st.success(f"{'Cloned' if clone_id else 'New'} document saved successfully with ID: {response['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
+        # 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
+        ))
 
-    st.error("Failed to save document after maximum retries.")
+        return response['id']
 
+    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)}")
 
+    return None