debug

app.py

@@ -105,10 +105,8 @@ def render_preview(code: str, width: int, height: int, scale: float) -> str:
     Preview renderer with both width and height control for the inner canvas.
     """
     try:
-        # Clean up the HTML code to remove problematic script and CSS references
         soup = BeautifulSoup(code, 'html.parser')
         
-        # For preview, we'll just remove problematic references since we don't have the output_dir context
         for script in soup.find_all('script'):
             src = script.get('src', '')
             if src and any(pattern in src for pattern in ['assets/', 'index-', 'iframeResizer']):
@@ -119,7 +117,6 @@ def render_preview(code: str, width: int, height: int, scale: float) -> str:
             if href and any(pattern in href for pattern in ['assets/', 'index-']):
                 link.decompose()
         
-        # Get the cleaned HTML
         cleaned_code = str(soup)
         
     except Exception as e:

screencoder/block_parsor.py

@@ -226,14 +226,43 @@ def main():
     json_output_path = os.path.join(tmp_dir, f"{run_id}_bboxes.json")
     annotated_image_output_path = os.path.join(tmp_dir, f"{run_id}_with_bboxes.png")
 
-    if not os.path.exists(image_path) or not os.path.exists(api_path):
-        print(f"Error: Input image or API key file not found.")
+    # Debug: Print all paths and check if files exist
+    print(f"Debug - base_dir: {base_dir}")
+    print(f"Debug - tmp_dir: {tmp_dir}")
+    print(f"Debug - image_path: {image_path}")
+    print(f"Debug - api_path: {api_path}")
+    print(f"Debug - image_path exists: {os.path.exists(image_path)}")
+    print(f"Debug - api_path exists: {os.path.exists(api_path)}")
+    
+    # List contents of tmp_dir if it exists
+    if os.path.exists(tmp_dir):
+        print(f"Debug - tmp_dir contents: {os.listdir(tmp_dir)}")
+    else:
+        print(f"Debug - tmp_dir does not exist: {tmp_dir}")
+
+    if not os.path.exists(image_path):
+        print(f"Error: Input image not found at {image_path}")
+        # Create empty json file so the pipeline doesn't break
+        save_bboxes_to_json({}, json_output_path)
+        exit(1)
+    
+    if not os.path.exists(api_path):
+        print(f"Error: API key file not found at {api_path}")
+        # Create empty json file so the pipeline doesn't break
+        save_bboxes_to_json({}, json_output_path)
         exit(1)
 
     print(f"--- Starting BBox Parsing for run_id: {run_id} ---")
     
     client = Doubao(api_path)
-    bbox_content = client.ask(PROMPT_MERGE, encode_image(image_path))
+    base64_image = encode_image(image_path)
+    if not base64_image:
+        print(f"Error: Failed to encode image {image_path}")
+        # Create empty json file so the pipeline doesn't break
+        save_bboxes_to_json({}, json_output_path)
+        exit(1)
+    
+    bbox_content = client.ask(PROMPT_MERGE, base64_image)
     bboxes = parse_bboxes(bbox_content)
     
     if bboxes:

screencoder/main.py

@@ -85,8 +85,15 @@ def generate_html_for_demo(image_path, instructions):
     try:
         # 1. Copy user-uploaded image to the temp input directory
         new_image_path = tmp_dir / f"{run_id}.png"
+        print(f"Debug - main.py: saving image from {image_path} to {new_image_path}")
+        print(f"Debug - main.py: image_path exists: {Path(image_path).exists()}")
+        
         img = Image.open(image_path)
         img.save(new_image_path, "PNG")
+        
+        print(f"Debug - main.py: saved image, new_image_path exists: {new_image_path.exists()}")
+        if new_image_path.exists():
+            print(f"Debug - main.py: saved image size: {new_image_path.stat().st_size} bytes")
 
         # 2. Run the processing scripts in sequence
         run_script_with_run_id("UIED/run_single.py", run_id)

screencoder/utils.py

@@ -11,19 +11,30 @@ import numpy as np
 
 def encode_image(image):
     if type(image) == str:
+        print(f"Debug - encode_image: trying to encode {image}")
+        print(f"Debug - encode_image: file exists: {os.path.exists(image)}")
+        if os.path.exists(image):
+            print(f"Debug - encode_image: file size: {os.path.getsize(image)} bytes")
+        
         try: 
             with open(image, "rb") as image_file:
                 encoding = base64.b64encode(image_file.read()).decode('utf-8')
+                print(f"Debug - encode_image: successfully encoded, length: {len(encoding)}")
+                return encoding
         except Exception as e:
-            print(e)
-            with open(image, "r", encoding="utf-8") as image_file:
-                encoding = base64.b64encode(image_file.read()).decode('utf-8')
-        return encoding
+            print(f"Error encoding image {image}: {e}")
+            return None
     
     else:
-        buffered = io.BytesIO()
-        image.save(buffered, format="PNG")
-        return base64.b64encode(buffered.getvalue()).decode('utf-8')
+        try:
+            buffered = io.BytesIO()
+            image.save(buffered, format="PNG")
+            encoding = base64.b64encode(buffered.getvalue()).decode('utf-8')
+            print(f"Debug - encode_image: successfully encoded PIL image, length: {len(encoding)}")
+            return encoding
+        except Exception as e:
+            print(f"Error encoding PIL image: {e}")
+            return None
 
 def image_mask(image_path: str, bbox_normalized: tuple[int, int, int, int]) -> Image.Image:
     """Creates a mask on the image in the specified normalized bounding box."""