Spaces:
Runtime error
Runtime error
| """Entry point for the Iqbal Poetry RAG application.""" | |
| import os | |
| import sys | |
| import requests | |
| import subprocess | |
| from app.RAGSystem import IqbalRAGSystem | |
| from app.gradio_interface import launch_gradio_app | |
| from app.config import JSON_FILE_PATH | |
| def check_ollama_availability(): | |
| """Check if Ollama is running and has the required models.""" | |
| try: | |
| # First check if Ollama service is running | |
| response = requests.get("http://localhost:11434/api/tags", timeout=5) | |
| if response.status_code == 200: | |
| # Check available models | |
| available_models = [model["name"].lower() for model in response.json().get("models", [])] | |
| if "llama3" not in available_models: | |
| print("Warning: llama3 model not found in Ollama.") | |
| # Try to check again with list command | |
| try: | |
| result = subprocess.run(["ollama", "list"], capture_output=True, text=True) | |
| if "llama3" in result.stdout.lower(): | |
| print("Model found via command line. Proceeding...") | |
| return True | |
| else: | |
| print("Please install it with: ollama pull llama3") | |
| return False | |
| except Exception: | |
| print("Could not check models via command line.") | |
| return False | |
| return True | |
| else: | |
| print(f"Warning: Ollama is running but API call failed with status {response.status_code}.") | |
| return False | |
| except requests.exceptions.ConnectionError: | |
| print("Error: Ollama server is not running. Please start Ollama with:") | |
| print("ollama serve") | |
| return False | |
| except Exception as e: | |
| print(f"Error checking Ollama: {str(e)}") | |
| return False | |
| def check_data_file(): | |
| """Check if the required data file exists.""" | |
| if not os.path.exists(JSON_FILE_PATH): | |
| print(f"Error: Required data file not found at {JSON_FILE_PATH}") | |
| print("Please ensure the iqbal_poems_rag.json file is present in the data/processed_data directory.") | |
| return False | |
| return True | |
| if __name__ == "__main__": | |
| # Check if data file exists | |
| if not check_data_file(): | |
| sys.exit(1) | |
| # Check if Ollama is available | |
| ollama_available = check_ollama_availability() | |
| if not ollama_available: | |
| print("Warning: Proceeding without verifying Ollama setup...") | |
| try: | |
| # Initialize RAG system (this will build the vector store if needed) | |
| print("Initializing Iqbal Poetry RAG system...") | |
| rag_system = IqbalRAGSystem() | |
| # Launch Gradio app | |
| print("Launching Gradio interface...") | |
| print(f"Access the application at http://localhost:{os.getenv('PORT', 7860)}") | |
| launch_gradio_app(system=rag_system) | |
| except Exception as e: | |
| print(f"Error: Failed to start the application: {str(e)}") | |
| sys.exit(1) | |