2b-agent / pages /04_Step 4: Story.py
dipta007's picture
Story fix
94546d6
raw
history blame contribute delete
No virus
2.77 kB
import streamlit as st
import pandas as pd
import random
from auth import check_password
if not check_password():
st.stop()
st.set_page_config(
page_title="ARL - Pilot 2a",
page_icon="πŸŽ†",
layout="wide",
)
df = pd.read_csv('./results.csv')
print(df.columns)
query = st.experimental_get_query_params()
if not query or 'id4' not in query or not query['id4']:
rnd_ind = random.randint(0, len(df) - 1)
# id = rnd_ind
id = 0
st.experimental_set_query_params(id4=id)
file_id = id
else:
file_id = int(query['id4'][0])
st.experimental_set_query_params(id4=file_id)
data = df.iloc[file_id]
# st.table(data)
def show_story():
st.subheader('Impacted Story')
number_of_story_lines = int(data['Input.number_of_lines'])
print(number_of_story_lines)
story = data['Input.story'].split('. ')
for i in range(number_of_story_lines):
row_name = f'Answer.line{i}_changed.on'
print(row_name, data[row_name])
if data[row_name] == 'true':
st.markdown(f'**Line {i+1}:** <mark style="background-color: white">{story[i]}</mark>', unsafe_allow_html=True)
else:
st.markdown(f'**Line {i+1}:** {story[i]}')
def show_data():
col1, col2 = st.columns(2)
with col1:
img1, img2, img3 = st.columns(3)
with img1:
st.image(data['Input.image1'])
with img2:
st.image(data['Input.image2'])
with img3:
st.image(data['Input.image3'])
st.subheader('Entity')
st.write(data['Input.entity'])
st.subheader('Agent')
st.write(data['Input.agent'])
st.subheader('Input Goal')
st.write(data['Input.goal'])
st.subheader('Story')
st.write(data['Input.story'])
with col2:
st.subheader('Goal')
st.write(data['Answer.goal'])
st.subheader('Condition Necessary')
st.write(data['Answer.condition'])
st.subheader('Altered Condition')
st.write(data['Answer.not_condition'])
st.subheader('Lead to Altered Condition')
st.write(data['Answer.not_condition_ep'])
print("*****", data['Answer.story_impact.yes'])
if data['Answer.story_impact.yes'] == 'true':
print("*****", data['Answer.story_impact.yes'])
show_story()
else:
st.subheader('Why no impact?')
st.text(data['Answer.why_no_impact'])
print(data)
from api import show_likert_scale
done = show_likert_scale('step_4', 'Rewritten Story', data['AssignmentId'])
if st.button("Next", disabled=not done):
id = file_id + 1
st.experimental_set_query_params(id4=id)
st.experimental_rerun()
# st.table(data)
show_data()