Update app.py

app.py

@@ -99,13 +99,13 @@ def SpeechSynthesis(result):
 def generate_filename(prompt, file_type, original_name=None):
     central = pytz.timezone('US/Central')
     safe_date_time = datetime.now(central).strftime("%m%d_%H%M")
+    safe_prompt = re.sub(r'[<>:"/\\|?*\n]', ' ', prompt).strip()[:50]
     if original_name and file_type == "md":  # For images
         base_name = os.path.splitext(original_name)[0]
-        safe_prompt = re.sub(r'[<>:"/\\|?*\n]', ' ', prompt).strip()[:50]
         file_stem = f"{safe_date_time}_{safe_prompt}_{base_name}"[:100]  # Cap at 100 chars
         return f"{file_stem}.{file_type}"
-    safe_prompt = re.sub(r'[<>:"/\\|?*\n]', ' ', prompt).strip()[:100]  # Cap at 100 chars
-    return f"{safe_date_time}_{safe_prompt}.{file_type}"
+    file_stem = f"{safe_date_time}_{safe_prompt}"[:100]  # Cap at 100 chars
+    return f"{file_stem}.{file_type}"
 
 def create_and_save_file(content, file_type="md", prompt=None, original_name=None, should_save=True):
     if not should_save:
@@ -152,7 +152,7 @@ def process_image(image_input, user_prompt):
         temperature=0.0
     )
     image_response = response.choices[0].message.content
-    filename = generate_filename(user_prompt, "md", original_name)
+    filename = generate_filename(user_prompt, "md", original_name)  # Include prompt in filename
     create_and_save_file(image_response, "md", user_prompt, original_name, should_save=should_save)
     return image_response
 
@@ -428,10 +428,10 @@ def FileSidebar():
         st.sidebar.markdown(get_zip_download_link(zip_file), unsafe_allow_html=True)
     
     for file in all_files:
-        ext = os.path.splitext(file)[1]
+        ext = os.path.splitext(file)[1].lower()
         col1, col2, col3, col4, col5 = st.sidebar.columns([1, 6, 1, 1, 1])
         with col1:
-            icon = "📜" if ext == ".md" else "📄" if ext == ".pdf" else "🖼️" if ext == ".png" else "🎵" if ext in [".wav", ".mp3"] else "🎥"
+            icon = "📜" if ext == ".md" else "📄" if ext == ".pdf" else "🖼️" if ext in [".png", ".jpg", ".jpeg"] else "🎵" if ext in [".wav", ".mp3"] else "🎥" if ext == ".mp4" else "📎"
             if st.button(icon, key=f"view_{file}"):
                 with open(file, "rb") as f:
                     content = f.read()
@@ -441,16 +441,30 @@ def FileSidebar():
                 elif ext == ".pdf":
                     st.download_button("Download PDF", content, file, "application/pdf")
                     st.write("PDF Viewer not natively supported; download to view.")
-                elif ext == ".png":
+                elif ext in [".png", ".jpg", ".jpeg"]:
                     st.image(content, use_column_width=True)
+                elif ext in [".wav", ".mp3"]:
+                    st.audio(content)
+                elif ext == ".mp4":
+                    st.video(content)
         with col2:
             st.markdown(get_table_download_link(file), unsafe_allow_html=True)
         with col3:
             if st.button("📂", key=f"open_{file}"):
-                st.session_state.update({'filename': file, 'filetext': open(file, "r", encoding="utf-8").read()})
+                with open(file, "rb") as f:
+                    content = f.read()
+                if ext in [".md", ".pdf"]:
+                    st.text_area(f"Editing {file}", value=content.decode("utf-8"), height=300, key=f"edit_{file}")
+                elif ext in [".png", ".jpg", ".jpeg"]:
+                    st.image(content, use_column_width=True, caption=f"Viewing {file}")
+                elif ext in [".wav", ".mp3"]:
+                    st.audio(content, format=f"audio/{ext[1:]}")
+                elif ext == ".mp4":
+                    st.video(content, format="video/mp4")
         with col4:
             if st.button("▶️", key=f"run_{file}"):
-                process_text(open(file, "r", encoding="utf-8").read())
+                if ext == ".md":
+                    process_text(open(file, "r", encoding="utf-8").read())
         with col5:
             if st.button("🗑", key=f"delete_{file}"):
                 os.remove(file)
@@ -475,8 +489,8 @@ def get_table_download_link(file_path):
         data = f.read()
     b64 = base64.b64encode(data).decode()
     file_name = os.path.basename(file_path)
-    ext = os.path.splitext(file_name)[1]
-    mime_type = "text/markdown" if ext == ".md" else "application/pdf" if ext == ".pdf" else "image/png" if ext == ".png" else "audio/wav" if ext == ".wav" else "audio/mpeg" if ext == ".mp3" else "video/mp4" if ext == ".mp4" else "application/octet-stream"
+    ext = os.path.splitext(file_name)[1].lower()
+    mime_type = "text/markdown" if ext == ".md" else "application/pdf" if ext == ".pdf" else "image/png" if ext in [".png", ".jpg", ".jpeg"] else "audio/wav" if ext == ".wav" else "audio/mpeg" if ext == ".mp3" else "video/mp4" if ext == ".mp4" else "application/octet-stream"
     return f'<a href="data:{mime_type};base64,{b64}" download="{file_name}">{file_name}</a>'
 
 # Main Function