Update app.py

app.py

@@ -188,6 +188,46 @@ def delete_record(container, record):
         
 # 💾 Save to Cosmos DB - Preserving data for future generations (or just until the next update)
 def save_to_cosmos_db(container, query, response1, response2):
+    try:
+        if container:
+            # Generate a unique ID that includes a timestamp
+            timestamp = datetime.utcnow().strftime('%Y%m%d%H%M%S%f')
+            unique_uuid = str(uuid.uuid4())
+            new_id = f"{timestamp}-{unique_uuid}"
+            
+            # Create new document with proper name field
+            record = {
+                "id": new_id,
+                "name": new_id,  # Set name equal to ID to avoid null name error
+                "query": query,
+                "response1": response1,
+                "response2": response2,
+                "timestamp": datetime.utcnow().isoformat(),
+                "type": "ai_response",  # Add document type for better organization
+                "version": "1.0"
+            }
+            
+            try:
+                # Create the new document
+                container.create_item(body=record)
+                st.success(f"Record saved successfully with ID: {record['id']}")
+                # Refresh the documents display
+                st.session_state.documents = get_documents(container)
+            except exceptions.CosmosHttpResponseError as e:
+                st.error(f"Error saving record to Cosmos DB: {e}")
+        else:
+            st.error("Cosmos DB container is not initialized.")
+    except Exception as e:
+        st.error(f"An unexpected error occurred: {str(e)}")
+
+
+
+
+
+
+
+
+def save_to_cosmos_db_old(container, query, response1, response2):
     try:
         if container:
             record = {
@@ -555,7 +595,67 @@ def display_file_viewer(file_path):
             mime="text/markdown"
         )
 
+
+
 def display_file_editor(file_path):
+    """Display file content in both Markdown and Code Editor views"""
+    # Initialize file content in session state if not already present
+    if 'file_content' not in st.session_state:
+        st.session_state.file_content = {}
+    
+    # Load content if not in session state or if it's a different file
+    if file_path not in st.session_state.file_content:
+        content = load_file_content(file_path)
+        if content is not None:
+            st.session_state.file_content[file_path] = content
+        else:
+            return
+    
+    st.markdown("### ✏️ File Editor")
+    st.markdown(f"**Editing:** {file_path}")
+    
+    # Create tabs for different views
+    markdown_tab, code_tab = st.tabs(["Markdown View", "Code Editor"])
+    
+    with markdown_tab:
+        st.markdown("### 📄 Markdown Preview")
+        st.markdown(st.session_state.file_content[file_path])
+    
+    with code_tab:
+        st.markdown("### 💻 Code Editor")
+        # Create a unique key for the text area
+        editor_key = f"editor_{hash(file_path)}"
+        
+        # Editor with syntax highlighting for markdown
+        new_content = st.text_area(
+            "Edit content below:",
+            value=st.session_state.file_content[file_path],
+            height=400,
+            key=editor_key
+        )
+    
+    # Add save and download buttons below both views
+    col1, col2 = st.columns([1, 5])
+    with col1:
+        if st.button("💾 Save Changes"):
+            if save_file_content(file_path, new_content):
+                st.session_state.file_content[file_path] = new_content
+                st.success("File saved successfully! 🎉")
+                time.sleep(1)
+                st.rerun()
+    
+    with col2:
+        st.download_button(
+            label="⬇️ Download File",
+            data=new_content,
+            file_name=os.path.basename(file_path),
+            mime="text/markdown"
+        )
+
+
+
+
+def display_file_editor_old(file_path):
     """Display file content in editor with save functionality"""
     # Initialize file content in session state if not already present
     if 'file_content' not in st.session_state: