Update backend.py

backend.py

@@ -64,64 +64,49 @@ def generate_testcases(user_story):
             stream=True
         )
 
+        # Initialize an empty string to accumulate the response
         test_cases_text = ""
 
+        # Accumulate the response from the streaming chunks
         for chunk in completion:
             if chunk.choices[0].delta.content is not None:
                 test_cases_text += chunk.choices[0].delta.content
 
+        # Print raw response for debugging
+        print("Raw response from model:", test_cases_text)
+
+        # Ensure the entire response is captured before cleaning
         if test_cases_text.strip() == "":
-            return [{"test_case": "No test cases generated or output was empty."}]
+            return [{"Test Case": "No test cases generated or output was empty."}]
 
+        # Clean the output by unescaping HTML entities and replacing <br> tags
         test_cases_text = clean_test_case_output(test_cases_text)
 
-        try:
-            test_cases = json.loads(test_cases_text)
-            if isinstance(test_cases, list):
-                return test_cases
-            else:
-                return [{"test_case": test_cases_text}]
-        except json.JSONDecodeError:
-            # If JSON decoding fails, attempt to parse manually
-            return parse_test_cases(test_cases_text)
+        # Print cleaned response for debugging
+        print("Cleaned response:", test_cases_text)
 
-    except requests.exceptions.RequestException as e:
-        print(f"API request failed: {str(e)}")
-        return []
+        # Split the output into individual test cases by detecting patterns
+        test_case_blocks = test_cases_text.split('\n\n')
+        test_cases = []
+        for block in test_case_blocks:
+            lines = block.split('\n')
+            if len(lines) >= 4:
+                test_case = {
+                    'Test Case': lines[0].replace('Test Case ', '').strip(),
+                    'Preconditions': lines[1].replace('Preconditions: ', '').strip(),
+                    'Steps': lines[2].replace('Steps: ', '').strip(),
+                    'Expected Result': lines[3].replace('Expected Result: ', '').strip(),
+                }
+                test_cases.append(test_case)
 
-def parse_test_cases(raw_text):
-    """
-    Parse raw text output into structured test cases.
-    
-    :param raw_text: Raw text returned from the model.
-    :return: List of dictionaries with structured test cases.
-    """
-    test_cases = []
-    case = {
-        "Test Case": "",
-        "Preconditions": "N/A",
-        "Steps": "N/A",
-        "Expected Result": "N/A"
-    }
-
-    lines = raw_text.split("\n")
-    for line in lines:
-        if line.startswith("Test Case"):
-            if case["Test Case"]:
-                test_cases.append(case)  # Save the previous test case
-                case = {"Test Case": "", "Preconditions": "N/A", "Steps": "N/A", "Expected Result": "N/A"}
-            case["Test Case"] = line.replace("Test Case:", "").strip()
-        elif "Preconditions" in line:
-            case["Preconditions"] = line.replace("Preconditions:", "").strip() or "N/A"
-        elif "Steps" in line:
-            case["Steps"] = line.replace("Steps:", "").strip() or "N/A"
-        elif "Expected Result" in line:
-            case["Expected Result"] = line.replace("Expected Result:", "").strip() or "N/A"
-    
-    if case["Test Case"]:  # Add the last case
-        test_cases.append(case)
+        if not test_cases:
+            return [{"Test Case": "No test cases generated or output was empty."}]
+
+        return test_cases
 
-    return test_cases
+    except requests.exceptions.RequestException as e:
+        print(f"API request failed: {str(e)}")
+        return [{"Test Case": "API request failed."}]
 
 def export_test_cases(test_cases):
     """