Update app.py

app.py

@@ -82,6 +82,13 @@ Multiplayer_Custom_Hosting_Game_Servers_For_Simulated_Worlds = """
 29. Valheim
 """
 
+def get_display_name(filename):
+    """Extract text from parentheses or return filename as is."""
+    match = re.search(r'\((.*?)\)', filename)
+    if match:
+        return match.group(1)
+    return filename
+
 def sanitize_filename(text):
     """Create a safe filename from text while preserving spaces."""
     # First replace unsafe characters with spaces
@@ -136,14 +143,13 @@ def get_file_download_link(file_path):
         with open(file_path, 'r', encoding='utf-8') as f:
             content = f.read()
         b64 = base64.b64encode(content.encode()).decode()
-        filename = os.path.basename(file_path)
-        return f'<a href="data:text/markdown;base64,{b64}" download="{filename}">{filename}</a>'
+        return f"data:text/markdown;base64,{b64}"
     except Exception as e:
         st.error(f"Error creating download link: {e}")
         return None
 
-# Function to parse markdown text and extract terms
 def extract_terms(markdown_text):
+    """Parse markdown text and extract terms."""
     lines = markdown_text.strip().split('\n')
     terms = []
     for line in lines:
@@ -152,8 +158,8 @@ def extract_terms(markdown_text):
             terms.append(line)
     return terms
 
-# Function to display terms with links
 def display_terms_with_links(terms):
+    """Display terms with various search links."""
     search_urls = {
         "🚀🌌ArXiv": lambda k: f"/?q={quote(k)}",
         "📖": lambda k: f"https://en.wikipedia.org/wiki/{quote(k)}",
@@ -166,8 +172,8 @@ def display_terms_with_links(terms):
         links_md = ' '.join([f"[{emoji}]({url(term)})" for emoji, url in search_urls.items()])
         st.markdown(f"- **{term}** {links_md}", unsafe_allow_html=True)
 
-# Function to perform AI lookup using Gradio client
 def perform_ai_lookup(query):
+    """Perform AI lookup using Gradio client."""
     st.write("Performing AI Lookup...")
     client = Client("awacke1/Arxiv-Paper-Search-And-QA-RAG-Pattern")
     result1 = client.predict(
@@ -217,55 +223,82 @@ def display_file_content(file_path):
         st.error(f"Error reading file: {e}")
 
 def file_management_sidebar():
-    """Enhanced sidebar with file management capabilities."""
+    """Redesigned sidebar with improved layout and additional functionality."""
     st.sidebar.title("📁 File Management")
 
-    # Get list of .md files
-    md_files = sorted(glob.glob("*.md"))
+    # Get list of .md files excluding README.md
+    md_files = [file for file in glob.glob("*.md") if file.lower() != 'readme.md']
+    md_files.sort()
     st.session_state.files = md_files
     
     if md_files:
-        st.sidebar.markdown("### Markdown Files")
+        st.sidebar.markdown("### Saved Files")
         for idx, file in enumerate(md_files):
-            col1, col2, col3, col4 = st.sidebar.columns([4, 2, 1, 1])
+            st.sidebar.markdown("---")  # Separator between files
             
-            with col1:
-                st.write(file)
+            # Display filename
+            st.sidebar.text(file)
             
-            with col2:
-                # Base64 download link
-                st.markdown(get_file_download_link(file), unsafe_allow_html=True)
+            # Display download link with simplified text
+            display_name = get_display_name(file)
+            download_link = get_file_download_link(file)
+            if download_link:
+                st.sidebar.markdown(f"📥 [{display_name}]({download_link})", unsafe_allow_html=True)
             
-            with col3:
-                # View button
-                if st.button("📄", key=f"view_{idx}"):
+            # Action buttons in a row
+            col1, col2, col3 = st.sidebar.columns(3)
+            
+            with col1:
+                if st.button("📄 View", key=f"view_{idx}"):
                     st.session_state.selected_file = file
                     st.session_state.view_mode = 'view'
             
-            with col4:
-                # Edit button
-                if st.button("✏️", key=f"edit_{idx}"):
+            with col2:
+                if st.button("✏️ Edit", key=f"edit_{idx}"):
                     st.session_state.selected_file = file
                     st.session_state.view_mode = 'edit'
+            
+            with col3:
+                if st.button("🔄 Rerun", key=f"rerun_{idx}"):
+                    try:
+                        with open(file, 'r', encoding='utf-8') as f:
+                            content = f.read()
+                            
+                        # Prepare the prompt with the prefix
+                        rerun_prefix = """For the markdown below reduce the text to a humorous fun outline with emojis and markdown outline levels in outline that convey all the facts and adds wise quotes and funny statements to engage the reader:
+
+"""
+                        full_prompt = rerun_prefix + content
+                        
+                        # Perform AI lookup and save results
+                        ai_result = perform_ai_lookup(full_prompt)
+                        saved_file = save_ai_interaction(f"Rerun of {file}", ai_result)
+                        
+                        if saved_file:
+                            st.success(f"Created fun version in {saved_file}")
+                            st.session_state.selected_file = saved_file
+                            st.session_state.view_mode = 'view'
+                            
+                    except Exception as e:
+                        st.error(f"Error during rerun: {e}")
 
+        st.sidebar.markdown("---")
         # Option to create a new markdown file
-        if st.sidebar.button("Create New Markdown File"):
-            timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
-            new_filename = f"note_{timestamp}.md"
-            with open(new_filename, 'w', encoding='utf-8') as f:
+        if st.sidebar.button("📝 Create New Note"):
+            filename = generate_timestamp_filename("New Note")
+            with open(filename, 'w', encoding='utf-8') as f:
                 f.write("# New Markdown File\n")
-            st.sidebar.success(f"Created new file: {new_filename}")
-            st.session_state.selected_file = new_filename
+            st.sidebar.success(f"Created: {filename}")
+            st.session_state.selected_file = filename
             st.session_state.view_mode = 'edit'
     else:
         st.sidebar.write("No markdown files found.")
-        if st.sidebar.button("Create New Markdown File"):
-            timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
-            new_filename = f"note_{timestamp}.md"
-            with open(new_filename, 'w', encoding='utf-8') as f:
+        if st.sidebar.button("📝 Create First Note"):
+            filename = generate_timestamp_filename("New Note")
+            with open(filename, 'w', encoding='utf-8') as f:
                 f.write("# New Markdown File\n")
-            st.sidebar.success(f"Created new file: {new_filename}")
-            st.session_state.selected_file = new_filename
+            st.sidebar.success(f"Created: {filename}")
+            st.session_state.selected_file = filename
             st.session_state.view_mode = 'edit'
 
 def main():