Update gradio demo

README.md

@@ -58,6 +58,7 @@ Input Data:
 
 #### Inference with M.2 Accelerator card
 ```
+$cd python
 $python3 gradio_demo.py
 [INFO] Available providers:  ['AXCLRTExecutionProvider']
 [INFO] Using provider: AXCLRTExecutionProvider
@@ -68,10 +69,14 @@ $python3 gradio_demo.py
 [INFO] SOC Name: AX650N
 [INFO] VNPU type: VNPUType.DISABLED
 [INFO] Compiler version: 5.0-patch1 2295293f
-* Running on local URL:  http://0.0.0.0:7860
-* To create a public link, set `share=True` in `launch()`.
+
+==================================================
+🌐 AI图片上色 Web UI 已启动！
+🔗 本地访问: http://127.0.0.1:7860
+🔗 局域网访问: http://10.126.33.124:7860
+==================================================
 ```
-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
+Then open the link in the browser to use the web UI:
 
 ![gradio demo](./assert/gradio_demo.JPG)
 

python/gradio_demo.py

@@ -5,6 +5,7 @@ from PIL import Image, ImageEnhance
 import numpy as np
 import axengine as axe
 import cv2
+import socket
 
 # ==============================
 # 模拟上色函数（请替换为你的实际模型）
@@ -172,7 +173,39 @@ with gr.Blocks(title="AI 图片上色工具") as demo:
         inputs=[input_image, model_choice],
         outputs=[output_image, download_btn]
     )
+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, theme=gr.themes.Soft(), css=custom_css)
+    # 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("🌐 AI图片上色 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()
+    )