Update app.py

app.py

@@ -9,6 +9,7 @@ import json
 import uuid 
 import glob
 import zipfile
+from PIL import Image
 
 EXCLUDED_FILES = ['app.py', 'requirements.txt', 'pre-requirements.txt', 'packages.txt', 'README.md','.gitattributes', "backup.py","Dockerfile"]
 URLS = {
@@ -197,11 +198,29 @@ def get_download_link(file):
     return href
 
 
+def show_file_content_new_style(file_path):
+    _, file_extension = os.path.splitext(file_path)
+    try:
+        if file_extension in ['.png', '.jpg', '.jpeg']:
+            with open(file_path, "rb") as file:
+                img = Image.open(file)
+                st.image(img, caption=os.path.basename(file_path))
+        elif file_extension in ['.md', '.markdown']:
+            # ... [Markdown file handling code]
+        elif file_extension in ['.html', '.txt']:
+            # ... [HTML/Text file handling code]
+    except Exception as e:
+        st.error(f"Error reading file {file_path}: {e}")
+        
 def show_file_content(file_path):
     _, file_extension = os.path.splitext(file_path)
     try:
         if file_extension in ['.png', '.jpg', '.jpeg']:
-            st.image(file_path)
+            with open(file_path, "rb") as file:
+                img = Image.open(file)
+                st.image(img, caption=os.path.basename(file_path))
+        #if file_extension in ['.png', '.jpg', '.jpeg']:
+        #    st.image(file_path)
         elif file_extension in ['.md', '.markdown']:
             with open(file_path, "r") as file:
                 content = file.read()