Upload app.py

app.py

@@ -50,6 +50,15 @@ def contains_forbidden(text: str) -> bool:
     return False
 
 # ---------------------------------------------------------------------------
+# Lee la API key de varias variables aceptadas (para que puedas usar OPENAI_API_KEY_ATAQUE)
+ENV_CANDIDATES = ["OPENAI_API_KEY", "OPENAI_API_KEY_ATAQUE", "OPENAI_APIKEY", "OPENAI_KEY", "HF_OPENAI_API_KEY"]
+
+def get_api_key() -> str | None:
+    for name in ENV_CANDIDATES:
+        v = os.environ.get(name)
+        if v:
+            return v
+    return None
 
 def call_openai_responses(prompt: str, api_key: str, models=None, timeout: int = 20) -> Tuple[bool, str]:
     if models is None:
@@ -121,9 +130,10 @@ def safe_parse_json_from_model(text: str):
         return {"raw": text}
 
 def generate_simulation(user_input: str, include_iocs: bool, include_mitigation: bool):
-    api_key = os.environ.get("OPENAI_API_KEY")
+    api_key = get_api_key()
     if not api_key:
-        return "<p style='color:crimson'><b>Error:</b> OPENAI_API_KEY no configurada en Settings → Variables and secrets.</p>", ""
+        return ("<p style='color:crimson'><b>Error:</b> Falta la API key. "
+                f"Añade una de estas variables en Settings → Variables and secrets: {', '.join(ENV_CANDIDATES)}.</p>", "")
 
     prompt = PROMPT_TEMPLATE.format(input=user_input)
     ok, out = call_openai_responses(prompt, api_key)