Update backend.py

backend.py

@@ -1,4 +1,3 @@
-import json
 import csv
 import io
 import requests
@@ -64,7 +63,6 @@ def generate_testcases(user_story):
             if chunk.choices[0].delta.content is not None:
                 test_cases_text += chunk.choices[0].delta.content
 
-
         # Ensure the entire response is captured before cleaning
         if test_cases_text.strip() == "":
             return [{"test_case": "No test cases generated or output was empty."}]
@@ -89,40 +87,34 @@ 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'):
+# Export test cases in CSV format
+def export_test_cases(test_cases, format='csv'):
     if not test_cases:
         return "No test cases to export."
 
     # Convert test cases (which are currently strings) into a structured format for CSV
-    structured_test_cases = [{'Test Case': case} for case in test_cases]
+    structured_test_cases = [{'Test Case': case.get('test_case', case)} for case in test_cases]
 
-    if format == 'json':
-        # Improve JSON export to be line-by-line formatted
-        return json.dumps(test_cases, indent=4, separators=(',', ': '))  # More readable format
-    elif format == 'csv':
+    if format == 'csv':
         if isinstance(test_cases, list) and isinstance(test_cases[0], dict):
             output = io.StringIO()
-            csv_writer = csv.DictWriter(output, fieldnames=test_cases[0].keys(), quoting=csv.QUOTE_ALL)
+            csv_writer = csv.DictWriter(output, fieldnames=structured_test_cases[0].keys(), quoting=csv.QUOTE_ALL)
             csv_writer.writeheader()
-            csv_writer.writerows(test_cases)
+            csv_writer.writerows(structured_test_cases)
             return output.getvalue()
         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'):
+# Save test cases as a CSV file
+def save_test_cases_as_file(test_cases, format='csv'):
     if not test_cases:
         return "No test cases to save."
 
-    if format == 'json':
-        with open('test_cases.json', 'w') as f:
-            json.dump(test_cases, f)
-    elif format == 'csv':
+    if format == 'csv':
         with open('test_cases.csv', 'w', newline='') as file:
-            dict_writer = csv.DictWriter(file, fieldnames=test_cases[0].keys())
+            dict_writer = csv.DictWriter(file, fieldnames=['Test Case'])
             dict_writer.writeheader()
-            dict_writer.writerows(test_cases)
+            dict_writer.writerows([{'Test Case': case.get('test_case', case)} for case in test_cases])
     else:
         return f"Unsupported format: {format}"
-    return f'{format} file saved'
+    return f'{format} file saved'