增强 vLLM 模式的导入处理，添加命令行参数支持以便在未安装 vLLM 时运行界面预览

app.py

@@ -6,17 +6,30 @@ Gradio 多模态聊天界面：直接在 app.py 内部调用 vLLM.LLM 进行推
 
 import base64
 import os
+import sys
 import threading
 import time
 from typing import Optional, Tuple
 
 import gradio as gr
 
+# 检查命令行参数，在导入 vllm 之前确定是否启用
+# 这样可以在没有安装 vllm 的情况下运行界面预览
+if "--no-vllm" in sys.argv:
+    os.environ["ENABLE_VLLM"] = "false"
+
 # 检查是否启用 vLLM 模式
 ENABLE_VLLM = os.getenv("ENABLE_VLLM", "true").lower() in ("true", "1", "yes")
 
 if ENABLE_VLLM:
-    from vllm import LLM, SamplingParams
+    try:
+        from vllm import LLM, SamplingParams
+    except ImportError:
+        print("[WARNING] 无法导入 vllm，自动切换到界面预览模式")
+        print("[INFO] 如需使用 vLLM，请先安装: pip install vllm")
+        ENABLE_VLLM = False
+        LLM = None
+        SamplingParams = None
 else:
     LLM = None
     SamplingParams = None
@@ -389,10 +402,8 @@ if __name__ == "__main__":
     
     args = parser.parse_args()
     
-    # 如果指定了 --no-vllm，覆盖环境变量
-    if args.no_vllm:
-        global ENABLE_VLLM
-        ENABLE_VLLM = False
+    # --no-vllm 参数已在文件开头处理，这里只是提示
+    if args.no_vllm and not ENABLE_VLLM:
         print("[INFO] 已禁用 vLLM，运行在界面预览模式")
     
     _set_llm_args(