from traning_zone.classe_prediction.prediction_classe import *
import streamlit as st 
import plotly.express as px
import warnings
warnings.filterwarnings("ignore")
import matplotlib.pyplot as plt
import seaborn as sns
sns.set()
sns.set(rc={'figure.figsize':(14.7,10.27)})

st.title("Classification")


def get_data() :
    data = None
    uploaded_file = st.file_uploader("Choose a CSV file with the separator ',' ", type=["csv"])
    
    if uploaded_file is not None:
            data = pd.read_csv(uploaded_file,dtype=str,  sep=",")
            return data
           

inputs = get_data()
inputs
if inputs != None :
    X = inputs.DESCRIPTION
    pred = PredictionV(X)
    data = pred.prediction("spacy_spacy") 
    st.dataframe(data)