Upload folder using huggingface_hub

grader.py

@@ -26,7 +26,7 @@ def _attempt_quality(rewards):
     return _safe_ratio(positive_attempts, len(rewards))
 
 
-def _feedback(score, category):
+def feedback(score, category):
     if score >= 0.9:
         return f"{category} performance is excellent and highly reliable."
     if score >= 0.7:
@@ -41,7 +41,7 @@ def grade_segmentation(app_obs: AppObservation):
     progress_score = _safe_ratio(len(app_obs.objectsFound), total_objects)
     quality_score = _attempt_quality(app_obs.rewardListSegment)
     score = _clamp_score((progress_score * 0.7) + (quality_score * 0.3))
-    return score, _feedback(score, "Segmentation")
+    return score, feedback(score, "Segmentation")
 
 
 def grade_placement(app_obs: AppObservation):
@@ -49,7 +49,7 @@ def grade_placement(app_obs: AppObservation):
     progress_score = _safe_ratio(app_obs.numberPlaced, total_objects)
     quality_score = _attempt_quality(app_obs.rewardListPlace)
     score = _clamp_score((progress_score * 0.7) + (quality_score * 0.3))
-    return score, _feedback(score, "Placement")
+    return score, feedback(score, "Placement")
 
 
 def grade_adjustment(app_obs: AppObservation):
@@ -58,4 +58,4 @@ def grade_adjustment(app_obs: AppObservation):
         return 0.0, "Adjustment was not attempted."
 
     score = _attempt_quality(rewards)
-    return score, _feedback(score, "Adjustment")
+    return score, feedback(score, "Adjustment")

inference.py

@@ -3,6 +3,7 @@ from dotenv import load_dotenv
 from openai import OpenAI
 import json
 from json import JSONDecodeError
+from urllib.parse import urlparse
 from numpy import set_printoptions
 
 try:
@@ -24,9 +25,38 @@ except ImportError:
 load_dotenv()
 set_printoptions(precision=2, suppress=True)
 
-API_URL = os.getenv("API_BASE_URL")
-MODEL = os.getenv("MODEL_NAME")
-API_KEY = os.getenv("API_KEY") or os.getenv("HF_TOKEN")
+
+def _get_env(*names):
+    for name in names:
+        value = os.getenv(name)
+        if value:
+            return value.strip().strip("\"'")
+    return None
+
+
+def _normalize_api_url(raw_url):
+    if not raw_url:
+        return None
+
+    url = raw_url.strip().strip("\"'")
+    if "://" not in url:
+        url = f"https://{url.lstrip('/')}"
+
+    parsed = urlparse(url)
+    if not parsed.scheme or not parsed.netloc:
+        raise RuntimeError(
+            "Invalid API base URL. Set API_BASE_URL (or OPENAI_BASE_URL)."
+            "URL such as 'https://generativelanguage.googleapis.com/v1beta/openai/'."
+        )
+
+    return url
+
+
+API_URL = _normalize_api_url(
+    _get_env("API_BASE_URL", "OPENAI_BASE_URL", "OPENAI_API_BASE")
+)
+MODEL = _get_env("MODEL_NAME", "OPENAI_MODEL")
+API_KEY = _get_env("API_KEY", "OPENAI_API_KEY", "HF_TOKEN")
 
 MAX_STEPS = 8
 TEMPERATURE = 0.2
@@ -185,6 +215,10 @@ def main() -> None:
         if observation.isDone:
             break
 
-    segmentGrade , 
+    segment = grade_segmentation(observation)
+    placing = grade_placement(observation)
+    adjust = grade_adjust(observation)
+
+
 if __name__ == "__main__":
     main()