eeg / app.py
Samiraghafari's picture
Upload app.py
6b413dd verified
import joblib
import streamlit as st
import os
import time
import pandas as pd
from PIL import Image
st.header("Emotional Predictive Model")
@st.cache_data
def main_sheet(n):
sheet=pd.read_excel(n)
return sheet
sheets=main_sheet('describe.xlsx')
@st.cache_data
def main_test(n):
test=pd.read_excel(n)
return test
test=main_test('test.xlsx')
@st.cache_resource
def load_model(n):
model=joblib.load(n)
return model
model=load_model("EEGP.json")
good=[]
bad=[]
natural=[]
b=st.button("strat the test")
if b:
for i in range(1,30):
s=sheets.query(f"image=={i}")
chatbot_tiny_logo = Image.open(f"E:/samira/eeg/image/{i}.jpg")
st.image(chatbot_tiny_logo, caption=s["description"].values[0])
d=test.sample(1)
X=d.iloc[:,:-1]
st.write(X)
emotion=model.predict(X)
st.write(emotion)
st.success("signal was recieved and analysed")
time.sleep(2)
if emotion=='POSITIVE':
good.append(s["description"].values[0])
elif emotion=='NEGATIVE':
bad.append(s["description"].values[0])
else:
natural.append(s["description"].values[0])
else:
st.write("true the toggle")
st.header("Design guidline report")
new_data=pd.DataFrame(good)
st.write(good)