ModuleNotFoundError: 'gui_rewalk.env' not found due to directory structure mismatch

Hi, thank you for this impressive work!
I am trying to run the run_random_walker.sh script following the README, but I encountered a ModuleNotFoundError.
It seems there is a mismatch between the import statements in the code and the actual directory structure in the repository.
The Error
The code attempts to import from gui_rewalk.env:

# In run_random_walker.py (and potentially other files)
from gui_rewalk.env.desktop_gui_gen_env import DesktopGUIGenEnv
from gui_rewalk.env.gui_gen_agent import GUIGenAgent
from gui_rewalk.env.gui_detector import GUIDecorator
Directory Structure
However, inspecting the repository structure, there is no env folder directly under gui_rewalk/. Instead, there is a src folder.
Current structure:

GUI-ReWalk/
├── gui_rewalk/
│ ├── src/ <-- The code seems to be here?
│ │ ├── core/
│ │ ├── utils/
│ │ └── ...
│ └── (No 'env' folder here)
├── run_random_walker.py
└── ...
Steps to Reproduce
Clone the repository.
Install dependencies (OSWorld, OmniParser).
Set up PYTHONPATH.
Run bash run_random_walker.sh.
Expected Behavior
The script should import DesktopGUIGenEnv correctly.
Possible Solution
It seems the code needs to be updated to reflect the new structure (e.g., from gui_rewalk.src.env...) or the env folder is missing from the commit.
Could you please help check this? Thanks!