Update app.py

app.py

@@ -6,36 +6,34 @@ import torch
 model = pipeline(
     "text-generation",
     model="gpt2",
-    max_length=200,
-    temperature=0.7,
-    early_stopping=True,
-    torch_dtype=torch.float32  # Use float32 for CPU compatibility
+    max_length=300,          # Increased length to allow more detailed output
+    temperature=0.7,         # Balance between creativity and stability
+    early_stopping=True,     # Stop generation early if completed
+    torch_dtype=torch.float32  # CPU compatibility
 )
 
 def generate_test_cases(requirement):
+    # Revised prompt to focus on the input requirement and avoid copying the example
     prompt = f"""
-    Generate test cases for '{requirement}' in JSON format. Output only the array, take the below as example but return testcases for the specified requirement.
-    
-    Example format:
+    Generate test cases for the requirement: '{requirement}' in JSON array format. 
+    Each test case must include "id", "title", "steps" (array of strings), and "expected_result". 
+
+    DO NOT use the example below in your response. Output only the JSON array, no explanations.
+
+    Example format (IGNORE this when generating):
     [
         {{
             "id": 1,
-            "title": "Valid Login",
-            "steps": ["Enter valid email", "Enter valid password", "Click login"],
-            "expected_result": "Redirect to dashboard"
-        }},
-        {{
-            "id": 2,
-            "title": "Invalid Password",
-            "steps": ["Enter valid email", "Enter wrong password", "Click login"],
-            "expected_result": "Error message displayed"
+            "title": "Example Test",
+            "steps": ["Step 1", "Step 2"],
+            "expected_result": "Expected outcome"
         }}
     ]
     """
     
     try:
         with torch.no_grad():
-            result = model(prompt, max_time=10)[0]["generated_text"]
+            result = model(prompt, max_time=15)[0]["generated_text"]
         return result.strip()
     except Exception as e:
         return f"Error: {str(e)}"