Update app.py

app.py

@@ -171,43 +171,33 @@ def evaluate_single_case(input_data):
             return {"status": "Exception", "error": "Input item must be a dictionary"}
             
         language = input_data.get('language')
-        completions = input_data.get('processed_completions', [])
+        code = input_data.get('code', "")
 
-        if not completions:
+        if code == "":
             return {"status": "Exception", "error": "No code provided"}
 
         # Use a retry mechanism for all languages for better reliability
         max_retries = 2  # One retry for all languages
         
-        status_list, stderr_list = [], []
-        for comp in completions:
-            code = input_data.get('prompt') + comp + '\n' + input_data.get('tests')
+        # Try up to max_retries times for all test cases
+        status, stderr = "", ""
+        for attempt in range(max_retries):
+            result = evaluate_code(code, language)
             
-            # Try up to max_retries times for all test cases
-            for attempt in range(max_retries):
-                result = evaluate_code(code, language)
-                
-                # If success or last attempt, return/record the result
-                if result["status"] == "OK":
-                    break
-                # For retries, briefly wait to allow resources to stabilize
-                time.sleep(0.3)
-
-            status_list.append(result["status"])
-            stderr_list.append(result["stderr"])
-        
-        processed_completions = input_data.pop('processed_completions', [])
-        completions = input_data.pop('completions', [])
+            # If success or last attempt, return/record the result
+            if result["status"] == "OK":
+                break
+            # For retries, briefly wait to allow resources to stabilize
+            time.sleep(0.3)
+
+            status = result["status"]
+            stderr = result["stderr"]
 
-        meta_data = [
-            {
-                'processed_completion': p_comp,
-                'completion': comp,
+        meta_data = {
+                'code': code,
                 'status': status,
                 'stderr': stderr
             }
-            for p_comp, comp, status, stderr in zip(processed_completions, completions, status_list, stderr_list)
-        ]
 
         input_data['meta_data'] = meta_data
         return input_data