Update app.py

app.py

@@ -12,6 +12,14 @@ import zipfile
 from PIL import Image
 
 EXCLUDED_FILES = ['app.py', 'requirements.txt', 'pre-requirements.txt', 'packages.txt', 'README.md','.gitattributes', "backup.py","Dockerfile"]
+# Emoji mapping for different file types
+FILE_EMOJIS = {
+    ".jpg": "🖼️", ".jpeg": "🖼️", ".png": "🖼️",
+    ".html": "🌐", ".htm": "🌐", ".txt": "📄",
+    ".pdf": "📚", ".doc": "📝", ".docx": "📝",
+    ".xls": "📊", ".xlsx": "📊", ".ppt": "📊", ".pptx": "📊",
+    # Add more mappings as needed
+}
 URLS = {
     "Lumiere": "https://lumiere-video.github.io/",
     "National Library of Medicine": "https://www.nlm.nih.gov/",
@@ -83,14 +91,18 @@ def download_html_and_files(url, subdir):
     html_content = requests.get(url).text
     soup = BeautifulSoup(html_content, 'html.parser')
     base_url = urllib.parse.urlunparse(urllib.parse.urlparse(url)._replace(path='', params='', query='', fragment=''))
+    file_urls = {}  # To store original URLs of files
     for link in soup.find_all('a'):
         file_url = urllib.parse.urljoin(base_url, link.get('href'))
         local_filename = os.path.join(subdir, urllib.parse.urlparse(file_url).path.split('/')[-1])
         if not local_filename.endswith('/') and local_filename != subdir:
             link['href'] = local_filename
-            download_file(file_url, local_filename)
+            if download_file(file_url, local_filename):
+                file_urls[local_filename] = file_url  # Store original URL
     with open(os.path.join(subdir, "index.html"), "w") as file:
         file.write(str(soup))
+    return file_urls
+
 
 def list_files(directory_path='.'):
     files = [f for f in os.listdir(directory_path) if os.path.isfile(os.path.join(directory_path, f))]
@@ -135,25 +147,21 @@ def show_file_operations(file_path, sequence_number):
 
 file_sequence_numbers = {}
 
-def show_file_content(file_path):
+def show_file_content(file_path, original_url):
     _, file_extension = os.path.splitext(file_path)
+    emoji = FILE_EMOJIS.get(file_extension, "📁")  # Default emoji for unknown file types
     try:
+        # Display file content and original URL with emoji
         if file_extension in ['.png', '.jpg', '.jpeg']:
             image_url = file_path.replace('File:','').replace('/','')
-            st.write('Image URL:' + image_url)
-            markdown_link = f"[![Image]({image_url})]({image_url})" #file_path
+            markdown_link = f"{emoji} [![Image]({image_url})]({original_url})"
             st.markdown(markdown_link, unsafe_allow_html=True)
-        elif file_extension in ['.md', '.markdown']:
-            with open(file_path, "r") as file:
-                content = file.read()
-            edited_content = st.text_area(f"Edit {os.path.basename(file_path)}", value=content, height=250)
-            if st.button(f"Save {os.path.basename(file_path)}"):
-                with open(file_path, "w") as file:
-                    file.write(edited_content)
-                st.success(f"Saved {os.path.basename(file_path)}!")
-        elif file_extension in ['.html', '.txt']:
+        elif file_extension in ['.html', '.htm', '.txt']:
             with open(file_path, "r") as file:
+                st.markdown(f"{emoji} [{os.path.basename(file_path)}]({original_url})", unsafe_allow_html=True)
                 st.markdown(file.read(), unsafe_allow_html=True)
+        else:
+            st.markdown(f"{emoji} [{os.path.basename(file_path)}]({original_url})", unsafe_allow_html=True)
     except Exception as e:
         st.error(f"Error reading file {file_path}: {e}")
 
@@ -231,14 +239,16 @@ def main():
                     json.dump(history, f)
 
         if st.sidebar.button('📥 Get All the Content', help="Download content from the selected URL"):
-            download_html_and_files(url, history[url])
-            show_download_links(history[url])
-    
+            file_urls = download_html_and_files(url, history[url])
+            for file in list_files(history[url]):
+                file_path = os.path.join(history[url], file)
+                original_url = file_urls.get(file_path, "#")
+                show_file_content(file_path, original_url)
+
         if st.sidebar.button('📂 Show Download Links', help="Show all available download links"):
             for subdir in history.values():
                 show_download_links(subdir)
 
-
         if st.sidebar.button("🗑 Delete All", help="Delete all downloaded content"):
             # Clear history file
             with open("history.json", "w") as f: