Update src/streamlit_app.py

src/streamlit_app.py

@@ -79,23 +79,43 @@ def save_uploaded_images(images, temp_dir):
         
         # Create mapping
         image_mapping[image.name] = f"images/{image.name}"
+        print(f"Saved image: {image.name} -> {image_path}")
     
     return image_mapping
 
-def process_html_with_images(html_content, image_mapping):
-    """Process HTML to handle image references"""
-    for original_name, new_path in image_mapping.items():
-        # Handle various image reference patterns
-        patterns = [
-            (f'src="{original_name}"', f'src="{new_path}"'),
-            (f"src='{original_name}'", f"src='{new_path}'"),
-            (f'href="{original_name}"', f'href="{new_path}"'),
-            (f"href='{original_name}'", f"href='{new_path}'"),
-        ]
+def process_html_with_images(html_content, temp_dir, image_mapping):
+    """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" or src='filename'
+        html_content = re.sub(
+            f'src=["\'](?:\.\/)?{re.escape(original_name)}["\']',
+            f'src="{file_url}"',
+            html_content,
+            flags=re.IGNORECASE
+        )
         
-        for old_pattern, new_pattern in patterns:
-            if old_pattern in html_content:
-                html_content = html_content.replace(old_pattern, new_pattern)
+        # Pattern 2: background-image: url(filename)
+        html_content = re.sub(
+            f'url\(["\']?(?:\.\/)?{re.escape(original_name)}["\']?\)',
+            f'url("{file_url}")',
+            html_content,
+            flags=re.IGNORECASE
+        )
+        
+        # Pattern 3: 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
 
@@ -380,8 +400,15 @@ with tab1:
                     # Process images if uploaded
                     if uploaded_images:
                         image_mapping = save_uploaded_images(uploaded_images, temp_dir)
-                        html_content = process_html_with_images(html_content, image_mapping)
+                        html_content = process_html_with_images(html_content, temp_dir, image_mapping)
                         st.info(f"📷 Processed {len(uploaded_images)} image(s)")
+                        # Debug info
+                        with st.expander("🔍 Debug: Image Mapping"):
+                            for orig, new in image_mapping.items():
+                                st.text(f"{orig} -> {new}")
+                                full_path = os.path.join(temp_dir, new)
+                                st.text(f"Full path: {full_path}")
+                                st.text(f"Exists: {os.path.exists(full_path)}")
                     
                     # Convert to PDF
                     pdf_bytes, error = convert_html_to_pdf(html_content, aspect_ratio_file, temp_dir)
@@ -527,8 +554,15 @@ with tab2:
                 processed_html = html_code
                 if uploaded_images_text:
                     image_mapping = save_uploaded_images(uploaded_images_text, temp_dir)
-                    processed_html = process_html_with_images(html_code, image_mapping)
+                    processed_html = process_html_with_images(html_code, temp_dir, image_mapping)
                     st.info(f"📷 Processed {len(uploaded_images_text)} image(s)")
+                    # Debug info
+                    with st.expander("🔍 Debug: Image Mapping"):
+                        for orig, new in image_mapping.items():
+                            st.text(f"{orig} -> {new}")
+                            full_path = os.path.join(temp_dir, new)
+                            st.text(f"Full path: {full_path}")
+                            st.text(f"Exists: {os.path.exists(full_path)}")
                 
                 # Convert to PDF
                 pdf_bytes, error = convert_html_to_pdf(processed_html, aspect_ratio_text, temp_dir)