Update app.py

app.py

@@ -89,6 +89,13 @@ def get_display_name(filename):
         return match.group(1)
     return filename
 
+def get_time_display(filename):
+    """Extract just the time portion from the filename."""
+    time_match = re.match(r'(\d{2}\d{2}[AP]M)', filename)
+    if time_match:
+        return time_match.group(1)
+    return filename
+
 def sanitize_filename(text):
     """Create a safe filename from text while preserving spaces."""
     # First replace unsafe characters with spaces
@@ -117,12 +124,31 @@ def generate_timestamp_filename(query):
     
     return filename
 
-def save_ai_interaction(query, ai_result):
+def delete_file(file_path):
+    """Delete a file and return success status."""
+    try:
+        os.remove(file_path)
+        return True
+    except Exception as e:
+        st.error(f"Error deleting file: {e}")
+        return False
+
+def save_ai_interaction(query, ai_result, is_rerun=False):
     """Save AI interaction to a markdown file with new filename format."""
     filename = generate_timestamp_filename(query)
     
-    # Format the content
-    content = f"""# Query: {query}
+    # Format the content differently for rerun vs normal query
+    if is_rerun:
+        content = f"""# Rerun Query
+Original file content used for rerun:
+
+{query}
+
+# AI Response (Fun Version)
+{ai_result}
+"""
+    else:
+        content = f"""# Query: {query}
 
 ## AI Response
 {ai_result}
@@ -143,7 +169,8 @@ 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()
-        return f"data:text/markdown;base64,{b64}"
+        filename = os.path.basename(file_path)
+        return f'<a href="data:text/markdown;base64,{b64}" download="{filename}">{get_display_name(filename)}</a>'
     except Exception as e:
         st.error(f"Error creating download link: {e}")
         return None
@@ -201,24 +228,25 @@ def display_file_content(file_path):
         with open(file_path, 'r', encoding='utf-8') as f:
             content = f.read()
         
-        # Display as code with line numbers
-        st.code(content, line_numbers=True)
-        
-        # Edit functionality
-        edited_content = st.text_area(
-            "Edit content",
-            content,
-            height=400,
-            key=f"edit_{os.path.basename(file_path)}"
-        )
-        
-        if st.button("Save Changes", key=f"save_{os.path.basename(file_path)}"):
-            try:
-                with open(file_path, 'w', encoding='utf-8') as f:
-                    f.write(edited_content)
-                st.success(f"Successfully saved changes to {file_path}")
-            except Exception as e:
-                st.error(f"Error saving changes: {e}")
+        if st.session_state.view_mode == 'view':
+            # Display as markdown when viewing
+            st.markdown(content)
+        else:
+            # Edit functionality
+            edited_content = st.text_area(
+                "Edit content",
+                content,
+                height=400,
+                key=f"edit_{os.path.basename(file_path)}"
+            )
+            
+            if st.button("Save Changes", key=f"save_{os.path.basename(file_path)}"):
+                try:
+                    with open(file_path, 'w', encoding='utf-8') as f:
+                        f.write(edited_content)
+                    st.success(f"Successfully saved changes to {file_path}")
+                except Exception as e:
+                    st.error(f"Error saving changes: {e}")
     except Exception as e:
         st.error(f"Error reading file: {e}")
 
@@ -236,17 +264,16 @@ def file_management_sidebar():
         for idx, file in enumerate(md_files):
             st.sidebar.markdown("---")  # Separator between files
             
-            # Display filename
-            st.sidebar.text(file)
+            # Display time
+            st.sidebar.text(get_time_display(file))
             
             # 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)
+                st.sidebar.markdown(download_link, unsafe_allow_html=True)
             
             # Action buttons in a row
-            col1, col2, col3 = st.sidebar.columns(3)
+            col1, col2, col3, col4 = st.sidebar.columns(4)
             
             with col1:
                 if st.button("📄 View", key=f"view_{idx}"):
@@ -272,7 +299,7 @@ def file_management_sidebar():
                         
                         # Perform AI lookup and save results
                         ai_result = perform_ai_lookup(full_prompt)
-                        saved_file = save_ai_interaction(f"Rerun of {file}", ai_result)
+                        saved_file = save_ai_interaction(content, ai_result, is_rerun=True)
                         
                         if saved_file:
                             st.success(f"Created fun version in {saved_file}")
@@ -281,6 +308,14 @@ def file_management_sidebar():
                             
                     except Exception as e:
                         st.error(f"Error during rerun: {e}")
+            
+            with col4:
+                if st.button("🗑️ Delete", key=f"delete_{idx}"):
+                    if delete_file(file):
+                        st.success(f"Deleted {file}")
+                        st.rerun()
+                    else:
+                        st.error(f"Failed to delete {file}")
 
         st.sidebar.markdown("---")
         # Option to create a new markdown file
@@ -331,10 +366,15 @@ def main():
     # Second priority: Display selected file content if any
     if st.session_state.selected_file:
         show_initial_content = False  # Hide initial content when showing file content
-        st.markdown(f"### Current File: {st.session_state.selected_file}")
-        display_file_content(st.session_state.selected_file)
+        if os.path.exists(st.session_state.selected_file):
+            st.markdown(f"### Current File: {st.session_state.selected_file}")
+            display_file_content(st.session_state.selected_file)
+        else:
+            st.error("Selected file no longer exists.")
+            st.session_state.selected_file = None
+            st.rerun()
 
-    # Show initial content: Either when first landing or when no interactive elements are active
+# Show initial content: Either when first landing or when no interactive elements are active
     if show_initial_content:
         # First show the clickable terms with links
         terms1 = extract_terms(Boxing_and_MMA_Commentary_and_Knowledge)