Update app.py

app.py

@@ -37,10 +37,12 @@ def validate_json_response(expected, actual) -> Tuple[bool, str]:
     except Exception as e:
         return False, f"Invalid JSON: {str(e)}"
 
-def run_single_test(endpoint: str, method: str, bearer_token: str, 
-                    expected_status: int, expected_body: str) -> Dict:
+def run_single_test(endpoint: str, method: str, bearer_token: str,
+                    expected_status: int, expected_body: str,
+                    payload: str = None, test_name: str = None) -> Dict:
     """Run a single API test with AI-powered semantic validation"""
     result = {
+        'TestName': test_name,
         'Endpoint': endpoint,
         'Method': method,
         'Expected Status': expected_status,
@@ -51,60 +53,64 @@ def run_single_test(endpoint: str, method: str, bearer_token: str,
         'Similarity': None,
         'Notes': ''
     }
-    
+
     try:
         headers = {}
         if bearer_token and bearer_token.strip():
             headers['Authorization'] = f'Bearer {bearer_token.strip()}'
-        
+
+        data = None
+        if payload and isinstance(payload, str) and payload.strip():
+            data = payload
+
         # Make API call
         if method.upper() == 'GET':
             response = requests.get(endpoint, headers=headers, timeout=10)
         elif method.upper() == 'POST':
-            response = requests.post(endpoint, headers=headers, timeout=10)
+            response = requests.post(endpoint, headers=headers, data=data, timeout=10)
         elif method.upper() == 'PUT':
-            response = requests.put(endpoint, headers=headers, timeout=10)
+            response = requests.put(endpoint, headers=headers, data=data, timeout=10)
         elif method.upper() == 'DELETE':
             response = requests.delete(endpoint, headers=headers, timeout=10)
         else:
             result['Notes'] = f'Unsupported method: {method}'
             return result
-        
+
         result['Actual Status'] = response.status_code
         actual_body = response.text
         result['Actual Body'] = actual_body[:100] + '...' if len(actual_body) > 100 else actual_body
-        
-        # Check status code
+
+        # Status code check
         if response.status_code != expected_status:
             result['Outcome'] = 'FAIL'
             result['Notes'] = f'Status mismatch: expected {expected_status}, got {response.status_code}'
             return result
-        
-        # Status matches - now check body
+
+        # Body validation
         if not expected_body or is_empty_response(expected_body):
             result['Outcome'] = 'PASS'
             result['Notes'] = 'Status matches, no body validation'
             return result
-        
+
         if is_empty_response(actual_body):
             result['Outcome'] = 'UNCERTAIN'
             result['Similarity'] = 0.0
             result['Notes'] = 'Response is empty (manual review needed)'
             return result
-        
-        # Try JSON validation first
+
+        # JSON validation
         json_match, _ = validate_json_response(expected_body, actual_body)
         if json_match:
             result['Outcome'] = 'PASS'
             result['Similarity'] = 1.0
             result['Notes'] = 'Exact JSON match'
             return result
-        
-        # Use AI semantic similarity
+
+        # Semantic similarity
         try:
             similarity = calculate_semantic_similarity(expected_body, actual_body)
             result['Similarity'] = round(similarity, 3)
-            
+
             if similarity >= 0.75:
                 result['Outcome'] = 'PASS'
                 result['Notes'] = f'High semantic similarity: {similarity:.1%}'
@@ -117,9 +123,9 @@ def run_single_test(endpoint: str, method: str, bearer_token: str,
         except Exception as sim_error:
             result['Outcome'] = 'UNCERTAIN'
             result['Notes'] = f'Cannot compute similarity: {str(sim_error)}'
-        
+
         return result
-        
+
     except requests.exceptions.Timeout:
         result['Notes'] = 'Request timeout'
         return result
@@ -134,31 +140,33 @@ def process_excel_file(file, bearer_token: str) -> Tuple[pd.DataFrame, str]:
     """Process uploaded Excel file and run all tests"""
     try:
         df = pd.read_excel(file)
-        
-        required_columns = ['Endpoint', 'Method', 'Expected Status Code', 'Expected Response Body']
+
+        required_columns = ['Endpoint', 'Method', 'ExpectedStatus', 'ExpectedBody']
         missing_columns = [col for col in required_columns if col not in df.columns]
-        
+
         if missing_columns:
             return None, f"❌ Missing columns: {', '.join(missing_columns)}"
-        
+
         results = []
         for _, row in df.iterrows():
             result = run_single_test(
                 endpoint=str(row['Endpoint']),
                 method=str(row['Method']),
                 bearer_token=bearer_token,
-                expected_status=int(row['Expected Status Code']),
-                expected_body=str(row['Expected Response Body'])
+                expected_status=int(row['ExpectedStatus']),
+                expected_body=str(row['ExpectedBody']),
+                payload=row.get('Payload', None),
+                test_name=row.get('TestName', None)
             )
             results.append(result)
-        
+
         results_df = pd.DataFrame(results)
-        
+
         total = len(results_df)
         passed = len(results_df[results_df['Outcome'] == 'PASS'])
         failed = len(results_df[results_df['Outcome'] == 'FAIL'])
         uncertain = len(results_df[results_df['Outcome'] == 'UNCERTAIN'])
-        
+
         summary = f"""
         📊 **Test Summary**
         - Total Tests: {total}
@@ -166,9 +174,9 @@ def process_excel_file(file, bearer_token: str) -> Tuple[pd.DataFrame, str]:
         - ❌ Failed: {failed}
         - ⚠️ Uncertain: {uncertain}
         """
-        
+
         return results_df, summary
-        
+
     except Exception as e:
         return None, f"❌ Error processing file: {str(e)}"
 
@@ -191,7 +199,7 @@ with gr.Blocks(title="API Test Runner with AI", theme=gr.themes.Soft()) as app:
     
     Upload an Excel file with your API test cases and get automated validation with semantic similarity analysis.
     """)
-    
+
     with gr.Row():
         with gr.Column(scale=2):
             file_input = gr.File(label="Upload Excel File (.xlsx)", file_types=['.xlsx'])
@@ -208,8 +216,8 @@ with gr.Blocks(title="API Test Runner with AI", theme=gr.themes.Soft()) as app:
     with gr.Row():
         results_output = gr.Dataframe(
             label="Test Results",
-            headers=['Endpoint', 'Method', 'Expected Status', 'Actual Status', 
-                    'Outcome', 'Similarity', 'Notes'],
+            headers=['TestName','Endpoint','Method','Expected Status','Actual Status',
+                     'Outcome','Similarity','Notes'],
             interactive=False
         )