Update main.py

main.py

@@ -6,6 +6,7 @@ import os
 
 # Load environment variables
 load_dotenv()
+# Trigger reload
 
 app = FastAPI(
     title="Network IDS API",
@@ -14,10 +15,19 @@ app = FastAPI(
     root_path=os.getenv("ROOT_PATH", "")
 )
 
-# Allow Streamlit frontend
+# Configure CORS
+# Default to ["*"] if not set. To restrict, set ALLOWED_ORIGINS="http://localhost:3000,https://my-app.hf.space"
+origins_env = os.getenv("ALLOWED_ORIGINS", "*")
+if origins_env == "*":
+    origins = ["*"]
+else:
+    origins = [origin.strip() for origin in origins_env.split(",") if origin.strip()]
+
+print(f"Allowed CORS Origins: {origins}")
+
 app.add_middleware(
     CORSMiddleware,
-    allow_origins=["*"],
+    allow_origins=origins,
     allow_credentials=True,
     allow_methods=["*"],
     allow_headers=["*"],
@@ -33,4 +43,4 @@ app.include_router(blockchain.router, prefix="/blockchain", tags=["Blockchain"])
 
 @app.get("/")
 def home():
-    return {"message": "Network IDS Backend Running!"}
+    return {"message": "Network IDS Backend Running!"}