Update app.py

app.py

@@ -249,23 +249,21 @@ 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 save_or_clone_to_cosmos_db(container, document=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())}"
 
     max_retries = 10
-    base_delay = 0.1  # 100 ms
+    base_delay = 0.1
 
     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,
@@ -275,44 +273,28 @@ def save_or_clone_to_cosmos_db(container, query=None, response=None, clone_id=No
                         'cloned_at': datetime.utcnow().isoformat()
                     }
                 except exceptions.CosmosResourceNotFoundError:
-                    st.error(f"Document with ID {clone_id} not found for cloning.")
-                    return None
+                    return False, f"Document with ID {clone_id} not found for cloning."
             else:
-                new_doc = {
-                    'id': new_id,
-                    'query': query,
-                    'response': response,
-                    'created_at': datetime.utcnow().isoformat()
-                }
-
-            # Attempt to create the item
+                if document is None:
+                    return False, "No document provided for saving"
+                
+                document['id'] = new_id
+                document['created_at'] = datetime.utcnow().isoformat()
+                new_doc = 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
-            ))
-            
-            return response['id']
+            return True, 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
+            if e.status_code == 409:
                 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
-
+                continue
+            return False, f"Error saving to Cosmos DB: {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
+            return False, f"An unexpected error occurred: {str(e)}"
+    
+    return False, "Failed to save document after maximum retries."
 
     
 # 📦 Archive current container - Packing up data like you're moving to a new digital house