Update api.py

api.py

@@ -44,24 +44,46 @@ def save_uploaded_images(images: List[UploadFile], temp_dir: str):
     return image_mapping
 
 def process_html_with_images(html_content: str, temp_dir: str, image_mapping: dict):
-    """Process HTML to handle image references"""
-    # Replace image references in HTML
-    for original_name, new_path in image_mapping.items():
-        # Handle various image reference patterns
-        patterns = [
-            f'src="{original_name}"',
-            f"src='{original_name}'",
-            f'src={original_name}',
-            f'href="{original_name}"',
-            f"href='{original_name}'"
-        ]
+    """Process HTML to handle image references with absolute file paths"""
+    import re
+    
+    for original_name, relative_path in image_mapping.items():
+        # Get absolute path for the image
+        absolute_path = os.path.abspath(os.path.join(temp_dir, relative_path))
+        file_url = f"file://{absolute_path}"
+        
+        # Replace various image reference patterns
+        # Pattern 1: src="filename"
+        html_content = re.sub(
+            f'src=["\'](?:\.\/)?{re.escape(original_name)}["\']',
+            f'src="{file_url}"',
+            html_content,
+            flags=re.IGNORECASE
+        )
+        
+        # Pattern 2: src='filename'
+        html_content = re.sub(
+            f"src=['\"](?:\.\/)?{re.escape(original_name)}['\"]",
+            f'src="{file_url}"',
+            html_content,
+            flags=re.IGNORECASE
+        )
+        
+        # Pattern 3: background-image: url(filename)
+        html_content = re.sub(
+            f'url\(["\']?(?:\.\/)?{re.escape(original_name)}["\']?\)',
+            f'url("{file_url}")',
+            html_content,
+            flags=re.IGNORECASE
+        )
         
-        for pattern in patterns:
-            if pattern in html_content:
-                html_content = html_content.replace(
-                    pattern,
-                    pattern.replace(original_name, new_path)
-                )
+        # Pattern 4: href for links
+        html_content = re.sub(
+            f'href=["\'](?:\.\/)?{re.escape(original_name)}["\']',
+            f'href="{file_url}"',
+            html_content,
+            flags=re.IGNORECASE
+        )
     
     return html_content