Update app.py

app.py

@@ -11,17 +11,17 @@ from PIL import Image
 import re
 from urllib.parse import quote
 
-# 1. Configuration
-Site_Name = '🤖🧠Ant🔬Claude📝'
-title="🤖🧠Ant🔬Claude📝"
+# 1. App Configuration
+Site_Name = '🤖🧠Claude35📝🔬'
+title="🤖🧠Claude35📝🔬"
 helpURL='https://huggingface.co/awacke1'
 bugURL='https://huggingface.co/spaces/awacke1'
 icons='🤖🧠🔬📝'
+
 st.set_page_config(
     page_title=title,
     page_icon=icons,
     layout="wide",
-    #initial_sidebar_state="expanded",
     initial_sidebar_state="auto",
     menu_items={
         'Get Help': helpURL,
@@ -37,36 +37,37 @@ client = anthropic.Anthropic(api_key=os.environ.get("ANTHROPIC_API_KEY"))
 if "chat_history" not in st.session_state:
     st.session_state.chat_history = []
 
-# Function to get file download link
+# Helper Functions: All Your Essentials 🚀
+
+# Function to get a file download link (because you deserve easy downloads 😎)
 def get_download_link(file_path):
     with open(file_path, "rb") as file:
         contents = file.read()
     b64 = base64.b64encode(contents).decode()
     file_name = os.path.basename(file_path)
-    return f'<a href="data:file/txt;base64,{b64}" download="{file_name}">Download {file_name}</a>'
+    return f'<a href="data:file/txt;base64,{b64}" download="{file_name}">Download {file_name}📂</a>'
 
-# Function to generate filename
+# Function to generate a filename based on prompt and time (because names matter 🕒)
 def generate_filename(prompt, file_type):
     central = pytz.timezone('US/Central')
     safe_date_time = datetime.now(central).strftime("%m%d_%H%M")
-    replaced_prompt = prompt.replace(" ", "_").replace("\n", "_")
-    safe_prompt = "".join(x for x in replaced_prompt if x.isalnum() or x == "_")[:90]
+    safe_prompt = re.sub(r'\W+', '_', prompt)[:90]
     return f"{safe_date_time}_{safe_prompt}.{file_type}"
 
-# Function to create file
+# Function to create and save a file (and avoid the black hole of lost data 🕳)
 def create_file(filename, prompt, response, should_save=True):
     if not should_save:
         return
     with open(filename, 'w', encoding='utf-8') as file:
         file.write(prompt + "\n\n" + response)
 
-# Function to load file content
+# Function to load file content (for revisiting the past 📜)
 def load_file(file_name):
     with open(file_name, "r", encoding='utf-8') as file:
         content = file.read()
     return content
 
-# Function to display glossary entity
+# Function to display handy glossary entity links (search like a pro 🔍)
 def display_glossary_entity(k):
     search_urls = {
         "🚀🌌ArXiv": lambda k: f"/?q={quote(k)}",
@@ -77,7 +78,7 @@ def display_glossary_entity(k):
     links_md = ' '.join([f"[{emoji}]({url(k)})" for emoji, url in search_urls.items()])
     st.markdown(f"**{k}** <small>{links_md}</small>", unsafe_allow_html=True)
 
-# Function to create zip of files
+# Function to create zip of files (because more is more 🧳)
 def create_zip_of_files(files):
     import zipfile
     zip_name = "all_files.zip"
@@ -86,15 +87,7 @@ def create_zip_of_files(files):
             zipf.write(file)
     return zip_name
 
-# Function to get zip download link
-def get_zip_download_link(zip_file):
-    with open(zip_file, 'rb') as f:
-        data = f.read()
-    b64 = base64.b64encode(data).decode()
-    href = f'<a href="data:application/zip;base64,{b64}" download="{zip_file}">Download All</a>'
-    return href
-
-# Function to create HTML for autoplaying and looping video
+# Function to create HTML for autoplaying and looping video (for the full cinematic effect 🎥)
 def get_video_html(video_path, width="100%"):
     video_url = f"data:video/mp4;base64,{base64.b64encode(open(video_path, 'rb').read()).decode()}"
     return f'''
@@ -104,7 +97,7 @@ def get_video_html(video_path, width="100%"):
     </video>
     '''
 
-# Function to create HTML for audio player
+# Function to create HTML for audio player (when life needs a soundtrack 🎶)
 def get_audio_html(audio_path, width="100%"):
     audio_url = f"data:audio/mpeg;base64,{base64.b64encode(open(audio_path, 'rb').read()).decode()}"
     return f'''
@@ -115,17 +108,14 @@ def get_audio_html(audio_path, width="100%"):
     '''
 
 
-
-# Streamlit app
+# Streamlit App Layout (like a house with better flow 🏡)
 def main():
     
-    # Sidebar
-    st.sidebar.title("🤖🧠Ant🔬Claude📝🔊 - Anthropic Claude 3.5 Sonnet")
+    # Sidebar with Useful Controls (All the VIP actions 🎛)
+    st.sidebar.title("🧠Claude📝")
     
-    # File management
     all_files = glob.glob("*.md")
-    all_files = [file for file in all_files if len(os.path.splitext(file)[0]) >= 10]
-    all_files.sort(key=lambda x: (os.path.splitext(x)[1], x), reverse=True)
+    all_files.sort(reverse=True)
 
     if st.sidebar.button("🗑 Delete All"):
         for file in all_files:
@@ -134,7 +124,7 @@ def main():
 
     if st.sidebar.button("⬇️ Download All"):
         zip_file = create_zip_of_files(all_files)
-        st.sidebar.markdown(get_zip_download_link(zip_file), unsafe_allow_html=True)
+        st.sidebar.markdown(get_download_link(zip_file), unsafe_allow_html=True)
 
     for file in all_files:
         col1, col2, col3, col4 = st.sidebar.columns([1,3,1,1])
@@ -153,10 +143,10 @@ def main():
                 os.remove(file)
                 st.rerun()
 
-    # Main content area
-    user_input = st.text_area("Enter your message:", height=100)
+    # Main Area: Chat with Claude (He’s a good listener 💬)
+    user_input = st.text_area("Message 📨:", height=100)
     
-    if st.button("Send"):
+    if st.button("Send 📨"):
         if user_input:
             response = client.messages.create(
                 model="claude-3-sonnet-20240229",
@@ -165,58 +155,56 @@ def main():
                     {"role": "user", "content": user_input}
                 ]
             )
-            
-            st.write("Claude's response:")
+            st.write("Claude's reply 🧠:")
             st.write(response.content[0].text)
 
-            # Save response to file
             filename = generate_filename(user_input, "md")
             create_file(filename, user_input, response.content[0].text)
 
-            # Add to chat history
             st.session_state.chat_history.append({"user": user_input, "claude": response.content[0].text})
 
-    # Display chat history
-    st.subheader("Chat History")
+    # Display Chat History (Never forget a good chat 💭)
+    st.subheader("Past Conversations 📜")
     for chat in st.session_state.chat_history:
-        st.text_area("You:", chat["user"], height=100, disabled=True)
-        st.text_area("Claude:", chat["claude"], height=200, disabled=True)
+        st.text_area("You said 💬:", chat["user"], height=100, disabled=True)
+        st.text_area("Claude replied 🤖:", chat["claude"], height=200, disabled=True)
         st.markdown("---")
 
-    # File content viewer/editor
+    # File Editor (When you need to tweak things ✏️)
     if hasattr(st.session_state, 'current_file'):
-        st.subheader(f"Editing: {st.session_state.current_file}")
-        new_content = st.text_area("File Content:", st.session_state.file_content, height=300)
-        if st.button("Save Changes"):
+        st.subheader(f"Editing: {st.session_state.current_file} 🛠")
+        new_content = st.text_area("File Content ✏️:", st.session_state.file_content, height=300)
+        if st.button("Save Changes 💾"):
             with open(st.session_state.current_file, 'w', encoding='utf-8') as file:
                 file.write(new_content)
-            st.success("File updated successfully!")
+            st.success("File updated successfully! 🎉")
 
-    # Image Gallery
-    st.subheader("Image Gallery")
+    # Image Gallery (For your viewing pleasure 📸)
+    st.subheader("Image Gallery 🖼")
     image_files = glob.glob("*.png") + glob.glob("*.jpg") + glob.glob("*.jpeg")
-    image_cols = st.slider("Image Gallery Columns", min_value=1, max_value=5, value=3)
+    image_cols = st.slider("Gallery Columns 🖼", min_value=1, max_value=15, value=5)
     cols = st.columns(image_cols)
     for idx, image_file in enumerate(image_files):
         with cols[idx % image_cols]:
             img = Image.open(image_file)
-            st.image(img, caption=image_file, use_column_width=True)
+            #st.image(img, caption=image_file, use_column_width=True)
+            st.image(img, use_column_width=True)
             display_glossary_entity(os.path.splitext(image_file)[0])
 
-    # Video Gallery
-    st.subheader("Video Gallery")
+    # Video Gallery (Let’s roll the tapes 🎬)
+    st.subheader("Video Gallery 🎥")
     video_files = glob.glob("*.mp4")
-    video_cols = st.slider("Video Gallery Columns", min_value=1, max_value=5, value=3)
+    video_cols = st.slider("Gallery Columns 🎬", min_value=1, max_value=5, value=3)
     cols = st.columns(video_cols)
     for idx, video_file in enumerate(video_files):
         with cols[idx % video_cols]:
             st.markdown(get_video_html(video_file, width="100%"), unsafe_allow_html=True)
             display_glossary_entity(os.path.splitext(video_file)[0])
 
-    # Audio Gallery
-    st.subheader("Audio Gallery")
+    # Audio Gallery (Tunes for the mood 🎶)
+    st.subheader("Audio Gallery 🎧")
     audio_files = glob.glob("*.mp3") + glob.glob("*.wav")
-    audio_cols = st.slider("Audio Gallery Columns", min_value=1, max_value=5, value=3)
+    audio_cols = st.slider("Gallery Columns 🎶", min_value=1, max_value=15, value=5)
     cols = st.columns(audio_cols)
     for idx, audio_file in enumerate(audio_files):
         with cols[idx % audio_cols]: