Spaces:
Running
Running
| import os | |
| import datetime | |
| import streamlit as st | |
| hide_streamlit_style = """ | |
| <style> | |
| footer {visibility: hidden;} | |
| </style> | |
| """ | |
| st.markdown(hide_streamlit_style, unsafe_allow_html=True) | |
| # Create a 'brain' folder in the current directory if it doesn't exist | |
| if not os.path.exists("brain"): | |
| os.makedirs("brain") | |
| # Name of the single journal file | |
| journal_file = "brain/brain_journal.txt" | |
| def parse_date(date_string): | |
| try: | |
| return datetime.datetime.strptime(date_string, "%m-%d-%Y %A") | |
| except ValueError: | |
| return datetime.datetime.strptime(date_string, "%m-%d-%Y") | |
| def get_journal_entries(): | |
| entries = [] | |
| if not os.path.exists(journal_file): | |
| return entries | |
| with open(journal_file, "r", encoding="utf-8") as f: | |
| for line in f: | |
| if line.startswith("Date: "): | |
| entry_date = parse_date(line[6:].strip()) | |
| entries.append(entry_date) | |
| entries.sort(reverse=True) | |
| return entries | |
| def read_entry(date): | |
| content = "" | |
| with open(journal_file, "r", encoding="utf-8") as f: | |
| lines = f.readlines() | |
| start_reading = False | |
| for line in lines: | |
| if line.startswith("Date: ") and start_reading: | |
| break | |
| if start_reading: | |
| content += line | |
| if line.startswith("Date: ") and date == parse_date(line[6:].strip()): | |
| start_reading = True | |
| return content | |
| def write_entry(date, content): | |
| new_entry = f"\nDate: {date}\n{content}\n\n" | |
| # If the journal file does not exist, create it with the new entry | |
| if not os.path.exists(journal_file): | |
| with open(journal_file, "w", encoding="utf-8") as f: | |
| f.write(new_entry) | |
| else: | |
| with open(journal_file, "r", encoding="utf-8") as f: | |
| lines = f.readlines() | |
| # Remove existing entry if present | |
| lines_to_remove = set() | |
| removing_entry = False | |
| for i, line in enumerate(lines): | |
| if line.startswith("Date: "): | |
| if date == line[6:].strip(): | |
| removing_entry = True | |
| lines_to_remove.add(i) | |
| else: | |
| removing_entry = False | |
| if removing_entry: | |
| lines_to_remove.add(i) | |
| lines = [line for i, line in enumerate(lines) if i not in lines_to_remove] | |
| # Find the correct position for the new entry based on its date | |
| new_entry_date = parse_date(date) | |
| position = None | |
| for i, line in enumerate(lines): | |
| if line.startswith("Date: "): | |
| entry_date = parse_date(line[6:].strip()) | |
| if new_entry_date < entry_date: | |
| position = i | |
| break | |
| # Insert the new entry at the correct position | |
| if position is None: | |
| lines.append(new_entry) | |
| else: | |
| lines.insert(position, new_entry) | |
| # Write the updated journal entries to the file | |
| with open(journal_file, "w", encoding="utf-8") as f: | |
| f.writelines(lines) | |
| st.title("Digital Brain Journal Entry ✍️") | |
| st.write("Write a diary journal entry or edit an existing one by selecting on the date picker.") | |
| selected_date = st.date_input("Select the date for the journal entry:", value=datetime.date.today()) | |
| formatted_date = selected_date.strftime("%m-%d-%Y %A") | |
| st.write(f"Selected date: {formatted_date}") | |
| entry = "" | |
| if selected_date in get_journal_entries(): | |
| entry = read_entry(selected_date) | |
| new_entry = st.text_area("Write your journal entry:", entry) | |
| if st.button("Submit"): | |
| write_entry(formatted_date, new_entry) | |
| st.success("Journal entry saved successfully!") | |
| st.header("Previous Journal Entries") | |
| entries = get_journal_entries() | |
| if entries: | |
| selected_entry_date = st.selectbox("Select an entry to view or edit:", entries, format_func=lambda x: x.strftime("%m-%d-%Y %A")) | |
| if st.button("Load Entry"): | |
| entry_text = read_entry(selected_entry_date) | |
| st.write(f"**{selected_entry_date.strftime('%m-%d-%Y %A')}**") | |
| st.markdown(entry_text.replace("\n", "<br>"), unsafe_allow_html=True) | |
| else: | |
| st.write("No previous entries found.") | |
| st.markdown("---") | |
| st.markdown("") | |
| st.markdown("<p style='text-align: center'><a href='https://github.com/Kaludii'>Github</a> | <a href='https://huggingface.co/Kaludi'>HuggingFace</a></p>", unsafe_allow_html=True) |