Update app.py

app.py

@@ -563,183 +563,110 @@ def get_file_size(file_path):
     
 
 def FileSidebar():
-    # File Sidebar for files 🌐View, 📂Open, ▶️Run, and 🗑Delete per file
+    # File Sidebar for files 🌐View, 📂Edit, and 🗑Delete per file
     all_files = glob.glob("*.md") + glob.glob("*_abstract.html") + glob.glob("*_abstract.pdf")
-    all_files = [file for file in all_files if len(os.path.splitext(file)[0]) >= 10 and os.path.basename(file)]  # exclude files with short names and empty names
-    all_files.sort(key=lambda x: (os.path.splitext(x)[1], x), reverse=True)  # sort by filename length which puts similar prompts together
+    all_files = [file for file in all_files if os.path.basename(file) and len(os.path.splitext(file)[0]) >= 10]
+    all_files.sort(key=lambda x: (os.path.splitext(x)[1], x), reverse=True)
 
     # ⬇️ Download
     Files1, Files2 = st.sidebar.columns(2)
     with Files1:
         if st.button("🗑 Delete All"):
             for file in all_files:
-                os.remove(file)
+                try:
+                    os.remove(file)
+                except Exception as e:
+                    st.error(f"Error deleting {file}: {str(e)}")
             st.rerun()
     with Files2:
         if st.button("⬇️ Download"):
             zip_file = create_zip_of_files(all_files)
             st.sidebar.markdown(get_zip_download_link(zip_file), unsafe_allow_html=True)
-    file_contents = ''
-    file_name = ''
-    next_action = ''
 
-    # Add files 🌐View, 📂Open, ▶️Run, and 🗑Delete per file
+    # Add files 🌐View, 📂Edit, and 🗑Delete per file
     for file in all_files:
-        # Generate a unique timestamp for each iteration
         timestamp = time.strftime("%H%M%S%f")[:-4]  # HHMMSSNN format
         
-        col1, col2, col3, col4, col5 = st.sidebar.columns([1,6,1,1,1])  # adjust the ratio as needed
+        col1, col2, col3, col4 = st.sidebar.columns([3,1,1,1])
         with col1:
-            if st.button("🌐", key=f"view_{timestamp}_{file}"):  # view emoji button
-                file_extension = os.path.splitext(file)[1].lower()
-                if file_extension == '.pdf':
-                    display_pdf(file)
-                elif file_extension == '.html':
-                    display_html(file)
-                else:
-                    file_contents = load_file(file)
-                    st.markdown(file_contents)
-                file_name = file
-                next_action = 'md'
-                st.session_state['next_action'] = next_action
+            st.markdown(f"**{file}**")
         with col2:
-            st.markdown(get_table_download_link(file), unsafe_allow_html=True)
+            if st.button("🌐", key=f"view_{timestamp}_{file}"):
+                view_file(file)
         with col3:
-            if st.button("📂", key=f"open_{timestamp}_{file}"):  # open emoji button
-                file_contents = load_file(file)
-                file_name = file
-                next_action = 'open'
-                st.session_state['lastfilename'] = file
-                st.session_state['filename'] = file
-                st.session_state['filetext'] = file_contents
-                st.session_state['next_action'] = next_action
+            if st.button("📂", key=f"edit_{timestamp}_{file}"):
+                edit_file(file)
         with col4:
-            if st.button("▶️", key=f"read_{timestamp}_{file}"):  # search emoji button
-                file_contents = load_file(file)
-                file_name = file
-                next_action = 'search'
-                st.session_state['next_action'] = next_action
-        with col5:
             if st.button("🗑", key=f"delete_{timestamp}_{file}"):
-                os.remove(file)
-                file_name = file
+                try:
+                    os.remove(file)
+                    st.success(f"Deleted {file}")
+                except Exception as e:
+                    st.error(f"Error deleting {file}: {str(e)}")
                 st.rerun()
-                next_action = 'delete'
-                st.session_state['next_action'] = next_action
 
-    
+def view_file(file):
+    file_extension = os.path.splitext(file)[1].lower()
+    try:
+        if file_extension == '.pdf':
+            display_pdf(file)
+        elif file_extension == '.html':
+            display_html(file)
+        else:  # For .md and other text files
+            with open(file, 'r', encoding='utf-8') as f:
+                content = f.read()
+            st.markdown(content)
+    except Exception as e:
+        st.error(f"Error viewing {file}: {str(e)}")
+
+def edit_file(file):
+    file_extension = os.path.splitext(file)[1].lower()
+    try:
+        if file_extension == '.pdf':
+            st.warning("PDF files cannot be edited directly. Please download and edit locally.")
+        elif file_extension == '.html':
+            with open(file, 'r', encoding='utf-8') as f:
+                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!")
+        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!")
+    except Exception as e:
+        st.error(f"Error editing {file}: {str(e)}")
+
 def display_pdf(file_path):
-    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="800" height="800" type="application/pdf"></iframe>'
-    st.markdown(pdf_display, unsafe_allow_html=True)
+    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)
+    except Exception as e:
+        st.error(f"Error displaying PDF {file_path}: {str(e)}")
 
 def display_html(file_path):
