Create app.py
Browse files
app.py
ADDED
@@ -0,0 +1,52 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import streamlit as st
|
2 |
+
import backend # Importing backend functions
|
3 |
+
import os
|
4 |
+
|
5 |
+
# Title
|
6 |
+
st.title("Test Case Generator - QA User Story")
|
7 |
+
|
8 |
+
# Input field for the user story
|
9 |
+
user_story = st.text_area("Enter the QA User Story", "")
|
10 |
+
|
11 |
+
# Button to generate test cases
|
12 |
+
if st.button("Generate Test Cases"):
|
13 |
+
if user_story:
|
14 |
+
# Show a spinner while the test cases are being generated
|
15 |
+
with st.spinner("Generating test cases..."):
|
16 |
+
test_cases = backend.generate_testcases(user_story)
|
17 |
+
st.subheader("Generated Test Cases")
|
18 |
+
st.write(test_cases)
|
19 |
+
st.session_state.test_cases = test_cases # Store test cases in session state for further use
|
20 |
+
else:
|
21 |
+
st.error("Please enter a user story to generate test cases.")
|
22 |
+
|
23 |
+
# Export test cases
|
24 |
+
st.sidebar.title("Export Test Cases")
|
25 |
+
format = st.sidebar.selectbox("Select Format", ["json", "csv"], key="export_format")
|
26 |
+
if st.sidebar.button("Export Test Cases", key="export_button"):
|
27 |
+
if 'test_cases' in st.session_state:
|
28 |
+
test_cases = st.session_state.test_cases
|
29 |
+
export_content = backend.export_test_cases(test_cases, format)
|
30 |
+
|
31 |
+
if format == 'json':
|
32 |
+
mime_type = "application/json"
|
33 |
+
elif format == 'csv':
|
34 |
+
mime_type = "text/csv"
|
35 |
+
else:
|
36 |
+
st.sidebar.error(f"Unsupported format: {format}")
|
37 |
+
export_content = None
|
38 |
+
|
39 |
+
if export_content:
|
40 |
+
st.sidebar.download_button(
|
41 |
+
label=f"Download Test Cases as {format.upper()}",
|
42 |
+
data=export_content,
|
43 |
+
file_name=f"test_cases.{format}",
|
44 |
+
mime=mime_type,
|
45 |
+
key="download_button"
|
46 |
+
)
|
47 |
+
else:
|
48 |
+
st.sidebar.error("No test cases available to export.")
|
49 |
+
|
50 |
+
# Footer with a clean divider
|
51 |
+
st.markdown("---")
|
52 |
+
st.write("Built by the QA Automation Team")
|