env varaible

main.py

@@ -6,6 +6,7 @@ import re
 import json
 from fastapi.responses import HTMLResponse
 import requests
+import os
 
 
 # ----------------------------
@@ -15,8 +16,12 @@ import requests
 MODEL_NAME = "Salesforce/codegen-350M-mono"
 DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 
-# Add your Hugging Face API key here
-HF_API_KEY = "REDACTED"
+# Remove hardcoded API key and use an environment variable
+HF_API_KEY = os.getenv("HF_API_KEY")
+
+if not HF_API_KEY:
+    raise RuntimeError("Hugging Face API key is not set. Please set the HF_API_KEY environment variable.")
+
 HF_MODEL_NAME = "bigcode/starcoder"  # Replace with the best model for code review
 
 # ----------------------------
@@ -82,9 +87,6 @@ def run_ai_inference(diff: str) -> str:
     """
     Sends the code diff to Hugging Face Inference API to get the review.
     """
-    if not HF_API_KEY:
-        raise RuntimeError("Hugging Face API key is not set.")
-
     # Better prompt for meaningful completions
     prompt = f"""# Code Review