fix: added robust descriptive labels for dataset strings

backend/main.py

@@ -6,15 +6,19 @@ from fastapi.staticfiles import StaticFiles
 from pydantic import BaseModel
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
 
-# Complexity descriptions 
+# Complexity descriptions (robust mapping for codeparrot labels)
 DESCRIPTIONS = {
-    "constant":  ("O(1)", "⚡ Constant Time", "Executes in the same time regardless of input size. Very fast!"),
-    "linear":    ("O(n)", "📈 Linear Time", "Execution time grows linearly with input size."),
-    "logn":      ("O(log n)", "🔍 Logarithmic Time", "Very efficient! Common in binary search algorithms."),
-    "nlogn":     ("O(n log n)", "⚙️ Linearithmic Time", "Common in efficient sorting algorithms like merge sort."),
-    "quadratic": ("O(n²)", "🐢 Quadratic Time", "Execution time grows quadratically. Common in nested loops."),
-    "cubic":     ("O(n³)", "🦕 Cubic Time", "Triple nested loops. Avoid for large inputs."),
-    "np":        ("O(2ⁿ)", "💀 Exponential Time", "NP-Hard complexity. Only feasible for very small inputs."),
+    "O(1)":       ("O(1)", "⚡ Constant Time", "Executes in the same time regardless of input size. Very fast!"),
+    "O(N)":       ("O(N)", "📈 Linear Time", "Execution time grows linearly with input size."),
+    "O(log N)":   ("O(log N)", "🔍 Logarithmic Time", "Very efficient! Common in binary search algorithms."),
+    "O(N log N)": ("O(N log N)", "⚙️ Linearithmic Time", "Common in efficient sorting algorithms like merge sort."),
+    "O(N^2)":     ("O(N²)", "🐢 Quadratic Time", "Execution time grows quadratically. Common in nested loops."),
+    "O(N^3)":     ("O(N³)", "🦕 Cubic Time", "Triple nested loops. Avoid for large inputs."),
+    "O(2^N)":     ("O(2ⁿ)", "💀 Exponential Time", "NP-Hard complexity. Only feasible for very small inputs."),
+    "O(NP)":      ("O(NP)", "💀 NP-Complete", "Infeasible for large inputs without approximation."),
+    "constant":   ("O(1)", "⚡ Constant Time", "Executes in the same time regardless of input size. Very fast!"),
+    "linear":     ("O(N)", "📈 Linear Time", "Execution time grows linearly with input size."),
+    "quadratic":  ("O(N²)", "🐢 Quadratic Time", "Execution time grows quadratically. Common in nested loops."),
 }
 
 app = FastAPI(title="Code Complexity Predictor API")