Update app.py

app.py

@@ -128,6 +128,18 @@ class TrafficSimEnv(gym.Env):
 # Prior to commit had a lot of errors regarding expected output errors
 def optimize_signal_rl(congestion_level):
     try:
+
+        # Map congestion levels (string to numeric)
+        congestion_map = {
+            "Low": 2,
+            "Medium": 5,
+            "High": 8
+        }
+
+        if isinstance(congestion_level, str):
+            congestion_level = congestion_map.get(congestion_level, 5)  # Default to "Medium" if unrecognized
+
+        
         # Create environment
         env = DummyVecEnv([lambda: TrafficSimEnv(congestion_level)])
 
@@ -145,15 +157,14 @@ def optimize_signal_rl(congestion_level):
             action, _ = model.predict(obs, deterministic=True)
             obs, reward, done, truncated, info = env.step(action)
 
-            # Ensure single observation extraction
+            # Unpack single observations (adjust for DummyVecEnv behavior)
             obs = obs[0]
             reward = reward[0]
             done = done[0]
             truncated = truncated[0]
-            info = info[0]
 
-            # Log each step for debugging
-            print(f"STEP: action={action}, obs={obs}, reward={reward}, done={done}, truncated={truncated}, info={info}")
+            # Debugging logs for better tracking
+            logging.debug(f"Action: {action}, Observation: {obs}, Reward: {reward}, Done: {done}")
 
             if done or truncated:
                 break