import gradio as gr
import joblib

# Chargez votre modèle à partir du fichier pkl
model = joblib.load('best_pipeline.pkl')

# Exemple de données pour un passager
passenger_data = {
    "Pclass": "1",  # Exemple : "1" pour 1ère classe
    "Sex": "male",  # Exemple : "male" ou "female"
    "Age": 30,  # Exemple : 30 ans
    "SibSp": 0,  # Exemple : 0
    "Parch": 0,  # Exemple : 0
    "Fare": 75.0,  # Exemple : 75.0
    "Name": "John Smith"  # Exemple : Nom du passager
}

def predict_survival(Pclass, Sex, Age, SibSp, Parch, Fare, Name):
    # Utilisez votre modèle pour faire des prédictions ici
    prediction = model.predict([Pclass, Sex, Age, SibSp, Parch, Fare])
    return 'Survécu' if prediction == 1 else 'Non survécu'

iface = gr.Interface(fn=predict_survival, inputs=["text", "text", "number", "number", "number", "number", "text"], outputs="text")
iface.launch()