refactor: proper package structure - move he_demo package to dedicated folder

This refactoring fixes the hacky pyproject.toml setup where the root was acting as the package.

Changes:
- Created he_demo/ subfolder with proper Python package structure
- Moved core files: __init__.py, client.py, models.py, task_graders.py, task_registry.py, grader_manifest.py
- Moved server/ to he_demo/server/ with all files (__init__.py, app.py, he_demo_environment.py)
- Updated pyproject.toml:
* Removed hacky package-dir = {he_demo: ., he_demo.server: server}
* Kept clean: packages = [he_demo, he_demo.server]
* Removed py-modules = [graders]
- Updated all root-level scripts to use proper imports:
* validate_comprehensive.py: server.he_demo_environment → he_demo.server.he_demo_environment
* gym_wrapper.py: server.he_demo_environment → he_demo.server.he_demo_environment
* test_environment.py: server.he_demo_environment → he_demo.server.he_demo_environment
* validate.py: server.he_demo_environment → he_demo.server.he_demo_environment

Benefits:
✓ Proper Python package structure (no more hacky root-level package dir)
✓ All imports use he_demo.* paths consistently
✓ Can be properly installed with pip install -e .
✓ Compatible with build tools and package managers
✓ Better IDE support and import resolution

gym_wrapper.py

@@ -20,7 +20,7 @@ he_demo.TaskSummary = TaskSummary
 sys.modules['he_demo'] = he_demo
 sys.modules['he_demo.models'] = he_demo
 
-from server.he_demo_environment import EnergyOptimizationEnvironment
+from he_demo.server.he_demo_environment import EnergyOptimizationEnvironment
 
 class EnergyOptimizationGymEnv(gym.Env):
     """Gym wrapper for the Energy Optimization Environment."""

__init__.py → he_demo/__init__.py

@@ -6,9 +6,9 @@
 
 """Energy & Memory RAM Optimization Environment."""
 
-from .client import EnergyOptimizationEnv
-from .models import EnergyOptimizationAction, EnergyOptimizationObservation, Task
-from .task_graders import (
+from he_demo.client import EnergyOptimizationEnv
+from he_demo.models import EnergyOptimizationAction, EnergyOptimizationObservation, Task
+from he_demo.task_graders import (
     TASK_GRADERS,
     get_grader,
     get_all_graders,
@@ -19,14 +19,14 @@ from .task_graders import (
     task_4_advanced_efficiency_grader,
     task_5_expert_optimization_grader,
 )
-from .task_registry import (
+from he_demo.task_registry import (
     TASK_REGISTRY,
     get_all_tasks_with_graders,
     get_task_grader,
     get_tasks_count,
     is_grader_requirement_met,
 )
-from .grader_manifest import (
+from he_demo.grader_manifest import (
     GRADERS_MANIFEST,
     get_graders_manifest,
     get_active_graders_count,

pyproject.toml

@@ -41,6 +41,4 @@ server = "he_demo.server.app:main"
 
 [tool.setuptools]
 include-package-data = true
-packages = ["he_demo", "he_demo.server"]
-package-dir = { "he_demo" = ".", "he_demo.server" = "server" }
-py-modules = ["graders"]
+packages = ["he_demo", "he_demo.server"]

test_environment.py

@@ -26,7 +26,7 @@ sys.modules['he_demo'] = he_demo
 sys.modules['he_demo.models'] = he_demo
 
 # Now import the environment
-from server.he_demo_environment import EnergyOptimizationEnvironment
+from he_demo.server.he_demo_environment import EnergyOptimizationEnvironment
 
 def test_environment():
     """Test the energy optimization environment."""

validate.py

@@ -31,7 +31,7 @@ sys.modules['he_demo.models'] = he_demo
 sys.modules['he_demo.task_graders'] = sys.modules.get('task_graders')
 
 # Now import the environment
-from server.he_demo_environment import EnergyOptimizationEnvironment
+from he_demo.server.he_demo_environment import EnergyOptimizationEnvironment
 
 def main():
     print("🔋 Energy & Memory RAM Optimization Environment - Final Validation")

validate_comprehensive.py

@@ -28,7 +28,7 @@ sys.modules['he_demo'] = he_demo
 sys.modules['he_demo.models'] = he_demo
 
 # Now import the environment
-from server.he_demo_environment import EnergyOptimizationEnvironment
+from he_demo.server.he_demo_environment import EnergyOptimizationEnvironment
 
 def create_observation(ram_usage, energy_consumption, steps_taken):
     """Helper to create observations for testing."""