Function create_visualization_interface

startup/R3-RAG.py:402–652

创建Gradio界面

()

            return f"❌ Error processing question: {str(e)}", ""

def create_visualization_interface():
    """创建Gradio界面"""
    
    # 创建模型管理器实例
    model_manager = ModelManager()
    
    # 获取本机IP
    local_ip = get_local_ip()
    
    # 自定义CSS样式
    custom_css = """
    @import url('https://fonts.googleapis.com/css2?family=Times+New+Roman&display=swap');
    
    * {
        font-family: 'SimHei', 'Times New Roman', serif !important;
    }
    
    .gradio-container {
        font-family: 'SimHei', 'Times New Roman', serif !important;
    }
    
    .gr-textbox input, .gr-textbox textarea {
        font-family: 'SimHei', 'Times New Roman', serif !important;
        font-size: 14px !important;
    }
    
    .gr-button {
        font-family: 'SimHei', Arial, sans-serif !important;
        font-weight: bold !important;
    }
    
    h1, h2, h3, h4, h5, h6 {
        font-family: 'SimHei', Arial, sans-serif !important;
    }
    
    .markdown {
        font-family: 'SimHei', 'Times New Roman', serif !important;
    }
    """
    
    # 创建Gradio界面
    with gr.Blocks(title="R3-RAG Visualization", theme=gr.themes.Soft(), css=custom_css) as demo:
        # 论文标题和介绍
        paper_title = "R3-RAG: Learning Step-by-Step Reasoning and Retrieval for LLMs via Reinforcement Learning"
        authors = "Yuan Li, Qi Luo, Xiaonan Li, Bufan Li, Qinyuan Cheng, Bo Wang, Yining Zheng, Yuxin Wang, Zhangyue Yin, Xipeng Qiu"
        affiliation = "School of Computer Science, Fudan University"
        github_url = "https://github.com/Yuan-Li-FNLP/R3-RAG"
        
        gr.Markdown(f"""
        # {paper_title}
        
        **Authors:** {authors}  
        **Affiliation:** {affiliation}  
        **GitHub:** [{github_url}]({github_url})  
        **Server Address:** `{local_ip}:7860`
        
        ---