| """ | |
| Initialize tab functionality for the Gradio app | |
| """ | |
| import gradio as gr | |
| from utils.rag_system import DocumentIngestion, RAGSystem | |
| def initialize_systems(global_vars): | |
| """Initialize the RAG systems""" | |
| try: | |
| print("🚀 Initializing document ingestion system...") | |
| global_vars['doc_ingestion'] = DocumentIngestion() | |
| print("🚀 Initializing RAG system...") | |
| global_vars['rag_system'] = RAGSystem() | |
| return "✅ Systems initialized successfully! You can now upload documents." | |
| except Exception as e: | |
| error_msg = f"❌ Error initializing systems: {str(e)}\n\n" | |
| if "sentence-transformers" in str(e): | |
| error_msg += """ | |
| **Possible solutions:** | |
| 1. Install sentence-transformers: `pip install sentence-transformers` | |
| 2. Or provide OpenAI API key in environment variables | |
| 3. Check that PyTorch is properly installed | |
| **For deployment:** | |
| - Ensure requirements.txt includes: sentence-transformers, torch, transformers | |
| """ | |
| return error_msg | |
| def create_initialize_tab(global_vars): | |
| """Create the Initialize System tab""" | |
| with gr.Tab("🚀 Initialize System", id="init"): | |
| gr.Markdown(""" | |
| ### Step 1: Initialize the System | |
| Click the button below to initialize the AI models and embedding systems. | |
| This may take a few moments on first run as models are downloaded. | |
| """) | |
| init_btn = gr.Button( | |
| "🚀 Initialize Systems", | |
| variant="primary", | |
| size="lg" | |
| ) | |
| init_status = gr.Textbox( | |
| label="Initialization Status", | |
| interactive=False, | |
| lines=8, | |
| placeholder="Click 'Initialize Systems' to start..." | |
| ) | |
| init_btn.click( | |
| lambda: initialize_systems(global_vars), | |
| outputs=init_status | |
| ) | |