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")
@@ -22,28 +21,17 @@ if st.button("Generate Test Cases"):
 
 # Export test cases
 st.sidebar.title("Export Test Cases")
-format = st.sidebar.selectbox("Select Format", ["json", "csv", "excel"], key="export_format")  # Added "excel" as an option
 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, format)
-
-        if format == 'json':
-            mime_type = "application/json"
-        elif format == 'csv':
-            mime_type = "text/csv"
-        elif format == 'excel':
-            mime_type = "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"
-        else:
-            st.sidebar.error(f"Unsupported format: {format}")
-            export_content = None
+        export_content = backend.export_test_cases(test_cases)
 
         if export_content:
             st.sidebar.download_button(
-                label=f"Download Test Cases as {format.upper()}",
+                label="Download Test Cases as EXCEL",
                 data=export_content,
-                file_name=f"test_cases.{format}",
-                mime=mime_type,
+                file_name="test_cases.xlsx",
+                mime="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet",
                 key="download_button"
             )
     else: