fix phase2 timeout

inference.py

@@ -1,6 +1,6 @@
-import os
 import threading
 import time
+import os
 import random
 import traceback
 from openai import OpenAI
@@ -11,63 +11,53 @@ from env.fake_server import run_server
 from env.env import DeceptionEnv
 from env.attacker import simulate_attack
 
+# Start server in background
+threading.Thread(target=run_server, daemon=True).start()
+time.sleep(3)
 
-# If running inside HuggingFace Space → start server only
-if os.getenv("SPACE_ID"):
-
-    run_server()
+API_BASE_URL = os.getenv("API_BASE_URL", "https://router.huggingface.co/v1")
+MODEL_NAME = os.getenv("MODEL_NAME", "Qwen/Qwen2.5-7B-Instruct")
+HF_TOKEN = os.getenv("HF_TOKEN")
 
+try:
 
-else:
+    client = OpenAI(
+        base_url=API_BASE_URL,
+        api_key=HF_TOKEN
+    )
 
-    API_BASE_URL = os.getenv("API_BASE_URL", "https://router.huggingface.co/v1")
-    MODEL_NAME = os.getenv("MODEL_NAME", "Qwen/Qwen2.5-7B-Instruct")
-    HF_TOKEN = os.getenv("HF_TOKEN")
+    env = DeceptionEnv()
+    state = env.reset()
 
-    try:
-
-        threading.Thread(target=run_server, daemon=True).start()
-        time.sleep(3)
-
-        client = OpenAI(
-            base_url=API_BASE_URL,
-            api_key=HF_TOKEN
-        )
-
-        env = DeceptionEnv()
-        state = env.reset()
-
-        print(
-            "[START] task=ai-deception env=cyber-security model=AI-agent",
-            flush=True
-        )
+    print(
+        "[START] task=ai-deception env=cyber-security model=AI-agent",
+        flush=True
+    )
 
-        rewards = []
-        history = []
+    rewards = []
+    history = []
 
-        for step in range(1, 4):
+    for step in range(1, 4):
 
-            try:
-                simulate_attack()
-            except:
-                pass
+        try:
+            simulate_attack()
+        except:
+            pass
 
-            try:
-                state = env.state()
-            except:
-                pass
+        try:
+            state = env.state()
+        except:
+            pass
 
-            summary = {
-                "failed_logins": state.get("failed_logins", 0),
-                "port_scans": state.get("port_scans", 0),
-                "suspicious_ips": len(state.get("suspicious_ips", []))
-            }
+        summary = {
+            "failed_logins": state.get("failed_logins", 0),
+            "port_scans": state.get("port_scans", 0),
+            "suspicious_ips": len(state.get("suspicious_ips", []))
+        }
 
-            prompt = f"""
-You are cybersecurity system.
-
-Previous: {history}
-Current: {summary}
+        prompt = f"""
+Previous actions: {history}
+Current summary: {summary}
 
 Return one:
 detect_attack
@@ -75,50 +65,50 @@ deploy_honeypot
 block_ip
 """
 
-            try:
-                response = client.chat.completions.create(
-                    model=MODEL_NAME,
-                    messages=[{"role": "user", "content": prompt}],
-                    temperature=0.2,
-                    max_tokens=20,
-                    timeout=15
-                )
-
-                action = response.choices[0].message.content.strip()
-
-            except:
+        try:
+            response = client.chat.completions.create(
+                model=MODEL_NAME,
+                messages=[{"role": "user", "content": prompt}],
+                temperature=0.2,
+                max_tokens=20,
+                timeout=15
+            )
 
-                if not history:
-                    action = "detect_attack"
-                elif history[-1] == "detect_attack":
-                    action = "deploy_honeypot"
-                else:
-                    action = "block_ip"
+            action = response.choices[0].message.content.strip()
 
-            history.append(action)
+        except:
 
-            state, reward, done, _ = env.step(action)
-            rewards.append(reward)
+            if not history:
+                action = "detect_attack"
+            elif history[-1] == "detect_attack":
+                action = "deploy_honeypot"
+            else:
+                action = "block_ip"
 
-            print(
-                f"[STEP] step={step} action={action} reward={reward:.2f} "
-                f"done={str(done).lower()} error=null",
-                flush=True
-            )
+        history.append(action)
 
-        score = min(sum(rewards), 1.0)
+        state, reward, done, _ = env.step(action)
+        rewards.append(reward)
 
         print(
-            f"[END] success=true steps={len(rewards)} "
-            f"score={score:.2f} rewards={','.join(f'{r:.2f}' for r in rewards)}",
+            f"[STEP] step={step} action={action} reward={reward:.2f} "
+            f"done={str(done).lower()} error=null",
             flush=True
         )
 
-    except Exception:
+    score = min(sum(rewards), 1.0)
 
-        traceback.print_exc()
+    print(
+        f"[END] success=true steps={len(rewards)} "
+        f"score={score:.2f} rewards={','.join(f'{r:.2f}' for r in rewards)}",
+        flush=True
+    )
 
-        print(
-            "[END] success=false steps=0 score=0.00 rewards=",
-            flush=True
-        )
+except Exception:
+
+    traceback.print_exc()
+
+    print(
+        "[END] success=false steps=0 score=0.00 rewards=",
+        flush=True
+    )