QA-virtual-testing 1.1.0__py3-none-any.whl

ai_virtual_testing/cli.py

@@ -0,0 +1,239 @@
+import sys
+import os
+import time
+import json
+from rich.console import Console
+from rich.prompt import Prompt, Confirm
+from playwright.sync_api import sync_playwright
+from .engine import AIVirtualTester
+from .models import BugList, BugReport
+
+console = Console()
+
+# --- PROFESSIONAL IN-BROWSER SIDEBAR (HTML & CSS ONLY) ---
+# Notice: No <script> tags here anymore!
+QA_SIDEBAR_UI = """
+<style>
+    #qac-wrapper {
+        position: fixed; top: 20px; right: 20px; z-index: 2147483647;
+        font-family: 'Segoe UI', Roboto, Helvetica, Arial, sans-serif;
+    }
+    #qac-panel {
+        width: 320px; background: #18181b; border: 1px solid #3f3f46;
+        border-radius: 12px; box-shadow: 0 10px 25px rgba(0,0,0,0.5);
+        padding: 20px; color: #f4f4f5; display: flex; flex-direction: column; gap: 12px;
+        transition: opacity 0.2s ease-in-out;
+    }
+    .minimized #qac-panel { display: none; }
+    .qac-header { display: flex; justify-content: space-between; align-items: center; }
+    .qac-header h3 { margin: 0; font-size: 16px; color: #0ea5e9; }
+    .qac-label { font-size: 12px; color: #a1a1aa; margin-bottom: 4px; }
+    .qac-input, .qac-select, .qac-textarea {
+        width: 100%; background: #27272a; border: 1px solid #3f3f46;
+        color: white; padding: 8px; border-radius: 6px; font-size: 14px; box-sizing: border-box;
+    }
+    .qac-btn {
+        width: 100%; padding: 10px; border: none; border-radius: 6px;
+        font-weight: 600; cursor: pointer; transition: 0.2s;
+    }
+    .qac-btn-primary { background: #0ea5e9; color: white; }
+    .qac-btn-primary:hover { background: #0284c7; }
+    .qac-btn-danger { background: #ef4444; color: white; margin-top: 5px; }
+    #qac-fab {
+        width: 50px; height: 50px; background: #0ea5e9; border-radius: 50%;
+        display: none; justify-content: center; align-items: center;
+        cursor: pointer; font-size: 24px; box-shadow: 0 4px 12px rgba(0,0,0,0.3);
+    }
+    .minimized #qac-fab { display: flex; }
+</style>
+
+<div id="qac-wrapper">
+    <div id="qac-fab" onclick="document.getElementById('qac-wrapper').classList.remove('minimized')">🤖</div>
+    <div id="qac-panel">
+        <div class="qac-header">
+            <h3>QA Copilot PRO</h3>
+            <button style="background:none; border:none; color:#a1a1aa; cursor:pointer;" onclick="document.getElementById('qac-wrapper').classList.add('minimized')">✖</button>
+        </div>
+        <div>
+            <label class="qac-label">Bug Title</label>
+            <input type="text" id="qac-title" class="qac-input" placeholder="What is the issue?">
+        </div>
+        <div>
+            <label class="qac-label">Severity</label>
+            <select id="qac-severity" class="qac-select">
+                <option value="Low">Low</option>
+                <option value="Medium" selected>Medium</option>
+                <option value="High">High</option>
+            </select>
+        </div>
+        <div>
+            <label class="qac-label">Notes</label>
+            <textarea id="qac-desc" class="qac-textarea" placeholder="Steps to reproduce..."></textarea>
+        </div>
+        <button class="qac-btn qac-btn-primary" onclick="window.submitBug()">📸 Capture & Log Bug</button>
+        <button class="qac-btn qac-btn-danger" onclick="window.pythonFinish()">💾 End Session & Save</button>
+        <div id="qac-status" style="display:none; font-size:12px; text-align:center; color:#10b981; margin-top:5px;">✅ Bug saved!</div>
+    </div>
+</div>
+"""
+# --- END UI ---
+
+def manual_mode_loop(tester: AIVirtualTester, url: str):
+    bugs_caught = []
+    pending_bugs = []
+    finished = [False]
+    
+    with sync_playwright() as p:
+        browser = p.chromium.launch(headless=False)
+        context = browser.new_context(record_video_dir=str(tester.video_dir))
+
+        # Safely package the HTML string
+        safe_html = json.dumps(QA_SIDEBAR_UI)
+        
+        # THE FIX: We inject both the HTML AND define the Javascript functions here natively
+        context.add_init_script(f"""
+            // 1. Inject the HTML when the page loads
+            document.addEventListener("DOMContentLoaded", () => {{
+                if (!document.getElementById('qac-wrapper')) {{
+                    document.body.insertAdjacentHTML('beforeend', {safe_html});
+                }}
+            }});
+
+            // 2. Define the Javascript functions globally so the buttons can find them
+            window.submitBug = () => {{
+                const titleInput = document.getElementById('qac-title');
+                if (!titleInput) return;
+
+                const data = {{
+                    title: titleInput.value,
+                    severity: document.getElementById('qac-severity').value,
+                    desc: document.getElementById('qac-desc').value
+                }};
+
+                if(!data.title) {{
+                    alert("Please enter a Bug Title");
+                    return;
+                }}
+                
+                // Hide UI for a clean screenshot
+                document.getElementById('qac-panel').style.opacity = '0';
+                
+                // Send data to Python
+                window.pythonLogBug(JSON.stringify(data));
+            }};
+
+            window.onBugSaved = () => {{
+                // Bring UI back
+                const panel = document.getElementById('qac-panel');
+                if (panel) panel.style.opacity = '1';
+                
+                // Clear the forms
+                document.getElementById('qac-title').value = '';
+                document.getElementById('qac-desc').value = '';
+                
+                // Show success message
+                const status = document.getElementById('qac-status');
+                if (status) {{
+                    status.style.display = 'block';
+                    setTimeout(() => status.style.display = 'none', 3000);
+                }}
+            }};
+        """)
+
+        page = context.new_page()
+
+        # Python Bridge: Receives the data from Javascript
+        def log_bug_callback(bug_json):
+            pending_bugs.append(json.loads(bug_json))
+
+        def finish_callback():
+            finished[0] = True
+
+        # Expose bridges BEFORE navigating
+        page.expose_function("pythonLogBug", log_bug_callback)
+        page.expose_function("pythonFinish", finish_callback)
+
+        page.goto(url)
+
+        console.print("\n[bold cyan]✨ Manual Copilot Active![/bold cyan]")
+        console.print("[white]Use the sidebar in the browser window to log issues.[/white]")
+
+        # Main Python Loop
+        while not finished[0]:
+            
+            # If a bug was submitted, process it
+            if pending_bugs:
+                data = pending_bugs.pop(0)
+                try:
+                    # Give UI time to hide visually
+                    page.wait_for_timeout(200) 
+                    
+                    shot_path = tester.img_dir / f"manual_bug_{int(time.time())}.png"
+                    page.screenshot(path=str(shot_path))
+                    
+                    bugs_caught.append(BugReport(
+                        issue=data['title'],
+                        severity=data['severity'],
+                        description=data['desc'],
+                        recommendation="Manual Review",
+                        suggested_code_fix="N/A",
+                        screenshot_path=str(shot_path)
+                    ))
+                    
+                    # Tell Javascript to show success message and bring UI back
+                    page.evaluate("if(window.onBugSaved) window.onBugSaved()")
+                except Exception as e:
+                    console.print(f"[red]Failed to capture bug: {e}[/red]")
+
+            # Watchdog: Ensure UI exists mid-navigation
+            try:
+                ui_missing = page.evaluate("() => document.body && !document.getElementById('qac-wrapper')")
+                if ui_missing:
+                    page.evaluate(f"() => document.body.insertAdjacentHTML('beforeend', {safe_html})")
+            except Exception:
+                pass
+
+            page.wait_for_timeout(100)
+            if page.is_closed(): 
+                break
+
+        context.close()
+        browser.close()
+        
+    return BugList(bugs=bugs_caught)
+
+def main():
+    os.system('cls' if os.name == 'nt' else 'clear')
+    console.print("\n[bold cyan]🤖 AI Virtual Tester PRO[/bold cyan]\n", justify="center")
+    
+    api_key = os.environ.get("GEMINI_API_KEY") or Prompt.ask("[yellow]🔑 Enter Gemini API Key[/yellow]", password=True)
+    tester = AIVirtualTester(api_key)
+    
+    url = Prompt.ask("\n[bold green]🌐 Target URL[/bold green]", default="http://localhost:8000")
+    
+    mode = Prompt.ask("\n[bold magenta]Testing Mode[/bold magenta]\n1. Autonomous AI Testing\n2. Interactive Manual Copilot", choices=["1", "2"], default="2")
+    
+    output_file = Prompt.ask("\n[bold green]📁 Excel Filename[/bold green]", default="QA_Report.xlsx")
+    if not output_file.endswith('.xlsx'): output_file += ".xlsx"
+
+    report = None
+
+    if mode == "1":
+        stack = Prompt.ask("[bold green]💻 Tech Stack[/bold green]", default="Django, HTML/CSS/JS")
+        instr = Prompt.ask("[bold green]📝 Instructions[/bold green]", default="Full UI check")
+        autofill = Confirm.ask("Enable Autofill dummy data?")
+        
+        with console.status("[bold magenta]🚀 AI is auditing...[/bold magenta]"):
+            report = tester.run_auto_audit(url, instr, stack, autofill)
+    else:
+        report = manual_mode_loop(tester, url)
+
+    if report and len(report.bugs) > 0:
+        tester.save_rich_excel(report, output_file)
+        console.print(f"\n[bold green]✅ Done! Logged {len(report.bugs)} issues.[/bold green]")
+        console.print(f"📊 Report: [bold underline]qa_reports/{output_file}[/bold underline]")
+    else:
+        console.print("\n[yellow]Session ended. No bugs logged.[/yellow]")
+
+if __name__ == "__main__":
+    main()

