|
import gradio as gr |
|
import numpy as np |
|
import joblib |
|
|
|
|
|
model = joblib.load('logistic_regression_model.pkl') |
|
scaler = joblib.load('scaler.pkl') |
|
|
|
def initial_risk_check(features): |
|
features_scaled = scaler.transform([features]) |
|
prediction = model.predict_proba(features_scaled)[:, 1] |
|
high_risk_threshold = 0.75 |
|
moderate_risk_threshold = 0.5 |
|
|
|
if prediction >= high_risk_threshold: |
|
return "High risk. Please consult a doctor immediately.", prediction |
|
elif prediction >= moderate_risk_threshold: |
|
return "Moderate risk. Consider lifestyle changes and monitor your health closely.", prediction |
|
else: |
|
return "Low risk. Maintain a healthy lifestyle and schedule routine check-ups.", prediction |
|
|
|
def launch_interface(): |
|
with gr.Blocks(css=\" |
|
.gradio-container { |
|
font-family: Arial, sans-serif; |
|
background-color: #f9f9f9; |
|
color: #333; |
|
} |
|
.gr-button { |
|
background-color: #007bff; |
|
color: white; |
|
} |
|
.gr-button:hover { |
|
background-color: #0056b3; |
|
} |
|
.gr-textbox { |
|
margin-bottom: 1em; |
|
} |
|
\") as app: |
|
|
|
gr.Markdown(\" |
|
# Gestational Diabetes Risk Calculator |
|
This calculator helps you assess your risk for gestational diabetes. Enter your details below and get an initial risk assessment. |
|
If you are above medium risk, please provide more detailed information for a more accurate assessment. |
|
\") |
|
|
|
with gr.Row(): |
|
pregnancies = gr.Number(label="Number of Pregnancies") |
|
age = gr.Number(label="Age") |
|
bmi = gr.Number(label="BMI") |
|
detailed = gr.Checkbox(label="Click if above medium risk for more detailed input", value=False) |
|
|
|
with gr.Row(): |
|
ogtt = gr.Number(label="OGTT", visible=False) |
|
sys_bp = gr.Number(label="Systolic Blood Pressure", visible=False) |
|
hdl = gr.Number(label="HDL", visible=False) |
|
|
|
submit_button = gr.Button("Submit") |
|
output = gr.Textbox(label="Result") |
|
|
|
def update_output(pregnancies, age, bmi, detailed, ogtt, sys_bp, hdl): |
|
if detailed: |
|
features = [pregnancies, ogtt, sys_bp, bmi, hdl, age] |
|
else: |
|
features = [pregnancies, 120, 120, bmi, 0.5, age] |
|
result, prediction = initial_risk_check(features) |
|
return result |
|
|
|
def toggle_visibility(detailed): |
|
return { |
|
ogtt: gr.update(visible=detailed), |
|
sys_bp: gr.update(visible=detailed), |
|
hdl: gr.update(visible=detailed) |
|
} |
|
|
|
detailed.change( |
|
toggle_visibility, |
|
inputs=[detailed], |
|
outputs=[ogtt, sys_bp, hdl] |
|
) |
|
|
|
submit_button.click( |
|
update_output, |
|
inputs=[pregnancies, age, bmi, detailed, ogtt, sys_bp, hdl], |
|
outputs=output |
|
) |
|
|
|
app.launch() |
|
|
|
if __name__ == "__main__": |
|
launch_interface() |
|
|
