Update app.py

app.py

@@ -1,6 +1,5 @@
 import streamlit as st
 import backend  # Importing backend functions
-import os
 
 # Title
 st.title("Test Case Generator - QA User Story")
@@ -14,7 +13,8 @@ if st.button("Generate Test Cases"):
         # Show a spinner while the test cases are being generated
         with st.spinner("Generating test cases..."):
             test_cases = backend.generate_testcases(user_story)
-        if test_cases:
+        
+        if test_cases and not test_cases[0].get('Test Case') == "No test cases generated or output was empty.":
             st.subheader("Generated Test Cases")
             # Display the test cases in a readable format on the page
             for idx, case in enumerate(test_cases, start=1):
@@ -47,6 +47,8 @@ if st.sidebar.button("Export Test Cases", key="export_button"):
                 mime="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet",
                 key="download_button"
             )
+        else:
+            st.sidebar.error("Error exporting test cases. Please try again.")
     else:
         st.sidebar.error("No test cases available to export.")