ai_virtual_testing/engine.py

@@ -0,0 +1,110 @@
+import os
+import json
+import re
+import time
+from pathlib import Path
+import pandas as pd
+from PIL import Image
+from openpyxl import Workbook
+from openpyxl.drawing.image import Image as XLImage
+from playwright.sync_api import sync_playwright
+import google.generativeai as genai
+from .models import BugList, BugReport
+
+class AIVirtualTester:
+    def __init__(self, api_key: str):
+        genai.configure(api_key=api_key)
+        self.model = genai.GenerativeModel('gemini-2.5-flash')
+        
+        # Create directories for advanced features
+        self.output_dir = Path("qa_reports")
+        self.img_dir = self.output_dir / "screenshots"
+        self.video_dir = self.output_dir / "videos"
+        self.img_dir.mkdir(parents=True, exist_ok=True)
+        self.video_dir.mkdir(parents=True, exist_ok=True)
+
+    def _inject_autofill(self, page):
+        """Advanced feature: Automatically fills forms with dummy data before testing."""
+        page.evaluate("""() => {
+            document.querySelectorAll('input[type="email"]').forEach(el => el.value = 'qa_tester@example.com');
+            document.querySelectorAll('input[type="text"]').forEach(el => el.value = 'QA Automation Test');
+            document.querySelectorAll('input[type="number"]').forEach(el => el.value = '42');
+            document.querySelectorAll('input[type="password"]').forEach(el => el.value = 'SecureTest123!');
+            document.querySelectorAll('textarea').forEach(el => el.value = 'This is an automated test injection.');
+        }""")
+
+    def run_auto_audit(self, url: str, instructions: str, tech_stack: str, autofill: bool) -> BugList:
+        """Fully autonomous mode."""
+        temp_shot = self.img_dir / f"auto_state_{int(time.time())}.png"
+        
+        with sync_playwright() as p:
+            # Launch with video recording enabled
+            browser = p.chromium.launch(headless=True)
+            context = browser.new_context(record_video_dir=str(self.video_dir))
+            page = context.new_page()
+            
+            try:
+                page.goto(url, wait_until="networkidle")
+                page.wait_for_timeout(2000)
+                
+                if autofill:
+                    self._inject_autofill(page)
+                    page.wait_for_timeout(1000) # Wait for UI to update after fill
+                
+                page.screenshot(path=str(temp_shot), full_page=True)
+            finally:
+                context.close()
+                browser.close()
+
+        # AI Analysis
+        with Image.open(temp_shot) as img:
+            prompt = f"""
+            Act as a Lead QA Engineer. Analyze this UI based on:
+            Instructions: {instructions}
+            Tech Stack: {tech_stack}
+
+            Output MUST be a raw JSON object only:
+            {{ "bugs": [ {{ "issue": "", "severity": "", "description": "", "recommendation": "", "suggested_code_fix": "" }} ] }}
+            """
+            response = self.model.generate_content([prompt, img])
+            
+        # Parse JSON
+        match = re.search(r'\{.*\}', response.text, re.DOTALL)
+        if not match:
+            raise ValueError("AI failed to return valid JSON.")
+            
+        bug_data = json.loads(match.group(0))
+        
+        # Attach the screenshot path to all found bugs in auto mode
+        for bug in bug_data.get("bugs", []):
+            bug["screenshot_path"] = str(temp_shot)
+            
+        return BugList(**bug_data)
+
+    def save_rich_excel(self, bug_data: BugList, filename: str):
+        """Advanced feature: Creates an Excel file with actual images embedded inside it."""
+        export_path = self.output_dir / filename
+        
+        # Create base dataframe
+        df = pd.DataFrame([b.model_dump(exclude={"screenshot_path"}) for b in bug_data.bugs])
+        df.to_excel(str(export_path), index=False, engine='openpyxl')
+        
+        # Re-open with openpyxl to inject images
+        import openpyxl
+        wb = openpyxl.load_workbook(str(export_path))
+        ws = wb.active
+        
+        # Add a column for screenshots
+        ws.cell(row=1, column=len(df.columns) + 1, value="Screenshot")
+        
+        for idx, bug in enumerate(bug_data.bugs):
+            row_num = idx + 2
+            ws.row_dimensions[row_num].height = 150 # Make row tall enough for image
+            
+            if bug.screenshot_path and os.path.exists(bug.screenshot_path):
+                img = XLImage(bug.screenshot_path)
+                # Resize image to fit nicely in Excel
+                img.width, img.height = 200, 150
+                ws.add_image(img, f"{openpyxl.utils.get_column_letter(len(df.columns) + 1)}{row_num}")
+                
+        wb.save(str(export_path))

