Adding browser resolutuion feature for chrome and resolving P0 bug of new installation with HF_HOME envrionment variable (#18)

### Summary 📝
Adding browser resolution feature for chrome and resolving P0 bug of new installation with HF_HOME envrionment variable

### Details
1. Borwser resolution feature for chrome

### Bugfixes 🐛
Blocker for installation, caused because HF_HOME env variable not handled properly in 0.0.20

Author: Shriyansh Agnihotri

.env-example

@@ -12,9 +12,10 @@ PROJECT_SOURCE_ROOT=folder where to save data.
 
 TOKENIZERS_PARALLELISM=false
 MODE=prod
-HEADLESS=true
+HEADLESS=false
 RECORD_VIDEO=false
-TAKE_SCREENSHOTS=false
+TAKE_SCREENSHOTS=true
 BROWSER_TYPE=chromium
-CAPTURE_NETWORK=false
+CAPTURE_NETWORK=true
 HF_HOME=./.cache
+BROWSER_RESOLUTION=1920,1080

helper_scripts/hercules_setup.sh

@@ -1,4 +1,5 @@
 #!/bin/bash
+set -ex
 
 # curl -sS https://bootstrap.pypa.io/get-pip.py | python3.11
 
@@ -18,8 +19,15 @@ export HEADLESS=false
 # create a new directory named 'opt'
 mkdir -p opt/input opt/output opt/test_data
 
-# create a intput/test.feature file
-echo "Feature: Open Google homepage\nScenario: User opens Google homepage\n  Given I have a web browser open\n  When I navigate to https://www.google.com\n  Then I should see the Google homepage" > opt/input/test.feature
+# create a input/test.feature file
+cat << 'EOF' > opt/input/test.feature
+Feature: Open Google homepage
+
+Scenario: User opens Google homepage
+  Given I have a web browser open
+  When I navigate to https://www.google.com
+  Then I should see the Google homepage
+EOF
 
 # get gpt-4o model API key by asking user
 echo "Enter your GPT-4o model API key:"

testzeus_hercules/config.py

@@ -165,6 +165,7 @@ def _merge_from_env(self) -> None:
             "TOKENIZERS_PARALLELISM",
             "DONT_CLOSE_BROWSER",
             "TOKEN_VERBOSE",
+            "BROWSER_RESOLUTION",
         ]
 
         for key in relevant_keys:
@@ -250,6 +251,7 @@ def _finalize_defaults(self) -> None:
             self._config["TOKENIZERS_PARALLELISM"] = "false"
 
         self._config.setdefault("TOKEN_VERBOSE", "false")
+        self._config.setdefault("BROWSER_RESOLUTION", "1920,1080")
 
         # HEADLESS, RECORD_VIDEO, etc. can also be "finalized" here if needed
         self._config.setdefault("HEADLESS", "true")
@@ -309,6 +311,9 @@ def get_browser_type(self) -> str:
 
     def should_capture_network(self) -> bool:
         return self._config["CAPTURE_NETWORK"].lower().strip() == "true"
+    
+    def get_hf_home(self) -> str:
+        return self._config["HF_HOME"]
 
     # -------------------------------------------------------------------------
     # Directory creation logic (mirroring your original code)
@@ -374,6 +379,9 @@ def get_project_temp_path(self, test_id: Optional[str] = None) -> str:
 
     def get_token_verbose(self) -> bool:
         return self._config["TOKEN_VERBOSE"].lower().strip() == "true"
+    
+    def get_resolution(self) -> str:
+        return self._config["BROWSER_RESOLUTION"]
 
     # -------------------------------------------------------------------------
     # Telemetry

testzeus_hercules/core/playwright_manager.py

@@ -220,12 +220,26 @@ async def prepare_extension(self) -> None:
 
     async def create_browser_context(self) -> None:
         user_dir: str = os.environ.get("BROWSER_STORAGE_DIR", "")
+        resolution: str = CONF.get_resolution()
         disable_args = [
-            "--disable-blink-features=AutomationControlled",
             "--disable-session-crashed-bubble",
+            "--disable-notifications",
+            "--no-sandbox",
+            "--disable-blink-features=AutomationControlled",
             "--disable-infobars",
+            "--disable-background-timer-throttling",
             "--disable-popup-blocking",
-            "--disable-notifications",
+            "--disable-backgrounding-occluded-windows",
+            "--disable-renderer-backgrounding",
+            "--disable-window-activation",
+            "--disable-focus-on-load",
+            "--no-first-run",
+            "--no-default-browser-check",
+            "--window-position=0,0",
+            "--disable-web-security",
+            "--disable-site-isolation-trials",
+            "--disable-features=IsolateOrigins,site-per-process",
+            f"--window-size={resolution}",
         ]
         if self.cdp_config:
             # Connect over CDP

testzeus_hercules/core/tools/api_sec_calls.py

@@ -15,7 +15,7 @@
 from testzeus_hercules.utils.logger import logger
 
 # Define cache and binary paths
-CACHE_DIR = Path(os.environ["HF_HOME"]) / "nuclei_tool"
+CACHE_DIR = Path(CONF.get_hf_home()) / "nuclei_tool"
 NUCLEI_BINARY = CACHE_DIR / "nuclei"