wip payload

app.py

@@ -6,6 +6,7 @@ import pandas as pd
 
 from questions import get_questions_data
 from submit import submit_answers
+from payload import get_answer_payload_results_log
 
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
@@ -13,13 +14,24 @@ API_URL = DEFAULT_API_URL
 QUESTIONS_URL = f"{API_URL}/questions"
 SUBMIT_URL = f"{API_URL}/submit"
 
+SAFETY_OFF = os.getenv("SAFETY_OFF")
+
 SPACE_ID = os.getenv("SPACE_ID")
 if SPACE_ID:
     AGENT_CODE = f"https://huggingface.co/spaces/{SPACE_ID}/tree/main"
     print(f"Agent Code URL: {AGENT_CODE}")
 
 
-def run_and_submit_all(profile: gr.OAuthProfile | None):
+def _check_agent_dependent_environment_variables() -> str:
+    openai_api_key = os.getenv("OPENAI_API_KEY")
+    if not openai_api_key:
+        return "OPENAI_API_KEY environment variable must be set"
+    return None
+
+
+def run_and_submit_all(
+    profile: gr.OAuthProfile | None,
+):  # pylint: disable=too-many-return-statements
     """Run and Submit All"""
     if profile:
         username = f"{profile.username}"
@@ -27,39 +39,25 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
     else:
         print("User not logged in.")
         return "Please Login to Hugging Face with the button.", None
-
     if not SPACE_ID:
         print("SPACE_ID must be set")
         return "SPACE_ID environment variable must be set", None
 
+    error_message = _check_agent_dependent_environment_variables()
+    if error_message:
+        return error_message, None
+
     questions_data = get_questions_data(questions_url=QUESTIONS_URL)
     if not questions_data:
         print("Questions list is empty.")
         return "Questions list is empty or invalid format.", None
     print(f"Retrieved {len(questions_data)} questions.")
 
-    # TEST STUB SECTION START
-
-    results_log = []
-    answers_payload = []
-    submitted_answer = "Mock Final Answer Test Data"
-
-    for item in questions_data:
-        task_id = item.get("task_id")
-        question_text = item.get("question")
-        if not task_id or question_text is None:
-            print(f"Skipping item with missing task_id or question: {item}")
-            continue
-        answers_payload.append(
-            {"task_id": task_id, "submitted_answer": submitted_answer}
-        )
-        results_log.append(
-            {
-                "Task ID": task_id,
-                "Question": question_text,
-                "Submitted Answer": submitted_answer,
-            }
-        )
+    answers_payload, results_log, error_message = get_answer_payload_results_log(
+        questions_data=questions_data
+    )
+    if error_message:
+        return error_message, None
 
     submission_data = {
         "username": username.strip(),
@@ -71,11 +69,13 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
     results_df = pd.DataFrame(results_log)
     print(f"Submitting {len(answers_payload)} answers to: {SUBMIT_URL}")
 
-    # TEST STUB SECTION END
-
-    final_status, error_message = submit_answers(
-        submit_url=SUBMIT_URL, submission_data=submission_data
-    )
+    if SAFETY_OFF:
+        final_status, error_message = submit_answers(
+            submit_url=SUBMIT_URL, submission_data=submission_data
+        )
+    else:
+        final_status = "\n\tSAFETY_OFF bypass active\n"
+        error_message = None
 
     if final_status:
         return final_status, results_df

payload.py

@@ -0,0 +1,26 @@
+"""Runner for Agents to output answers"""
+
+
+def get_answer_payload_results_log(questions_data: dict) -> tuple:
+    """Get Answer Payload, Results Log or Error Message"""
+    results_log = []
+    answers_payload = []
+    submitted_answer = "Mock Final Answer Test Data"
+
+    for item in questions_data:
+        task_id = item.get("task_id")
+        question_text = item.get("question")
+        if not task_id or question_text is None:
+            print(f"Skipping item with missing task_id or question: {item}")
+            continue
+        answers_payload.append(
+            {"task_id": task_id, "submitted_answer": submitted_answer}
+        )
+        results_log.append(
+            {
+                "Task ID": task_id,
+                "Question": question_text,
+                "Submitted Answer": submitted_answer,
+            }
+        )
+    return answers_payload, results_log, None