ArvindSelvaraj's picture
Update app.py
00eb694 verified
raw
history blame
1.52 kB
import streamlit as st
import backend # Importing backend functions
import os
# Title
st.title("Test Case Generator - QA User Story")
# Input field for the user story
user_story = st.text_area("Enter the QA User Story", "")
# Button to generate test cases
if st.button("Generate Test Cases"):
if user_story:
# Show a spinner while the test cases are being generated
with st.spinner("Generating test cases..."):
test_cases = backend.generate_testcases(user_story)
st.subheader("Generated Test Cases")
st.write(test_cases)
st.session_state.test_cases = test_cases # Store test cases in session state for further use
else:
st.error("Please enter a user story to generate test cases.")
# Export test cases
st.sidebar.title("Export Test Cases")
if st.sidebar.button("Export Test Cases", key="export_button"):
if 'test_cases' in st.session_state:
test_cases = st.session_state.test_cases
export_content = backend.export_test_cases(test_cases, 'csv')
mime_type = "text/csv"
if export_content:
st.sidebar.download_button(
label="Download Test Cases as CSV",
data=export_content,
file_name="test_cases.csv",
mime=mime_type,
key="download_button"
)
else:
st.sidebar.error("No test cases available to export.")
# Footer with a clean divider
st.markdown("---")
st.write("Built by the QA Automation Team")