Fix image path for Gradio rendering

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 import re
+import os
 
 def load_and_fix_markdown():
     with open("README.md", "r", encoding="utf-8") as f:
@@ -11,22 +12,36 @@ def load_and_fix_markdown():
         if len(parts) >= 3:
             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)
+        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('./')
+        
+        # 确保路径存在
+        if os.path.exists(clean_path):
+            return f'![{alt_text}](/file={clean_path})'
+        else:
+            # 如果文件不存在，尝试在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})'
     
+    content = re.sub(r'!\[(.*?)\]\((.*?)\)', replace_path, content)
     return content
 
 with gr.Blocks() as demo:
     gr.Markdown(load_and_fix_markdown())
 
-demo.launch()
+# 允许访问当前目录和可能的assets目录
+demo.launch(allowed_paths=["./", "./assets"])