Merge pull request #143 from web-genie-ai/hotfix/v-issue

Hotfix/v issue

Author: sangar-1028

webgenie/constants.py

@@ -3,7 +3,7 @@
 import psutil
 # Change this value when updating your code base.
 # Define the version of the webgenie.
-__VERSION__ = "1.2.0" # version
+__VERSION__ = "1.2.1" # version
 
 SPEC_VERSION = (
     (1000 * int(__VERSION__.split(".")[0]))

webgenie/datasets/central_dataset.py

@@ -14,9 +14,6 @@ class CentralDataset(Dataset):
 
     def __init__(self):
         pass
-
-    async def generate_context(self) -> DatasetEntry:
-        pass
 
     async def generate_context(self, **kwargs)->DatasetEntry:
         try:

webgenie/tasks/image_task_generator.py

@@ -82,44 +82,6 @@ async def generate_task(self, **kwargs) -> Tuple[Task, bt.Synapse]:
             timeout=IMAGE_TASK_TIMEOUT,
         )
 
-        return (
-            image_task,  
-            WebgenieImageSynapse(base64_image=base64_image, task_id=image_task.task_id),
-        )
-
-    async def generate_task(self, session:int, task_number:int)->Tuple[Task, bt.Synapse]:
-        bt.logging.info("Generating Image task")
-        
-        dataset, _ = random.choices(self.datasets, weights=[weight for _, weight in self.datasets])[0]
-        dataset_entry = await dataset.generate_context(session, task_number)
-        bt.logging.debug(f"Generated dataset entry: {dataset_entry.url}")
-
-        ground_truth_html = preprocess_html(dataset_entry.ground_truth_html)
-        bt.logging.info(f"Preprocessed ground truth html")
-        if not ground_truth_html :
-            raise ValueError("Invalid ground truth html")
-
-        if is_empty_html(ground_truth_html):
-            raise ValueError("Empty ground truth html")
-        
-        base64_image = await html_to_screenshot(ground_truth_html, page_load_time=GROUND_TRUTH_HTML_LOAD_TIME)   
-        # Check image dimensions ratio
-        image = base64_to_image(base64_image)
-        width, height = image.size
-        aspect_ratio = height / width
-        if aspect_ratio > 7:  # If height is more than 7x the width
-            raise ValueError(f"Image aspect ratio too extreme: {aspect_ratio:.2f}. Height should not exceed 7x width.")
-        
-        bt.logging.debug(f"Screenshot generated for {dataset_entry.url}")
-        image_task = ImageTask(
-            base64_image=base64_image,
-            ground_truth_html=ground_truth_html,
-            generator=self,
-            src=dataset_entry.src,
-            task_id=hashlib.sha256(dataset_entry.url.encode()).hexdigest(),
-            timeout=IMAGE_TASK_TIMEOUT,
-        )
-        
         return (
             image_task,  
             WebgenieImageSynapse(base64_image=base64_image, task_id=image_task.task_id),