Enterprise-Grade Test Case Generator & Selenium Automation Code Builder (Powered by Gemini 1.5 Flash & Streamlit)
The QE Test Automation Suite is an all-in-one platform for quality engineers to:
- ✨ Generate professional test cases from requirements or user stories using AI
- 🧑💻 Produce production-ready Java Selenium automation code (TestNG, Page Object Model, Allure, Log4j2, etc.)
- 🏗️ Build complete automation frameworks adhering to enterprise standards
- 📄 Manage, edit, and download test cases and automation artifacts in a user-friendly Streamlit interface
This app leverages Google's Gemini 1.5 Flash model, providing industry-leading test design and code generation capabilities.
-
AI-Powered Test Case Generation:
Input requirements or user stories, and instantly generate comprehensive test cases in JSON format. -
Manual Test Case Authoring:
Create, edit, and manage your own test cases with rich forms and attachment support. -
Bulk Test Case Management:
Select, copy, delete, and batch-generate automation code for multiple test cases. -
Automation Code Generation:
Generate Java Selenium automation code (TestNG, POM, Allure, Log4j2, explicit waits, thread safety, etc.) for:- Individual test cases
- Combined test suite (multiple test cases in one suite)
-
Downloadable Artifacts:
Download generated code as a ready-to-use ZIP archive. -
Enterprise UI/UX:
Modern, responsive Streamlit web app with advanced CSS styling and usability features.
- Streamlit — UI framework
- Google Gemini 1.5 Flash — AI test case/code generation
- PyPDF2, python-docx, pandas — File parsing
- Base64 — Attachment encoding
git clone https://github.com/pasindu-kalubowila/QA_Test_Automation_Streamlit_App.git
cd QA_Test_Automation_Streamlit_App
pip install -r requirements.txt
Create a .env
file in the project root:
GEMINI_API_KEY=your_gemini_api_key_here
Note: You need access to Google Gemini API.
streamlit run app.py
- Welcome & feature summary
- Navigation sidebar
- Manual Creation:
Fill in scenario, steps, expected results, and attach files/screenshots. - AI Generation:
Enter user stories or requirements → Get instant, structured test cases. - Bulk Actions:
Select, copy, delete, or send test cases to automation.
- Combined Suite:
Generate a single Java test class for multiple selected test cases. - Separate Files:
Generate separate Java classes per test case. - Download ZIP:
Download all Java source files as a ready-to-import zip.
- Java 17
- Selenium WebDriver
- TestNG
- Page Object Model (with
@FindBy
) - WebDriver Factory (Factory Pattern)
- Singleton configuration
- Log4j2 logging
- Allure reporting
- Explicit waits with
WebDriverWait
- Thread-safe implementation
- Meaningful assertions
- Test Case Attachments:
- Images: PNG, JPG, JPEG
- Docs: PDF, TXT
- Requirement Uploads:
- TXT, PDF, DOCX, CSV, XLSX (auto-parsed)
See requirements.txt
:
streamlit
python-dotenv
google-generativeai
PyPDF2
python-docx
pandas
openpyxl
Pull requests are welcome!
For major changes, please open an issue first to discuss what you would like to change.
QE Test Automation Suite | Powered by Gemini 1.5 Flash & Streamlit