import gradio as gr
import requests

# Prediction function

def predict_risk(age, systolic, diastolic, bs, temp, heart_rate):
    input_data = {
        "Age": age,
        "SystolicBP": systolic,
        "DiastolicBP": diastolic,
        "BS": bs,
        "BodyTemp": temp,
        "HeartRate": heart_rate
    }
    try:
        response = requests.post("http://0.0.0.0:7860/predict", json=input_data)
        response.raise_for_status()
        result = response.json()
        return result.get("Risk", "Error: Unexpected response format")
    except Exception as e:
        return f"Error: {str(e)}"

# Gradio Interface

demo = gr.Interface(
    fn=predict_risk,
    inputs=[
        gr.Number(label="Age"),
        gr.Number(label="Systolic BP"),
        gr.Number(label="Diastolic BP"),
        gr.Number(label="Blood Sugar"),
        gr.Number(label="Body Temperature"),
        gr.Number(label="Heart Rate")
    ],
    outputs=gr.Text(label="Predicted Risk Level"),
    title="Pregnancy Risk Predictor",
    description="Enter patient vitals to get a low/medium/high risk assessment."
)

# Launch the interface
demo.launch(share=True)