Update app.py

app.py

@@ -18,62 +18,32 @@ def extract_text_from_block(pdf_path, block_pages):
     return text
 
 def generate_test_cases(text, prompt_template):
-    # Define a termination phrase that signals completion
-    termination_phrase = "### END OF TEST CASES ###"
-    
-    # Append termination instruction to the prompt template
-    full_prompt_template = prompt_template + (
-        "\n\nPlease ensure that your response ends with '" 
-        + termination_phrase 
-        + "' to indicate that no further test cases are needed."
-    )
-    
-    prompt = full_prompt_template.format(text=text)
+    prompt = prompt_template.format(text=text)
     messages = [
         {"role": "system", "content": "You are a helpful assistant that generates test cases based on given text."},
         {"role": "user", "content": prompt}
     ]
     
-    full_content = ""
-    max_iterations = 10 
-    iteration = 0
-
-    while iteration < max_iterations:
-        iteration += 1
-        try:
-            response = openai.ChatCompletion.create(
-                model="gpt-4o-mini", 
-                messages=messages,
-                max_tokens=1000,
-                n=1,
-                temperature=0.7,
-                timeout=30  # Timeout after 30 seconds
-            )
-        except Exception as e:
-            full_content += f"\n[Error during API call: {str(e)}]"
-            break
-
-        part = response.choices[0].message['content'].strip()
-        full_content += "\n" + part
-        
-        # Check if the termination phrase is in the output
-        if termination_phrase in part:
-            # Optionally remove the termination phrase before returning
-            full_content = full_content.replace(termination_phrase, "")
-            break
-
-        finish_reason = response.choices[0].get("finish_reason", None)
-        # If the response appears truncated, ask the model to continue
-        if finish_reason == "length" or part.endswith("..."):
-            messages.append({"role": "assistant", "content": part})
-            messages.append({"role": "user", "content": "Please continue with the remaining test cases."})
-        else:
-            break
-
-    if iteration == max_iterations:
-        full_content += "\n[WARNING: Maximum iterations reached. The output may be incomplete.]"
+    try:
+        response = openai.ChatCompletion.create(
+            model="gpt-4o-mini", 
+            messages=messages,
+            max_tokens=3000,
+            n=1,
+            temperature=0.7,
+            timeout=30  # Timeout after 30 seconds
+        )
+    except Exception as e:
+        return f"[Error during API call: {str(e)}]"
+    
+    content = response.choices[0].message['content'].strip()
+    finish_reason = response.choices[0].get("finish_reason", None)
+    
+    # Check if the response appears truncated; if so, warn the user.
+    if finish_reason == "length" or content.endswith("..."):
+        content += "\n[WARNING: The output may be truncated due to token limits.]"
     
-    return full_content.strip()
+    return content.strip()