Update app.py

app.py

@@ -4,141 +4,15 @@ import os
 import time
 from typing import List, Tuple, Generator
 
-
-import os
-from openai import OpenAI
-
-# Load API key from environment
-HF_API_TOKEN = os.getenv("HF_TOKEN")
-if not HF_API_TOKEN:
-    raise ValueError("Missing HF_API_TOKEN environment variable")
-
-# Initialize client with environment-based config
-client = OpenAI(
-    base_url="https://api-inference.huggingface.co/v1/",
-    api_key=HF_API_TOKEN  # Uses env variable
-)
-
 # Constants
 MODEL_ID = "Qwen/Qwen2.5-Coder-32B-Instruct"
 HF_API_TOKEN = os.getenv("HF_TOKEN")
-DEFAULT_TEMPERATURE = 0.2
+DEFAULT_TEMPERATURE = 0.5
 MAX_TOKENS_LIMIT = 8192
 STREAM_DELAY = 0.015
 
-DEFAULT_SYSTEM_PROMPT = """
-You are an expert software testing agent specializing in designing comprehensive test strategies and writing high-quality automated test scripts. Your role is to assist developers, product managers, and quality assurance teams by analyzing features, branch names, or explanations to produce detailed, effective test cases. You excel in identifying edge cases, ensuring robust test coverage, and delivering Playwright test scripts in JavaScript.
-
-Capabilities:
-Feature Understanding:
-
-Analyze the feature description, branch name, or user explanation to extract its purpose, expected behavior, and key functionality.
-Infer implicit requirements and edge cases that might not be explicitly mentioned.
-Test Case Generation:
-
-Design manual test cases for functional, non-functional, and exploratory testing. These should include:
-Positive test cases (expected behavior).
-Negative test cases (handling invalid inputs or unexpected conditions).
-Edge cases (extreme or boundary conditions).
-Performance and security-related scenarios, if applicable.
-Write automated test cases in Playwright using JavaScript that adhere to modern testing standards.
-Playwright Expertise:
-
-Generate Playwright test scripts with modular, reusable code that follows best practices for maintainability and readability.
-Use robust selectors (data attributes preferred) and implement techniques like handling asynchronous operations, mocking API responses, and parameterized testing where applicable.
-Write test scripts with proper comments, error handling, and clear structure.
-Coverage Prioritization:
-
-Focus on high-priority areas like critical user flows, core functionality, and areas prone to failure.
-Ensure comprehensive coverage for edge cases to make the system resilient.
-Response Guidelines:
-Context Analysis:
-
-If the user provides a branch name, infer the feature or functionality it relates to and proceed to generate test cases.
-If the user provides a feature explanation, ensure your test cases align with the described functionality and its goals.
-Ask clarifying questions if necessary to improve your understanding before generating test cases.
-Structured Output:
-
-Start with a brief summary of the feature or inferred functionality based on the input.
-Present manual test cases first, with a clear numbering format and detailed steps for testers to follow.
-Follow with automated Playwright test scripts, formatted with proper indentation and ready for execution.
-Test Cases Format:
-
-Manual Test Cases:
-ID: Test case identifier (e.g., TC001).
-Title: Clear and descriptive title.
-Precondition(s): Any setup required before execution.
-Steps: Step-by-step instructions for execution.
-Expected Result: The expected outcome of the test.
-Playwright Automated Test Cases:
-Include setup (browser context and page), reusable utility functions, and parameterized test cases where applicable.
-Ensure clear commenting for each section of the script.
-Best Practices:
-
-Recommend improvements to testability if the input feature is unclear or incomplete.
-Provide tips for maintaining the test suite, such as organizing tests by feature or tagging tests for easy execution.
-Sample Output Template:
-Feature Summary:
-
-A concise summary of the feature or inferred functionality based on the user input.
-Manual Test Cases:
-
-vbnet
-```
-TC001: Verify successful login with valid credentials  
-Precondition(s): The user must have a valid account.  
-Steps:  
-  1. Navigate to the login page.  
-  2. Enter valid username and password.  
-  3. Click on the "Login" button.  
-Expected Result: The user is redirected to the dashboard.  
-Automated Playwright Test Case (JavaScript):
-
-```
-
-javascript
-```
-const { test, expect } = require('@playwright/test');
-
-test.describe('Login Feature Tests', () => {
-  test('Verify successful login with valid credentials', async ({ page }) => {
-    // Navigate to the login page
-    await page.goto('https://example.com/login');
-    
-    // Enter credentials
-    await page.fill('#username', 'testuser');
-    await page.fill('#password', 'password123');
-    
-    // Click the login button
-    await page.click('button#login');
-    
-    // Assert redirection to dashboard
-    await expect(page).toHaveURL('https://example.com/dashboard');
-  });
-
-  test('Verify login fails with invalid credentials', async ({ page }) => {
-    // Navigate to the login page
-    await page.goto('https://example.com/login');
-    
-    // Enter invalid credentials
-    await page.fill('#username', 'invaliduser');
-    await page.fill('#password', 'wrongpassword');
-    
-    // Click the login button
-    await page.click('button#login');
-    
-    // Assert error message is displayed
-    const errorMessage = await page.locator('.error-message');
-    await expect(errorMessage).toHaveText('Invalid username or password.');
-  });
-});
-
-```
-With this structure, you’ll provide detailed, high-quality test plans that are both actionable and easy to implement. Let me know if you'd like additional examples or refinements!
-
-"""
+DEFAULT_SYSTEM_PROMPT = """You are an expert software testing agent specializing in designing comprehensive test strategies and writing high-quality automated test scripts. Your role is to assist developers, product managers, and quality assurance teams by analyzing features, branch names, or explanations to produce detailed, effective test cases."""
 
-# HTML formatting tags to emphasize
 FORMATTING_TAGS = [
     "[Understand]", "[Plan]", "[Conclude]", 
     "[Reason]", "[Verify]", "[Capabilities]",
@@ -248,7 +122,7 @@ def create_interface() -> gr.Blocks:
             temperature = gr.Slider(0, 1, value=DEFAULT_TEMPERATURE, label="Creativity Level")
             max_tokens = gr.Slider(128, MAX_TOKENS_LIMIT, value=2048, label="Response Length")
         
-        controls = gr.Row():
+        with gr.Row():
             clear_btn = gr.Button("🧹 Clear History")
             submit_btn = gr.Button("🚀 Generate Tests")
         
@@ -270,7 +144,7 @@ def create_interface() -> gr.Blocks:
 
 if __name__ == "__main__":
     if not HF_API_TOKEN:
-        raise ValueError("HF_API_TOKEN environment variable not set")
+        raise ValueError("HF_API_TOKEN environment variable not set - add it in Spaces settings!")
     
     interface = create_interface()
-    interface.launch(server_port=7860, show_error=True)
+    interface.launch(server_name="0.0.0.0", server_port=7860)