Update CORS

app/main.py

@@ -9,9 +9,9 @@ from .ml import get_model_prediction, check_model_status
 
 app = FastAPI(title="AI-Powered Phishing Email Detection System")
 
-# Determine your specific Hugging Face Space direct URL
-# Example: "https://username-spacename.hf.space"
+# Hugging face URLs
 HF_SPACE_DIRECT_URL = "https://lleratodev-multinomial-nb-phishing-email-detection-api.hf.space"
+HF_SPACE_MAIN = "https://huggingface.co/spaces/lleratodev/multinomial-nb-phishing-email-detection-api"
 HF_SPACE_OLD_URL = "https://huggingface.co/spaces/lleratodev/multinomial-nb-phishing-email-detection-api"
 
 # Define allowed origins for CORS
@@ -19,6 +19,7 @@ origins = [
     "https://ai-powered-phishing-email-detection-system.vercel.app",  # Our Next.js frontend hosted on Vercel 
     "http://localhost:3000",  # Local development
     HF_SPACE_DIRECT_URL,      # Specific direct URL for Hugging Face Space
+    HF_SPACE_MAIN,            # API URL
     HF_SPACE_OLD_URL,         # Older Hugging Face Space URL
     # Regex to match any other *.hf.space subdomains if needed.
     # This matches https://<any-subdomain(s)>.hf.space
@@ -59,12 +60,10 @@ async def get_debug_info():
         cwd = os.getcwd()
         ls_output = subprocess.check_output(["ls", "-la", cwd], text=True)
         env_vars = dict(os.environ)
-        # Add more commands or info as needed
         return {
             "cwd": cwd,
             "ls_output": ls_output,
             "environment_variables": env_vars
-            # Be careful not to expose sensitive environment variables
         }
     except Exception as e:
         return {"error": str(e)}

app/ml_logic.py

@@ -31,7 +31,6 @@ except Exception as e:
     model = None
 
 # --- Load BERT-mini model and tokenizer from Hugging Face Hub ---
-# Replace with your actual Hugging Face model ID
 BERT_MODEL_ID = "lleratodev/720-bert-mini-phishing" # e.g., "LeratoLetsepe/phishing-bert-mini"
 try:
     bert_tokenizer = AutoTokenizer.from_pretrained(BERT_MODEL_ID)