Spaces:
Sleeping
Sleeping
File size: 4,217 Bytes
8e0b903 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 |
import os
import sys
import tempfile
import pyvista as pv
import streamlit as st
import subprocess
from stpyvista import stpyvista
from PIL import Image
from streamlit_image_zoom import image_zoom
######### Import PATH #######
script_dir = os.path.dirname(os.path.abspath(__file__))
imgto3d_path = os.path.join(script_dir, '..', 'image_to_3D')
visualize_x3d = os.path.join(script_dir, '..', 'visualize_x3d')
sys.path.append(imgto3d_path)
sys.path.append(visualize_x3d)
######### Preprocessing DICOM Image #########
from visualize_x3d.dicom_toImg import convert_dcm_to_png
######### Convert Image to 3D #########
from image_to_3D.ui import process_image
try:
import torchmcubes
import torch
import torchvision
import fpdf
except ImportError:
subprocess.check_call(['pip', 'install', 'git+https://github.com/tatsy/torchmcubes.git'])
subprocess.check_call(['pip', 'install','fpdf'])
status_file=False
st.markdown("<h1 style='text-align: center;'>Chào mừng tới Lung x3D Viewer 🎈</h1>", unsafe_allow_html=True)
col_1, col_3,col_2 = st.columns([7.4,0.3,10])
with st.sidebar:
uploaded_files = st.file_uploader("Truyền ảnh vào đây", type=["png", "jpg", "jpeg", "dicom"], key="Uploader_sidebar")
with col_1:
with st.expander("Intructions"):
st.markdown("1. Để mở ảnh thì bạn có thể ấn vào chọn **Browse Files** và duyệt file. Hoặc bạn có thể chọn **Drag and Drop** và chuyển file vào đó.")
st.markdown("2. Sau khi đã truyền ảnh vào hãy ấn **Convert to 3D**. Sau đó vui lòng đợi một lúc để hệ thống xử lý ảnh và đưa ra hình 3D")
if uploaded_files:
# Handle single file or multiple files
if not isinstance(uploaded_files, list):
uploaded_files = [uploaded_files] # Convert single file to list
for uploaded_file in uploaded_files:
# Check the type of uploaded file
if hasattr(uploaded_file, 'name'):
file_type = uploaded_file.name.split('.')[-1].lower()
else:
file_type = 'unknown'
if file_type in ["jpg", "jpeg", "png"]:
img = Image.open(uploaded_file)
st.markdown("<div style='display: flex; justify-content: center;'>", unsafe_allow_html=True)
width, height = img.size
image_zoom(img, mode="both")
st.markdown("</div>", unsafe_allow_html=True)
btn_convert=st.button("Convert to 3D")
if btn_convert:
output_file = process_image(uploaded_file, output_filename='temp_image_3d')
status_file=True
elif file_type in ["dicom", "dcm"]:
convert_dcm_to_png(uploaded_file)
img = Image.open('a.png').convert('RGB')
st.markdown("<div style='display: flex; justify-content: center;'>", unsafe_allow_html=True)
width, height = img.size
image_zoom(img, mode="both", size=(width // 5, height // 5), keep_aspect_ratio=True,
zoom_factor=4.0, increment=0.2)
st.markdown("</div>", unsafe_allow_html=True)
btn_convert = st.button("Convert to 3D")
if btn_convert:
output_file = process_image("a.png", output_filename='temp_image_3d')
status_file=True
else:
st.info("Bạn cần truyền ảnh và quét ảnh trước khi xem ảnh 3D")
############### VISUALIZE_x3D file #########################
from visualize_x3d.visualize_3d_file import option_stl_1
from visualize_x3d.visualize_3d_file import option_stl_2
from visualize_x3d.glt2stl import convertgtb2stl
with col_2:
with st.expander("Intructions"):
st.markdown("Truyền ảnh vào sau đó ấn nút sau để xem ảnh 3D")
if status_file==False:
option_stl_1()
if status_file==True:
file_path=r"temp_image_3d.glb"
convertgtb2stl(file_path,"temp_image_3d.stl")
option_stl_2("temp_image_3d.stl")
|