-    with open(file_path, 'r', encoding='utf-8') as f:
-        html_content = f.read()
-    st.components.v1.html(html_content, height=800, scrolling=True)
-
-                
-    # 🚩File duplicate detector - useful to prune and view all.  Pruning works well by file size detection of two similar and flags the duplicate.
-    file_sizes = [get_file_size(file) for file in all_files]
-    previous_size = None
-    st.sidebar.title("File Operations")
-    for file, size in zip(all_files, file_sizes):
-        duplicate_flag = "🚩" if size == previous_size else ""
-        with st.sidebar.expander(f"File: {file} {duplicate_flag}"):
-            st.text(f"Size: {size} bytes")
-
-            if st.button("View", key=f"view_{file}"):
-                try:
-                    with open(file, "r", encoding='utf-8') as f:  # Ensure the file is read with UTF-8 encoding
-                        file_content = f.read()
-                    st.code(file_content, language="markdown")
-                except UnicodeDecodeError:
-                    st.error("Failed to decode the file with UTF-8. It might contain non-UTF-8 encoded characters.")
-
-            if st.button("Delete", key=f"delete3_{file}"):
-                os.remove(file)
-                st.rerun()
-        previous_size = size  # Update previous size for the next iteration
-
-    if len(file_contents) > 0:
-        if next_action=='open':  # For "open", prep session state if it hasn't been yet
-            if 'lastfilename' not in st.session_state:
-                st.session_state['lastfilename'] = ''
-            if 'filename' not in st.session_state:
-                st.session_state['filename'] = ''
-            if 'filetext' not in st.session_state:
-                st.session_state['filetext'] = ''
-            open1, open2 = st.columns(spec=[.8,.2])
-            
-            with open1:
-                # Use onchange functions to autoexecute file name and text save functions.
-                file_name_input = st.text_input(key='file_name_input', on_change=SaveFileNameClicked, label="File Name:",value=file_name )
-                file_content_area = st.text_area(key='file_content_area', on_change=SaveFileTextClicked, label="File Contents:", value=file_contents, height=300)
-
-                ShowButtons = False  #  Having buttons is redundant.  They work but if on change event seals the deal so be it - faster save is less impedence - less context breaking
-                if ShowButtons:
-                    bp1,bp2 = st.columns([.5,.5])
-                    with bp1:
-                        if st.button(label='💾 Save Name'):
-                            SaveFileNameClicked()
-                    with bp2:
-                        if st.button(label='💾 Save File'):
-                            SaveFileTextClicked()
-                
-                new_file_content_area = st.session_state['file_content_area']
-                if new_file_content_area != file_contents:
-                    st.markdown(new_file_content_area) #changed
-
-            if st.button("🔍 Run AI Meta Strategy", key="filecontentssearch"):
-                #search_glossary(file_content_area)
-                filesearch = PromptPrefix + file_content_area
-                st.markdown(filesearch)
-                
-                if st.button(key=rerun, label='🔍AI Search' ):
-                    search_glossary(filesearch)
-
-        if next_action=='md':
-            st.markdown(file_contents)
-            SpeechSynthesis(file_contents)
-
-            buttonlabel = '🔍Run'
-            if st.button(key='Runmd', label = buttonlabel):
-                MODEL = "gpt-4o-2024-05-13"
-                openai.api_key = os.getenv('OPENAI_API_KEY')
-                openai.organization = os.getenv('OPENAI_ORG_ID')
-                client = OpenAI(api_key= os.getenv('OPENAI_API_KEY'), organization=os.getenv('OPENAI_ORG_ID'))
-                st.session_state.messages.append({"role": "user", "content": transcript})
-                with st.chat_message("user"):
-                    st.markdown(transcript)
-                with st.chat_message("assistant"):
-                    completion = client.chat.completions.create(
-                        model=MODEL,
-                        messages = st.session_state.messages,
-                        stream=True
-                    )
-                    response = process_text2(text_input=prompt)
-                st.session_state.messages.append({"role": "assistant", "content": response})
-                #try:
-                #search_glossary(file_contents)
-            #except:
-                #st.markdown('GPT is sleeping.  Restart ETA 30 seconds.')
-
-        if next_action=='search':
-            file_content_area = st.text_area("File Contents:", file_contents, height=500)
-            user_prompt = file_contents
-        #try:
-            #search_glossary(file_contents)
-            filesearch = PromptPrefix2 + file_content_area
-            st.markdown(filesearch)
-            if st.button(key=rerun, label='🔍Re-Code' ):
-                #search_glossary(filesearch)
-                search_arxiv(filesearch)
-                    
-        #except:
-            #st.markdown('GPT is sleeping.  Restart ETA 30 seconds.')
-    # ----------------------------------------------------- File Sidebar for Jump Gates ------------------------------------------
+    try:
+        with open(file_path, 'r', encoding='utf-8') as f:
+            html_content = f.read()
+        st.components.v1.html(html_content, height=800, scrolling=True)
+    except Exception as e:
+        st.error(f"Error displaying HTML {file_path}: {str(e)}")
 
+def get_file_size(file_path):
+    try:
+        return os.path.getsize(file_path)
+    except Exception as e:
+        print(f"Error getting file size for {file_path}: {str(e)}")
+        return 0
+        
 # Randomly select a title
 titles = [
     "🧠🎭 Semantic Symphonies 🎹🎸 & Episodic Encores 🥁🎻",