Fix image path for Gradio rendering

README.md

@@ -80,7 +80,7 @@ This replaces labor-intensive human labeling with **autonomous, human-aligned re
 - At inference, **add vectors to internal activations** (e.g., Layer 18 of Llama 3.1) to guide behavior.
 - **No fine-tuning needed**—works on closed-source models like GPT-5.
 
-![MENTOR Architecture](assets/mentor_arch.png)
+![MENTOR Architecture](/file=./assets/mentor_arch.png)
 
 > *Figure: The MENTOR framework (from paper). Shell implements this full pipeline.*
 

app.py

@@ -1,9 +1,6 @@
 import gradio as gr
-import re
-import os
-import base64
 
-def load_and_fix_markdown():
+def load_markdown():
     with open("README.md", "r", encoding="utf-8") as f:
         content = f.read()
     
@@ -13,56 +10,9 @@ def load_and_fix_markdown():
         if len(parts) >= 3:
             content = parts[2].strip()
     
-    # 方法1: 使用HTML img标签
-    def replace_with_html(match):
-        alt_text = match.group(1) or ""
-        path = match.group(2)
-        
-        # 处理本地文件路径
-        clean_path = path.lstrip('./')
-        
-        # 尝试找到实际文件
-        actual_path = None
-        if os.path.exists(clean_path):
-            actual_path = clean_path
-        else:
-            # 尝试在assets目录中查找
-            filename = os.path.basename(clean_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_with_html, content)
     return content
 
 with gr.Blocks() as demo:
-    # 使用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>
-    """)
+    gr.Markdown(load_markdown())
 
-# 确保允许访问所有可能的路径
-demo.launch(allowed_paths=["./", "./assets", "./images", "./static"])
+demo.launch(allowed_paths=["./", "./assets"])