import gradio as gr
import sys
import os

# 1. 确保可以导入本地模块
gradio_dir = os.path.dirname(__file__)
sys.path.append(gradio_dir)

# 2. 导入 rag.py 中的 demo（gr.Blocks 或 gr.Interface 实例）
from rag import demo

# 3. 启动 Gradio 应用
if __name__ == "__main__":
    # Hugging Face Spaces 推荐 server_name=\"0.0.0.0\"，端口 7860
    # queue() 支持多用户排队
    demo.queue().launch(
        server_name="0.0.0.0",
        server_port=7860,
        share=False,  # Spaces 不需要 share
        show_api=False,  # 可选：不显示 API 文档
        max_threads=40   # 可选：根据需要调整并发线程数
    )