Fix image path for Gradio rendering

app.py

@@ -1,8 +1,7 @@
 import gradio as gr
-import os
+import re
 
-
-def load_markdown():
+def load_and_fix_markdown():
     with open("README.md", "r", encoding="utf-8") as f:
         content = f.read()
     
@@ -10,11 +9,24 @@ def load_markdown():
     if content.startswith("---"):
         parts = content.split("---", 2)
         if len(parts) >= 3:
-            return parts[2].strip()
+            content = parts[2].strip()
+    
+    # 修复所有图片路径 - 这是关键！
+    def replace_path(match):
+        path = match.group(1)
+        # 如果是绝对URL，保留域名但简化路径
+        if path.startswith("https://huggingface.co/spaces/"):
+            return f'![](/assets/{path.split("assets/")[-1]})'
+        # 如果是相对路径，转换为应用根路径
+        elif "assets/" in path:
+            return f'![](/assets/{path.split("assets/")[-1]})'
+        return match.group(0)
+    
+    content = re.sub(r'!\[.*?\]\((.*?)\)', replace_path, content)
     
     return content
 
 with gr.Blocks() as demo:
-    gr.Markdown(load_markdown())
+    gr.Markdown(load_and_fix_markdown())
 
-demo.launch(allowed_paths=['./'])
+demo.launch()