Upload 3 files

backend.py

@@ -0,0 +1,122 @@
+import json
+import csv
+import io
+import requests
+import html  # For escaping HTML characters
+from bs4 import BeautifulSoup
+from openai import OpenAI 
+
+# Initialize OpenAI API with Nvidia's Mistral model
+client = OpenAI(
+    base_url="https://integrate.api.nvidia.com/v1",
+    api_key="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"
+)
+
+def clean_test_case_output(text):
+    """
+    Cleans the output to handle HTML characters and unwanted tags.
+    """
+    # Unescape HTML entities (convert &lt; back to < and &gt; back to >)
+    text = html.unescape(text)
+    
+    # Use BeautifulSoup to handle HTML tags more comprehensively
+    soup = BeautifulSoup(text, 'html.parser')
+    
+    # Convert <br> tags to newlines and remove all other tags
+    cleaned_text = soup.get_text(separator="\n").strip()
+    
+    return cleaned_text
+
+def generate_testcases(user_story):
+    """
+    Generates advanced QA test cases based on a provided user story by interacting 
+    with Nvidia's Mistral model API. The prompt is refined for clarity, 
+    and the output is processed for better quality.
+    
+    :param user_story: A string representing the user story for which to generate test cases.
+    :return: A list of test cases in the form of dictionaries.
+    """
+    try:
+        # Example few-shot learning prompt to guide the model
+        completion = client.chat.completions.create(
+            model="nv-mistralai/mistral-nemo-12b-instruct",  # Using Mistral model
+            messages=[
+                {"role": "user", "content": f"Generate QA test cases for the following user story: {user_story}"}
+            ],
+            temperature=0.06,  # Further lowering temperature for precise and deterministic output
+            top_p=0.5,  # Prioritize high-probability tokens even more
+            max_tokens=2048,  # Increase max tokens to allow longer content
+            stream=True  # Streaming the response for faster retrieval
+        )
+
+        # 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
+
+
+        # Ensure the entire response is captured before cleaning
+        if test_cases_text.strip() == "":
+            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:
+            # Try to parse the output as JSON, assuming the model returns structured test cases
+            test_cases = json.loads(test_cases_text)
+            if isinstance(test_cases, list):
+                return test_cases  # Return structured test cases
+
+            else:
+                return [{"test_case": test_cases_text}]  # Return as a list with the text wrapped in a dict
+
+        except json.JSONDecodeError:
+            # Fallback: return the raw text if JSON parsing fails
+            return [{"test_case": test_cases_text}]
+    
+    except requests.exceptions.RequestException as e:
+        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."
+
+    # Convert test cases (which are currently strings) into a structured format for CSV
+    structured_test_cases = [{'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':
+        # Check if test_cases is a list of dicts
+        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.writeheader()
+            csv_writer.writerows(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'):
+    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':
+        with open('test_cases.csv', 'w', newline='') as file:
+            dict_writer = csv.DictWriter(file, fieldnames=test_cases[0].keys())
+            dict_writer.writeheader()
+            dict_writer.writerows(test_cases)
+    else:
+        return f"Unsupported format: {format}"
+    return f'{format} file saved'
+

frontend.py

@@ -0,0 +1,52 @@
+import streamlit as st
+import backend  # Importing backend functions
+import os
+
+# Title
+st.title("Test Case Generator - QA User Story")
+
+# Input field for the user story
+user_story = st.text_area("Enter the QA User Story", "")
+
+# Button to generate test cases
+if st.button("Generate Test Cases"):
+    if user_story:
+        # Show a spinner while the test cases are being generated
+        with st.spinner("Generating test cases..."):
+            test_cases = backend.generate_testcases(user_story)
+        st.subheader("Generated Test Cases")
+        st.write(test_cases)
+        st.session_state.test_cases = test_cases  # Store test cases in session state for further use
+    else:
+        st.error("Please enter a user story to generate test cases.")
+
+# Export test cases
+st.sidebar.title("Export Test Cases")
+format = st.sidebar.selectbox("Select Format", ["json", "csv"], key="export_format")
+if st.sidebar.button("Export Test Cases", key="export_button"):
+    if 'test_cases' in st.session_state:
+        test_cases = st.session_state.test_cases
+        export_content = backend.export_test_cases(test_cases, format)
+
+        if format == 'json':
+            mime_type = "application/json"
+        elif format == 'csv':
+            mime_type = "text/csv"
+        else:
+            st.sidebar.error(f"Unsupported format: {format}")
+            export_content = None
+
+        if export_content:
+            st.sidebar.download_button(
+                label=f"Download Test Cases as {format.upper()}",
+                data=export_content,
+                file_name=f"test_cases.{format}",
+                mime=mime_type,
+                key="download_button"
+            )
+    else:
+        st.sidebar.error("No test cases available to export.")
+
+# Footer with a clean divider
+st.markdown("---")
+st.write("Built by the QA Automation Team")

requirements.rtf

@@ -0,0 +1,14 @@
+{\rtf1\ansi\ansicpg1252\cocoartf2761
+\cocoatextscaling0\cocoaplatform0{\fonttbl\f0\froman\fcharset0 Times-Roman;}
+{\colortbl;\red255\green255\blue255;\red0\green0\blue0;}
+{\*\expandedcolortbl;;\cssrgb\c0\c0\c0;}
+\paperw11900\paperh16840\margl1440\margr1440\vieww11520\viewh8400\viewkind0
+\deftab720
+\pard\pardeftab720\partightenfactor0
+
+\f0\fs24 \cf0 \expnd0\expndtw0\kerning0
+streamlit \
+cohere\
+\pard\pardeftab720\partightenfactor0
+\cf0 \outl0\strokewidth0 \strokec2 beautifulsoup4\
+# Force rebuild}