Update backend.py

backend.py

@@ -4,7 +4,7 @@ import io
 import requests
 import html  # For escaping HTML characters
 from bs4 import BeautifulSoup
-from openai import OpenAI 
+import pandas as pd  # Add pandas for Excel export
 
 # Initialize OpenAI API with Nvidia's Llama model
 client = OpenAI(
@@ -89,7 +89,6 @@ def generate_testcases(user_story):
         print(f"API request failed: {str(e)}")
         return []
 
-# Add options for multiple test case formats
 def export_test_cases(test_cases, format='json'):
     if not test_cases:
         return "No test cases to export."
@@ -110,7 +109,6 @@ def export_test_cases(test_cases, format='json'):
         else:
             raise ValueError("Test cases must be a list of dictionaries for CSV export.")
 
-# 2. Save test cases as a downloadable file
 def save_test_cases_as_file(test_cases, format='json'):
     if not test_cases:
         return "No test cases to save."
@@ -123,6 +121,10 @@ def save_test_cases_as_file(test_cases, format='json'):
             dict_writer = csv.DictWriter(file, fieldnames=test_cases[0].keys())
             dict_writer.writeheader()
             dict_writer.writerows(test_cases)
+    elif format == 'excel':
+        # Use pandas to export the test cases to Excel
+        df = pd.DataFrame(test_cases)
+        df.to_excel('test_cases.xlsx', index=False)
     else:
         return f"Unsupported format: {format}"
     return f'{format} file saved'