Update app.py

app.py

@@ -639,29 +639,60 @@ def edit_file(file):
                 content = f.read()
             new_content = st.text_area("Edit HTML content", content, height=400)
             if st.button("Save HTML"):
-                with open(file, 'w', encoding='utf-8') as f:
-                    f.write(new_content)
-                st.success("File updated successfully!")
+                new_file_path = update_file_with_timestamp(file, new_content)
+                st.success(f"File updated successfully! New file: {new_file_path}")
         else:  # For .md and other text files
             with open(file, 'r', encoding='utf-8') as f:
                 content = f.read()
             new_content = st.text_area("Edit content", content, height=400)
             if st.button("Save"):
-                with open(file, 'w', encoding='utf-8') as f:
-                    f.write(new_content)
-                st.success("File updated successfully!")
+                new_file_path = update_file_with_timestamp(file, new_content)
+                st.success(f"File updated successfully! New file: {new_file_path}")
     except Exception as e:
         st.error(f"Error editing {file}: {str(e)}")
-
+        
 def display_pdf(file_path):
     try:
         with open(file_path, "rb") as f:
-            base64_pdf = base64.b64encode(f.read()).decode('utf-8')
-        pdf_display = f'<iframe src="data:application/pdf;base64,{base64_pdf}" width="100%" height="800" type="application/pdf"></iframe>'
-        st.markdown(pdf_display, unsafe_allow_html=True)
+            pdf_reader = PyPDF2.PdfReader(f)
+            num_pages = len(pdf_reader.pages)
+            
+            st.write(f"Total Pages: {num_pages}")
+            
+            for page_num in range(num_pages):
+                page = pdf_reader.pages[page_num]
+                st.subheader(f"Page {page_num + 1}")
+                st.write(page.extract_text())
+                
+                if page_num < num_pages - 1:
+                    st.markdown("---")  # Add a separator between pages
     except Exception as e:
         st.error(f"Error displaying PDF {file_path}: {str(e)}")
 
+
+
+def save_file_with_timestamp(file_path, content, file_type):
+    directory, filename = os.path.split(file_path)
+    name, ext = os.path.splitext(filename)
+    timestamp = datetime.datetime.now().strftime("%m%d_%H%M")
+    new_filename = f"{timestamp}_{name}{ext}"
+    new_file_path = os.path.join(directory, new_filename)
+    
+    if file_type == 'pdf':
+        with open(new_file_path, 'wb') as f:
+            f.write(content)
+    else:  # For HTML and other text files
+        with open(new_file_path, 'w', encoding='utf-8') as f:
+            f.write(content)
+    
+    return new_file_path
+
+def update_file_with_timestamp(file_path, new_content):
+    file_type = os.path.splitext(file_path)[1].lower()[1:]  # Get file extension without the dot
+    new_file_path = save_file_with_timestamp(file_path, new_content, file_type)
+    os.remove(file_path)  # Remove the old file
+    return new_file_path
+
 def display_html(file_path):
     try:
         with open(file_path, 'r', encoding='utf-8') as f: