Update app.py

app.py

@@ -200,10 +200,21 @@ def get_download_link(file):
 def show_file_content(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))
+            image_url = file_path
+            # Check if the base_url needs to be prepended
+            if base_url and not file_path.startswith(("http://", "https://")):
+                image_url = os.path.join(base_url, file_path)
+            
+            # Create a Markdown link to view the image
+            markdown_link = f"[![View Image]({image_url})]({image_url})"
+            st.markdown(markdown_link, unsafe_allow_html=True)
+        
+        #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))
         #if file_extension in ['.png', '.jpg', '.jpeg']:
         #    st.image(file_path)
         elif file_extension in ['.md', '.markdown']:
@@ -217,6 +228,11 @@ def show_file_content(file_path):
         elif file_extension in ['.html', '.txt']:
             with open(file_path, "r") as file:
                 st.markdown(file.read(), unsafe_allow_html=True)
+        elif file_extension in ['.pdf']:
+            pdf_file = open(file_path, "rb")
+            base64_pdf = base64.b64encode(pdf_file.read()).decode('utf-8')
+            pdf_display = f'<iframe src="data:application/pdf;base64,{base64_pdf}" width="700" height="1000" type="application/pdf"></iframe>'
+            st.markdown(pdf_display, unsafe_allow_html=True)
     except Exception as e:
         st.error(f"Error reading file {file_path}: {e}")