Update gradio demo

README.md

@@ -111,10 +111,13 @@ $ python gradio_demo.py
 [INFO] SOC Name: AX650N
 [INFO] VNPU type: VNPUType.DISABLED
 [INFO] Compiler version: 4.2 6bff2f67
-* Running on local URL:  http://0.0.0.0:7860
-* To create a public link, set `share=True` in `launch()`.
-```
-Then use the M.2 Accelerator card IP instead of the 0.0.0.0, and use chrome open the URL: http://[your ip]:7860
+==================================================
+🌐 SuperResolution 超分辨率 Web UI 已启动！
+🔗 本地访问: http://127.0.0.1:7860
+🔗 局域网访问: http://10.126.33.124:7860
+==================================================
+
+Then open the link in the browser to use the web UI:
 
 ![gradio_demo](./assert/gradio_demo.jpg)
 

python/gradio_demo.py

@@ -7,6 +7,7 @@ import time
 import axengine as axe
 import common
 import imgproc
+import socket
 
 rgb_range=255
 scale=2
@@ -227,7 +228,7 @@ def show_sr():
 # ======================
 # Gradio UI
 # ======================
-with gr.Blocks(title="超分辨率可视化工具", theme=gr.themes.Soft()) as demo:
+with gr.Blocks(title="超分辨率可视化工具") as demo:
     gr.Markdown("## 🚀 超分辨率模型效果可视化")
     gr.Markdown("上传图片或视频，选择模型，点击箭头切换原图/超分图！")
 
@@ -321,6 +322,39 @@ with gr.Blocks(title="超分辨率可视化工具", theme=gr.themes.Soft()) as d
             download_video
         ]
     )
+def get_local_ip():
+    """获取本机局域网IP地址"""
+    try:
+        # 创建一个UDP连接（不会真正发送数据）
+        with socket.socket(socket.AF_INET, socket.SOCK_DGRAM) as s:
+            s.connect(("8.8.8.8", 80))  # 连接到公共DNS（Google）
+            ip = s.getsockname()[0]
+        return ip
+    except Exception:
+        # 回退到 localhost
+        return "127.0.0.1"
+
 
 if __name__ == "__main__":
-    demo.launch(server_name="0.0.0.0", server_port=7860)
+    # demo.launch(server_name="0.0.0.0", server_port=7860, theme=gr.themes.Soft())
+
+    server_port = 7860
+    server_name = "0.0.0.0"
+    
+    # 获取本机IP
+    local_ip = get_local_ip()
+    
+    # 打印可点击的URL（大多数终端支持点击）
+    print("\n" + "="*50)
+    print("🌐 SuperResolution 超分辨率 Web UI 已启动！")
+    print(f"🔗 本地访问: http://127.0.0.1:{server_port}")
+    if local_ip != "127.0.0.1":
+        print(f"🔗 局域网访问: http://{local_ip}:{server_port}")
+    print("="*50 + "\n")
+
+    # 启动Gradio应用
+    demo.launch(
+        server_name=server_name,
+        server_port=server_port,
+        theme=gr.themes.Soft()
+    )