Fix image path for Gradio rendering

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 import re
 import os
+import base64
 
 def load_and_fix_markdown():
     with open("README.md", "r", encoding="utf-8") as f:
@@ -12,36 +13,56 @@ def load_and_fix_markdown():
         if len(parts) >= 3:
             content = parts[2].strip()
     
-    # 修复所有图片路径
-    def replace_path(match):
+    # 方法1: 使用HTML img标签
+    def replace_with_html(match):
         alt_text = match.group(1) or ""
         path = match.group(2)
         
-        # 如果已经是完整URL，保持不变
-        if path.startswith(("http://", "https://")):
-            return f'![{alt_text}]({path})'
-        
-        # 处理相对路径
+        # 处理本地文件路径
         clean_path = path.lstrip('./')
         
-        # 确保路径存在
+        # 尝试找到实际文件
+        actual_path = None
         if os.path.exists(clean_path):
-            return f'![{alt_text}](/file={clean_path})'
+            actual_path = clean_path
         else:
-            # 如果文件不存在，尝试在assets目录中查找
+            # 尝试在assets目录中查找
             filename = os.path.basename(clean_path)
-            assets_path = f"assets/{filename}"
-            if os.path.exists(assets_path):
-                return f'![{alt_text}](/file={assets_path})'
-            else:
-                # 文件不存在，返回原始路径（可能会有警告）
-                return f'![{alt_text}]({path})'
+            possible_paths = [
+                f"assets/{filename}",
+                f"assets/{clean_path}",
+                f"images/{filename}",
+                filename
+            ]
+            for p in possible_paths:
+                if os.path.exists(p):
+                    actual_path = p
+                    break
+        
+        if actual_path:
+            # 使用HTML img标签
+            return f'<img src="/file={actual_path}" alt="{alt_text}" style="max-width: 100%; height: auto; display: block; margin: 10px 0;">'
+        else:
+            # 如果找不到文件，保留原始URL（可能是外部URL）
+            return f'![{alt_text}]({path})'
     
-    content = re.sub(r'!\[(.*?)\]\((.*?)\)', replace_path, content)
+    content = re.sub(r'!\[(.*?)\]\((.*?)\)', replace_with_html, content)
     return content
 
 with gr.Blocks() as demo:
-    gr.Markdown(load_and_fix_markdown())
+    # 使用HTML组件而不是Markdown组件
+    gr.HTML("""
+    <div style="max-width: 800px; margin: 0 auto; padding: 20px;">
+        <div id="markdown-content">
+    """)
+    
+    # 动态加载内容
+    gr.HTML(load_and_fix_markdown())
+    
+    gr.HTML("""
+        </div>
+    </div>
+    """)
 
-# 允许访问当前目录和可能的assets目录
-demo.launch(allowed_paths=["./", "./assets"])
+# 确保允许访问所有可能的路径
+demo.launch(allowed_paths=["./", "./assets", "./images", "./static"])