Spaces:
Runtime error
Runtime error
| import streamlit as st | |
| import mahotas as mh | |
| import pandas as pd | |
| import plotly.express as px | |
| import urllib.request | |
| from skimage import io | |
| # Define a list of medical conditions | |
| conditions = [ | |
| {"name": "Depression", "test_for": "Patient Health Questionnaire-9 (PHQ-9)"}, | |
| {"name": "Anxiety", "test_for": "Generalized Anxiety Disorder-7 (GAD-7)"}, | |
| {"name": "Diabetes", "test_for": "Hemoglobin A1C test"}, | |
| {"name": "Hypertension", "test_for": "Blood pressure measurement"}, | |
| {"name": "Asthma", "test_for": "Pulmonary function test"}, | |
| {"name": "Cancer", "test_for": "Biopsy or imaging tests (e.g., CT scan, MRI)"}, | |
| {"name": "Arthritis", "test_for": "X-ray, MRI, or ultrasound"}, | |
| {"name": "Heart disease", "test_for": "Electrocardiogram (ECG)"}, | |
| {"name": "Obesity", "test_for": "Body mass index (BMI)"}, | |
| {"name": "Substance use disorder", "test_for": "Substance Abuse Subtle Screening Inventory (SASSI)"} | |
| ] | |
| # Define a function to process images using Mahotas | |
| def process_image(image): | |
| # Convert the image to grayscale | |
| grayscale_image = mh.colors.rgb2gray(image) | |
| # Apply a Gaussian filter to the image to reduce noise | |
| filtered_image = mh.gaussian_filter(grayscale_image, 4) | |
| # Threshold the image to create a binary image | |
| binary_image = filtered_image > mh.otsu(filtered_image) | |
| # Compute the connected components in the binary image | |
| labels, num_labels = mh.label(binary_image) | |
| # Compute the size of each connected component | |
| sizes = mh.labeled.labeled_size(labels) | |
| # Sort the sizes in descending order | |
| sorted_sizes = sorted(sizes, reverse=True) | |
| # Return the top 10 sizes | |
| return sorted_sizes[:10] | |
| # Define the Streamlit app | |
| def app(): | |
| # Add a title to the app | |
| st.title("Mahotas Demo") | |
| # Add a sidebar to the app | |
| st.sidebar.title("Medical Conditions") | |
| selected_condition = st.sidebar.selectbox("Select a condition", [c["name"] for c in conditions]) | |
| # Get the selected condition | |
| condition = next(c for c in conditions if c["name"] == selected_condition) | |
| # Display the selected condition | |
| st.header(condition["name"]) | |
| st.write("Test for:", condition["test_for"]) | |
| # Load an example medical image | |
| if selected_condition == "Depression": | |
| image_url = "https://i.imgur.com/kPQoD8C.jpg" | |
| elif selected_condition == "Anxiety": | |
| image_url = "https://i.imgur.com/ZWyKjJN.jpg" | |
| elif selected_condition == "Diabetes": | |
| image_url = "https://i.imgur.com/1gOEMO5.jpg" | |
| elif selected_condition == "Hypertension": | |
| image_url = "https://i.imgur.com/BoSUwio.jpg" | |
| elif selected_condition == "Asthma": | |
| image_url = "https://i.imgur.com/BLKjzJF.jpg" | |
| elif selected_condition == "Cancer": | |
| image_url = "https://i.imgur.com/nq3vV8.jpg" | |
| elif selected_condition == "Arthritis": | |
| image_url = "https://i.imgur.com/ffzd6Fo.jpg" | |
| elif selected_condition == "Heart disease": | |
| image_url = "https://i.imgur.com/1I7axhd.jpg" | |
| elif selected_condition == "Obesity": | |
| image_url = "https://i.imgur.com/nZ1EjJr.jpg" | |
| else: | |
| image_url = "https://i.imgur.com/RUBZOWF.jpg" | |
| image = io.imread(image_url) | |
| # Process the image using Mahotas | |
| sizes = process_image(image) | |
| # Display the top 10 connected component sizes | |
| df = pd.DataFrame({"Size": sizes}) | |
| st.write(df) | |
| # Create a sunburst chart using Plotly | |
| fig = px.sunburst( | |
| df, | |
| path=["Size"], | |
| values="Size", | |
| color="Size", | |
| color_continuous_scale="blues" | |
| ) | |
| st.plotly_chart(fig) | |
| st.markdown(""" | |
| # Alternate Image Links Per Condition: | |
| Depression: https://www.pexels.com/photo/woman-sitting-on-grass-field-while-holding-her-head-7127866/ | |
| Anxiety: https://www.pexels.com/photo/woman-sitting-on-rock-and-looking-at-the-ocean-7119798/ | |
| Diabetes: https://www.pexels.com/photo/man-taking-blood-sugar-test-4050305/ | |
| Hypertension: https://www.pexels.com/photo/woman-measuring-blood-pressure-with-sphygmomanometer-5691686/ | |
| Asthma: https://www.pexels.com/photo/woman-having-asthma-attack-in-park-7127511/ | |
| Cancer: https://www.pexels.com/photo/close-up-of-pink-ribbon-on-cancer-awareness-banner-4219366/ | |
| Arthritis: https://www.pexels.com/photo/man-with-back-pain-lying-on-bed-4050323/ | |
| Heart disease: https://www.pexels.com/photo/woman-touching-chest-during-chest-pain-7127487/ | |
| Obesity: https://www.pexels.com/photo/woman-in-black-pants-lying-on-bed-7127516/ | |
| """) | |
| app() |