Refactor app entry point and initialize env variables

app.py

@@ -1,3 +1,4 @@
 from synthetic_dataset_generator import launch
 
-launch()
+if __name__ == "__main__":
+    launch()

src/synthetic_dataset_generator/__init__.py

@@ -1,11 +1,14 @@
 import inspect
-
 from gradio import TabbedInterface
 
 from synthetic_dataset_generator import (  # noqa
     _distiset,
     _inference_endpoints,
 )
+from synthetic_dataset_generator.load_env import init_environment
+
+# Initialize environment variables
+init_environment()
 
 
 def launch(*args, **kwargs):
@@ -14,7 +17,6 @@ def launch(*args, **kwargs):
     Parameters: https://www.gradio.app/docs/gradio/tabbedinterface
     """
     from synthetic_dataset_generator.app import demo
-
     return demo.launch(*args, **kwargs)
 
 

src/synthetic_dataset_generator/load_env.py

@@ -0,0 +1,52 @@
+import os
+from pathlib import Path
+from typing import Dict
+
+def load_env_file(env_path: Path | str) -> Dict[str, str]:
+    """
+    Load environment variables from a .env file.
+    
+    Args:
+        env_path: Path to the .env file
+        
+    Returns:
+        Dict of environment variables loaded
+    
+    Raises:
+        FileNotFoundError: If the .env file doesn't exist
+    """
+    env_vars = {}
+    
+    with open(env_path) as f:
+        for line in f:
+            line = line.strip()
+            if line and not line.startswith('#'):
+                key, value = line.split('=', 1)
+                key = key.strip()
+                value = value.strip()
+                os.environ[key] = value
+                env_vars[key] = value
+                
+    return env_vars
+
+
+def init_environment() -> Dict[str, str]:
+    """
+    Initialize environment variables from .env file in project root.
+    
+    Returns:
+        Dict of environment variables loaded
+        
+    Raises:
+        FileNotFoundError: If the .env file doesn't exist
+    """
+    root_dir = Path(__file__).parent.parent.parent
+    env_file = root_dir / ".env"
+    
+    if not env_file.exists():
+        raise FileNotFoundError(
+            f"Environment file not found at {env_file}. "
+            "Please create a .env file in the project root directory."
+        )
+        
+    return load_env_file(env_file)