Update app.py

app.py

@@ -21,6 +21,8 @@ import zipfile
 import hashlib
 import time
 
+
+
 # 🎭 App Configuration - Because every app needs a good costume!
 Site_Name = '🐙Git🌌Cosmos💫 - Azure Cosmos DB and Github Agent'
 title = "🐙Git🌌Cosmos💫 - Azure Cosmos DB and Github Agent"
@@ -308,7 +310,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
 
-
     
 # 💾 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):
@@ -452,6 +453,56 @@ def search_glossary(query):
 
     return result, result2, result3, response2
 
+
+
+
+
+# 📝 Generate a safe filename from the first few lines of content
+def generate_filename_from_content(content, file_type="md"):
+    # Extract the first few lines or sentences
+    first_sentence = content.split('\n', 1)[0][:90]  # Limit the length to 90 characters
+    # Remove special characters to make it a valid filename
+    safe_name = re.sub(r'[^\w\s-]', '', first_sentence)
+    # Limit length to be compatible with Windows and Linux
+    safe_name = safe_name[:50].strip()  # Adjust length limit
+    return f"{safe_name}.{file_type}"
+
+# 💾 Create and save a file
+def create_file_from_content(content, should_save=True):
+    if not should_save:
+        return
+    filename = generate_filename_from_content(content)
+    with open(filename, 'w', encoding='utf-8') as file:
+        file.write(content)
+    return filename
+
+# 📂 Display list of saved .md files in the sidebar
+def display_saved_files_in_sidebar():
+    all_files = glob.glob("*.md")
+    all_files.sort(reverse=True)
+    all_files = [file for file in all_files if not file.lower().startswith('readme')]  # Exclude README.md
+
+    st.sidebar.markdown("## 📁 Saved Markdown Files")
+    for file in all_files:
+        col1, col2, col3 = st.sidebar.columns([6, 2, 1])
+        with col1:
+            st.markdown(f"📄 {file}")
+        with col2:
+            st.sidebar.download_button(
+                label="⬇️ Download",
+                data=open(file, 'rb').read(),
+                file_name=file
+            )
+        with col3:
+            if st.sidebar.button("🗑", key=f"delete_{file}"):
+                os.remove(file)
+                st.rerun()
+
+
+                
+
+
+
 # 🎭 Main function - Where the magic happens (and occasionally breaks)
 def main():
     st.title("🐙Git🌌Cosmos💫 - Azure Cosmos DB and Github Agent")
@@ -482,6 +533,28 @@ def main():
             # 🕵️‍♂️ We have a query! Let's process it!
             result, result2, result3, response2 = search_glossary(query)
             # When saving results, pass the container
+            try:
+                # 📝 Automatically save the AI-generated content
+                if st.button("Save AI Output"):
+
+                    # 🕵️‍♂️ Save our nice research summaries for later sleuthing
+                    filename = create_file_from_content(result)
+                    st.success(f"File saved: {filename}")                    
+                    filename = create_file_from_content(result2)
+                    st.success(f"File saved: {filename}")
+                    filename = create_file_from_content(result3)
+                    st.success(f"File saved: {filename}")
+                    filename = create_file_from_content(response2)
+                    st.success(f"File saved: {filename}")
+                    
+                    #st.rerun()  # Rerun the app to update the sidebar
+            
+                # 📁 Display saved markdown files in the sidebar
+                display_saved_files_in_sidebar()
+            except Exception as e:
+                st.error(f"An unexpected error occurred: {str(e)} 😱")
+
+                
             try:
                 save_to_cosmos_db(st.session_state.cosmos_container, query, result, result)
                 save_to_cosmos_db(st.session_state.cosmos_container, query, result2, result2)