1

seg2med_app/app_utils/image_utils.py

@@ -13,10 +13,32 @@ from io import BytesIO
 default_orientation_type = 'transpose'
 default_plt_origin_type = 'upper'
 
-def image_to_base64(img):
+     
+def image_to_base64(img, width=200):
     buffered = BytesIO()
-    img.save(buffered, format="JPEG")
-    return base64.b64encode(buffered.getvalue()).decode()
+    if img.mode == "RGBA":
+        img.save(buffered, format="PNG")
+        result = base64.b64encode(buffered.getvalue()).decode()
+        st.markdown(
+            f"""
+            <div style='text-align: center;'>
+                <img src="data:image/png;base64,{result}" width="{width}">
+            </div>
+            """,
+            unsafe_allow_html=True
+            )
+    else:
+        img.save(buffered, format="JPEG")
+        result = base64.b64encode(buffered.getvalue()).decode()
+        st.markdown(
+            f"""
+            <div style='text-align: center;'>
+                <img src="data:image/jpeg;base64,{result}" width="{width}">
+            </div>
+            """,
+            unsafe_allow_html=True
+            )
+        
 
 def processing_slice_to_right_orientation(img_slice, type=default_orientation_type):
     if type == 'transpose':

tutorial8_app.py

@@ -66,24 +66,8 @@ def reset_app():
     print("App has been reset. Login information is preserved.")
     st._rerun()
     
-image = Image.open(os.path.join(app_root, "frankenstein.jpg"))
-st.markdown(
-    f"""
-    <div style='text-align: center;'>
-        <img src="data:image/jpeg;base64,{image_to_base64(image)}" width="150">
-    </div>
-    """,
-    unsafe_allow_html=True
-)
-# copyright for logo
-st.markdown(
-    """
-    <p style="font-size:10px; color:gray; text-align: right;">
-        Logo Frankenstein – Designed by <a href="https://www.freepik.com" target="_blank" style="color:gray;">Freepik</a>
-    </p>
-    """,
-    unsafe_allow_html=True
-)
+image = Image.open(os.path.join(app_root, "Frankenstein0.png"))
+image_to_base64(image)
 
 
 st.title("\U0001F9E0 Frankenstein - multimodal medical image generation")