Update app.py

app.py

@@ -308,15 +308,6 @@ def save_or_clone_to_cosmos_db(container, query=None, response=None, clone_id=No
     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
@@ -681,7 +672,16 @@ def main():
                                         if st.button("🤖 Run With AI", key=f'run_with_ai_button_{idx}'):
                                             # Use the entire document as input
                                             search_glossary(json.dumps(editable_doc, indent=2))
-                        
+                                            # 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.")
+
                         elif selected_view == 'Clone Document':
                             # 🧬 Clone Document per record
                             st.markdown("#### Clone a document:")