ai_virtual_testing/models.py

@@ -0,0 +1,13 @@
+from pydantic import BaseModel, Field
+from typing import List, Optional
+
+class BugReport(BaseModel):
+    issue: str = Field(description="A short title of the bug")
+    severity: str = Field(description="High, Medium, or Low")
+    description: str = Field(description="Detailed observation")
+    recommendation: str = Field(description="How to fix it")
+    suggested_code_fix: str = Field(description="Exact code snippet to fix the issue")
+    screenshot_path: Optional[str] = Field(default=None, description="Path to the saved image")
+
+class BugList(BaseModel):
+    bugs: List[BugReport]

qa_virtual_testing-1.1.0.dist-info/METADATA

@@ -0,0 +1,22 @@
+Metadata-Version: 2.4
+Name: QA_virtual_testing
+Version: 1.1.0
+Summary: An AI-powered virtual QA tester with Auto-Healing code generation.
+Author: Mohammad Shahzeb Ali Talha
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+Requires-Dist: playwright
+Requires-Dist: google-generativeai
+Requires-Dist: pillow
+Requires-Dist: pandas
+Requires-Dist: openpyxl
+Requires-Dist: pydantic
+Requires-Dist: rich
+
+# AI Virtual Testing
+An AI-powered QA engineer in your terminal.
+
+## Installation
+```bash
+pip install ai_virtual_testing
+playwright install chromium

qa_virtual_testing-1.1.0.dist-info/RECORD

@@ -0,0 +1,9 @@
+ai_virtual_testing/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+ai_virtual_testing/cli.py,sha256=uDn4HxBiBSlVbGL-oJ8BBHWj9baVwnv0keydke2vdjs,10216
+ai_virtual_testing/engine.py,sha256=q6TtT31FQfAxuQPGg2F8NmIO1cXtflAneu3WEoOz_gI,4817
+ai_virtual_testing/models.py,sha256=_o7r91iWUS4aAb5QJnDXuSu00_MrrF5y4dBiC1u7Vh8,599
+qa_virtual_testing-1.1.0.dist-info/METADATA,sha256=Edeu2oHhQZCu4uan3mBdZK17PDjwAjCG8XGXNX-hUh8,581
+qa_virtual_testing-1.1.0.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
+qa_virtual_testing-1.1.0.dist-info/entry_points.txt,sha256=YP_PlgnQlAP4Zc6iQxgQmJ5B15BXpX6nyzibyu4FpEw,56
+qa_virtual_testing-1.1.0.dist-info/top_level.txt,sha256=0N-5mRTZh6Q4CWjYeNA-SB-JjTo6pxG1ZJNQ37mkb30,19
+qa_virtual_testing-1.1.0.dist-info/RECORD,,

qa_virtual_testing-1.1.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (82.0.1)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

qa_virtual_testing-1.1.0.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+ai-test = ai_virtual_testing.cli:main

qa_virtual_testing-1.1.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+ai_virtual_testing