Update src/streamlit_app.py

src/streamlit_app.py

@@ -92,27 +92,32 @@ def process_html_with_images(html_content, temp_dir, image_mapping):
         absolute_path = os.path.abspath(os.path.join(temp_dir, relative_path))
         file_url = f"file://{absolute_path}"
         
+        # Escape the filename for regex
+        escaped_name = re.escape(original_name)
+        
         # Replace various image reference patterns
-        # Pattern 1: src="filename" or src='filename'
+        # Match filename with or without directory paths (images/, src/images/, ./images/, etc.)
+        
+        # Pattern 1: src with any path prefix
         html_content = re.sub(
-            rf'src=["\'](?:\./)?{re.escape(original_name)}["\']',
-            f'src="{file_url}"',
+            rf'src=(["\'])(?:[^"\']*/)?' + escaped_name + r'\1',
+            f'src=\\1{file_url}\\1',
             html_content,
             flags=re.IGNORECASE
         )
         
-        # Pattern 2: background-image: url(filename)
+        # Pattern 2: url() with any path prefix
         html_content = re.sub(
-            rf'url\(["\']?(?:\./)?{re.escape(original_name)}["\']?\)',
+            rf'url\((["\']?)(?:[^)"\']*/)?{escaped_name}\1\)',
             f'url("{file_url}")',
             html_content,
             flags=re.IGNORECASE
         )
         
-        # Pattern 3: href for links
+        # Pattern 3: href with any path prefix
         html_content = re.sub(
-            rf'href=["\'](?:\./)?{re.escape(original_name)}["\']',
-            f'href="{file_url}"',
+            rf'href=(["\'])(?:[^"\']*/)?' + escaped_name + r'\1',
+            f'href=\\1{file_url}\\1',
             html_content,
             flags=re.IGNORECASE
         )