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 as CSV", 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")